From d7bcb4c0fe6499c2cca52bcbf8df590a5fe2c336 Mon Sep 17 00:00:00 2001
From: Waqar Ahmed <waqarahmedjoyia@live.com>
Date: Sun, 26 Jan 2025 23:08:24 +0500
Subject: [PATCH] Add integration test for docker migration

---
 tests/api2/test_docker_migration.py | 50 +++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 tests/api2/test_docker_migration.py

diff --git a/tests/api2/test_docker_migration.py b/tests/api2/test_docker_migration.py
new file mode 100644
index 0000000000000..b63ab1a7d7a3e
--- /dev/null
+++ b/tests/api2/test_docker_migration.py
@@ -0,0 +1,50 @@
+import pytest
+
+from middlewared.test.integration.assets.docker import docker
+from middlewared.test.integration.assets.pool import another_pool
+from middlewared.test.integration.utils import call
+
+
+APP_NAME = 'actual-budget'
+APP2_NAME = 'syncthing'
+BACKUP_NAME = 'test_backup'
+ENC_POOL_PASSWORD = 'test1234'
+
+
+@pytest.fixture(scope='module')
+def docker_pool():
+    with another_pool({'name': 'test_docker_migration_pool1'}) as pool:
+        with docker(pool) as docker_config:
+            yield docker_config
+
+
+@pytest.fixture(scope='module')
+def migration_pool():
+    with another_pool({'name': 'test_docker_migration_pool2'}) as pool:
+        yield pool
+
+
+def test_install_docker_apps(docker_pool):
+    call('app.create', {
+        'app_name': APP_NAME,
+        'train': 'community',
+        'catalog_app': 'actual-budget',
+    }, job=True)
+    call('app.create', {
+        'app_name': APP2_NAME,
+        'train': 'stable',
+        'catalog_app': 'syncthing',
+    }, job=True)
+    for app_name in (APP2_NAME, APP_NAME):
+        assert call('app.get_instance', app_name)['name'] == app_name
+
+
+def test_docker_app_migration(docker_pool, migration_pool):
+    try:
+        call('docker.update', {'pool': migration_pool['name'], 'migrate_applications': True}, job=True)
+        assert call('docker.config')['pool'] == migration_pool['name']
+        for app_name in (APP2_NAME, APP_NAME):
+            assert call('app.get_instance', app_name)['name'] == app_name
+    finally:
+        call('docker.update', {'pool': None}, job=True)
+        assert call('docker.config')['pool'] is None