Fix tests

Author: badrogger

tests/cli/mirage_cli_test.py

@@ -15,7 +15,6 @@
     init_boot,
     register_boot,
     signature_boot,
-    migrate_boot,
 )
 
 
@@ -150,25 +149,6 @@ def test_mirage_boot_init(mock_init_core, valid_env_file):
     mock_init_core.assert_called_once_with(valid_env_file)
 
 
-@mock.patch('node_cli.cli.mirage_boot.migrate')
-def test_mirage_boot_migrate(mock_migrate_core, valid_env_file):
-    runner = CliRunner()
-    result = runner.invoke(migrate_boot, ['--yes', valid_env_file])
-
-    assert result.exit_code == 0, f'Output: {result.output}\nException: {result.exception}'
-    mock_migrate_core.assert_called_once_with(valid_env_file, None)
-
-
-@mock.patch('node_cli.cli.mirage_boot.migrate')
-def test_mirage_boot_migrate_pull_config(mock_migrate_core, valid_env_file):
-    runner = CliRunner()
-    schain_name = 'my-schain-config'
-    result = runner.invoke(migrate_boot, ['--yes', '--pull-config', schain_name, valid_env_file])
-
-    assert result.exit_code == 0, f'Output: {result.output}\nException: {result.exception}'
-    mock_migrate_core.assert_called_once_with(valid_env_file, schain_name)
-
-
 @mock.patch('node_cli.cli.mirage_node.migrate_from_boot')
 def test_mirage_node_migrate(mock_migrate_core, valid_env_file):
     runner = CliRunner()

tests/configs/configs_env_validate_test.py

@@ -18,7 +18,7 @@
     MirageUserConfig,
     SkaleUserConfig,
     SyncUserConfig,
-    get_user_config_type,
+    get_user_config_class,
     get_validated_user_config,
     validate_env_type,
 )
@@ -47,7 +47,7 @@ def json(self):
     ids=['regular', 'sync', 'mirage_boot', 'mirage_regular'],
 )
 def test_build_env_params_keys(node_type, is_mirage_boot, expected_type):
-    env_type = get_user_config_type(node_type=node_type, is_mirage_boot=is_mirage_boot)
+    env_type = get_user_config_class(node_type=node_type, is_mirage_boot=is_mirage_boot)
     assert env_type == expected_type
 
 

tests/core/core_mirage_test.py

@@ -1,7 +1,7 @@
 from unittest import mock
 
 from node_cli.configs import SKALE_DIR
-from node_cli.core.mirage_boot import init as init_boot, migrate, update
+from node_cli.core.mirage_boot import init as init_boot, update
 from node_cli.core.mirage_node import restore_mirage, migrate_from_boot
 from node_cli.operations.mirage import MirageUpdateType
 from node_cli.utils.node_type import NodeType
@@ -60,41 +60,6 @@ def test_init_mirage_boot(
     mock_is_alive.assert_called_once_with(node_type=NodeType.MIRAGE, is_mirage_boot=True)
 
 
-@mock.patch('node_cli.utils.decorators.is_user_valid', return_value=True)
-@mock.patch('node_cli.core.mirage_boot.is_base_containers_alive', return_value=True)
-@mock.patch('node_cli.core.mirage_boot.time.sleep')
-@mock.patch('node_cli.core.mirage_boot.migrate_mirage_boot_op')
-@mock.patch('node_cli.core.mirage_boot.compose_node_env')
-def test_migrate_mirage_boot(
-    mock_compose_env,
-    mock_migrate_op,
-    mock_sleep,
-    mock_is_alive,
-    mock_is_user_valid,
-    valid_env_file,
-    inited_node,
-    resource_alloc,
-    meta_file_v3,
-):
-    mock_env = {'ENV_TYPE': 'devnet'}
-    mock_compose_env.return_value = mock_env
-    mock_migrate_op.return_value = True
-    pull_config_for_schain = 'mirage'
-
-    migrate(valid_env_file, pull_config_for_schain)
-
-    mock_compose_env.assert_called_once_with(
-        valid_env_file,
-        inited_node=True,
-        sync_schains=False,
-        pull_config_for_schain=pull_config_for_schain,
-        node_type=NodeType.MIRAGE,
-    )
-    mock_migrate_op.assert_called_once_with(valid_env_file, mock_env)
-    mock_sleep.assert_called_once()
-    mock_is_alive.assert_called_once_with(node_type=NodeType.MIRAGE)
-
-
 @mock.patch('node_cli.utils.decorators.is_user_valid', return_value=True)
 @mock.patch('node_cli.core.mirage_boot.is_base_containers_alive', return_value=True)
 @mock.patch('node_cli.core.mirage_boot.time.sleep')