def test_migrate_apps(monkeypatch):
"""Test that migrate_apps() migrates the correct apps."""
migrate_app_mock = Mock()
monkeypatch.setattr('datahub.search.migrate.migrate_app', migrate_app_mock)
apps = {app.name for app in list(get_search_apps())[:2]}
migrate_apps(apps)
assert {args[0][0] for args in migrate_app_mock.call_args_list} == apps
def handle(self, *args, **options):
"""Executes the command."""
apps = get_search_apps_by_name(options['model'])
if not are_apps_initialised(apps):
raise CommandError(
f'Index and mapping not initialised, please run `init_es` first.',
)
with advisory_lock('leeloo_migrate_es'):
migrate_apps(apps)
def handle(self, *args, **options):
"""Executes the command."""
apps = get_search_apps_by_name(options['model'])
with advisory_lock('migrate-opensearch-lock-id'):
migrate_apps(apps)