Merge pull request #369 from CiscoDevNet/mig_payload

Migration MT-ST new field is aded in payload for export endpoint 20.13
CiscoDevNet · Sep 19, 2023 · ce3b110 · ce3b110
2 parents df86647 + e16d056
commit ce3b110
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/vmngclient/model/tenant.py b/vmngclient/model/tenant.py
@@ -47,3 +47,7 @@ class Tenant(BaseModel):
 
     class Config:
         allow_population_by_field_name = True
+
+
+class MigrationTenant(Tenant):
+    is_destination_overlay_mt: Optional[bool] = Field(alias="isDestinationOverlayMT")
diff --git a/vmngclient/tests/test_tenant_migration_api.py b/vmngclient/tests/test_tenant_migration_api.py
@@ -6,7 +6,7 @@
 from vmngclient.api.task_status_api import Task
 from vmngclient.api.tenant_migration_api import ImportTask, TenantMigrationAPI
 from vmngclient.endpoints.tenant_migration import ImportInfo, MigrationInfo
-from vmngclient.model.tenant import Tenant
+from vmngclient.model.tenant import MigrationTenant, Tenant
 
 
 class TestTenantMigrationAPI(unittest.TestCase):
@@ -20,6 +20,17 @@ def test_export_tenant(self):
         task = self.api.export_tenant(tenant=tenant)
         self.assertIsInstance(task, Task)
 
+    def test_export_migration_tenant(self):
+        tenant = MigrationTenant(
+            desc="Test Tenant",
+            name="test_tenant",
+            subdomain="test_subdomain",
+            org_name="test_org",
+            is_destination_overlay_mt=False,
+        )
+        task = self.api.export_tenant(tenant=tenant)
+        self.assertIsInstance(task, Task)
+
     def test_download(self):
         content = b"\xFFtest_data"
         with tempfile.TemporaryDirectory() as tmpdir: