Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions node_cli/operations/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,6 +71,7 @@
compose_up,
docker_cleanup,
remove_dynamic_containers,
rm_legacy_containers,
system_prune,
)
from node_cli.utils.helper import cleanup_dir_content, rm_dir
Expand Down Expand Up @@ -123,6 +124,7 @@ def wrapper(
@checked_host
def update(settings: BaseNodeSettings, compose_env: dict, node_mode: NodeMode) -> bool:
compose_rm(node_type=NodeType.SKALE, node_mode=node_mode, env=compose_env)
rm_legacy_containers()
remove_dynamic_containers()

sync_skale_node()
Expand Down Expand Up @@ -273,6 +275,7 @@ def init_passive(

def update_passive(settings: BaseNodeSettings, compose_env: dict) -> bool:
compose_rm(env=compose_env, node_type=NodeType.SKALE, node_mode=NodeMode.PASSIVE)
rm_legacy_containers()
remove_dynamic_containers()
cleanup_volume_artifacts(settings.block_device)
download_skale_node(settings.node_version, settings.container_configs_dir or None)
Expand Down Expand Up @@ -326,6 +329,7 @@ def update_passive(settings: BaseNodeSettings, compose_env: dict) -> bool:
def turn_off(compose_env: dict, node_type: NodeType, node_mode: NodeMode) -> None:
logger.info('Turning off the node...')
compose_rm(env=compose_env, node_type=node_type, node_mode=node_mode)
rm_legacy_containers()
remove_dynamic_containers()
logger.info('Node was successfully turned off')

Expand Down
3 changes: 3 additions & 0 deletions node_cli/operations/fair.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@
docker_cleanup,
is_admin_running,
remove_dynamic_containers,
rm_legacy_containers,
start_container_by_name,
stop_container_by_name,
system_prune,
Expand Down Expand Up @@ -201,6 +202,7 @@ def update_fair_boot(
node_mode: NodeMode = NodeMode.ACTIVE,
) -> bool:
compose_rm(node_type=NodeType.FAIR, node_mode=node_mode, env=compose_env)
rm_legacy_containers()
remove_dynamic_containers()
cleanup_volume_artifacts(settings.block_device)

Expand Down Expand Up @@ -260,6 +262,7 @@ def update(
force_skaled_start: bool,
) -> bool:
compose_rm(node_type=NodeType.FAIR, node_mode=node_mode, env=compose_env)
rm_legacy_containers()
if update_type not in (FairUpdateType.INFRA_ONLY, FairUpdateType.FROM_BOOT):
remove_dynamic_containers()

Expand Down
5 changes: 5 additions & 0 deletions node_cli/utils/docker_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,6 +163,11 @@ def remove_telegraf() -> None:
remove_containers(telegraf, timeout=TELEGRAF_REMOVE_TIMEOUT)


def rm_legacy_containers() -> None:
containers = docker_client().containers.list(all=True, filters={'name': 'skale_'})
remove_containers(containers, timeout=DOCKER_DEFAULT_STOP_TIMEOUT)


def remove_containers(containers, timeout):
for container in containers:
safe_rm(container, timeout=timeout)
Expand Down
46 changes: 46 additions & 0 deletions tests/docker_utils_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
docker_cleanup,
get_all_ima_containers,
get_all_skaled_containers,
rm_legacy_containers,
save_container_logs,
safe_rm,
)
Expand Down Expand Up @@ -111,6 +112,27 @@ def _make_container(name: str) -> MagicMock:
return c


def test_get_all_skaled_containers_real(dclient):
containers = []
names = ['sk_skaled_test_chain', 'skale_schain_test_chain']
try:
for name in names:
c = dclient.containers.run('alpine', 'true', name=name, detach=True)
containers.append(c)
for c in containers:
c.wait()
result = get_all_skaled_containers()
result_names = {c.name for c in result}
for name in names:
assert name in result_names
finally:
for c in containers:
try:
c.remove(force=True)
except Exception:
pass


def test_get_all_skaled_containers_both_prefixes():
new_container = _make_container('sk_skaled_chain1')
legacy_container = _make_container('skale_schain_chain2')
Expand Down Expand Up @@ -155,3 +177,27 @@ def fake_list(all=True, filters=None):
assert len(result) == 2
names = {c.name for c in result}
assert names == {'sk_ima_chain1', 'skale_ima_chain2'}


def test_rm_legacy_containers(dclient, removed_containers_folder):
names = ['skale_sync_admin', 'skale_api', 'skale_schain_old']
containers = []
try:
for name in names:
c = dclient.containers.run('alpine', 'true', name=name, detach=True)
containers.append(c)
for c in containers:
c.wait()

rm_legacy_containers()

remaining = dclient.containers.list(all=True, filters={'name': 'skale_'})
remaining_names = {c.name for c in remaining}
for name in names:
assert name not in remaining_names
finally:
for c in containers:
try:
c.remove(force=True)
except Exception:
pass
Loading