Unit tests for Feature 10910

Added unit tests in NBI for feature 10910 - Migration of
Openstack based VM instances

Change-Id: Ied4ef3cb4b9754ace4667fe43abc303fc374e9b8
Signed-off-by: elumalai <deepika.e@tataelxsi.co.in>

diff --git a/osm_nbi/tests/test_instance_topics.py b/osm_nbi/tests/test_instance_topics.py
index 2ad596d..2c10632 100644 (file)
--- a/osm_nbi/tests/test_instance_topics.py
+++ b/osm_nbi/tests/test_instance_topics.py
@@ -281,6 +281,104 @@ class TestNsLcmOpTopic(unittest.TestCase):
             "Database record must contain 'updateType=REMOVE_VNF'",
         )
 
+    def test_migrate(self):
+        vnfr_id = self.db.get_list("vnfrs")[0]["_id"]
+        session = {}
+        self.db.set_one(
+            "nsrs",
+            {"_id": self.nsr_id},
+            {"_admin.nsState": "INSTANTIATED"},
+        )
+        session = {
+            "force": False,
+            "admin": False,
+            "public": False,
+            "project_id": [self.nsr_project],
+            "method": "write",
+        }
+        rollback = []
+        headers = {}
+
+        with self.subTest(i=1, t="Migration for Specific VM"):
+            indata = {
+                "lcmOperationType": "migrate",
+                "nsInstanceId": self.nsr_id,
+                "migrateToHost":"sample02",
+                "vdu": {
+                    "vduCountIndex": 0,
+                    "vduId": "mgmtVM"
+                },
+                "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f"
+            }
+            nslcmop_id, _ = self.nslcmop_topic.new(
+                rollback, session, indata, kwargs=None, headers=headers
+            )
+
+            self.assertEqual(
+                self.db.create.call_count,
+                1,
+                "database create not called, or called more than once",
+            )
+            _call = self.db.create.call_args_list[0]
+            self.assertEqual(
+                _call[0][0], "nslcmops", "nslcmops entry must be created at database"
+            )
+            created_nslcmop = _call[0][1]
+            self.assertEqual(
+                self.nsr_id,
+                created_nslcmop["nsInstanceId"],
+                "mismatch between nsId '_id' in created nslcmop and database nsr",
+            )
+            self.assertTrue(
+                created_nslcmop["lcmOperationType"] == "migrate",
+                "Database record must contain 'lcmOperationType=migrate'",
+            )
+        with self.subTest(i=2, t="Migration of all VDUs in a VNF"):
+            indata = {
+                "lcmOperationType": "migrate",
+                "nsInstanceId": self.nsr_id,
+                "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f"
+            }
+            nslcmop_id, _ = self.nslcmop_topic.new(
+                rollback, session, indata, kwargs=None, headers=headers
+            )
+
+            self.assertEqual(
+                self.db.create.call_count,
+                2,
+                "database create not called, or called more than once",
+            )
+            _call = self.db.create.call_args_list[0]
+            self.assertEqual(
+                _call[0][0], "nslcmops", "nslcmops entry must be created at database"
+            )
+            created_nslcmop = _call[0][1]
+            self.assertEqual(
+                self.nsr_id,
+                created_nslcmop["nsInstanceId"],
+                "mismatch between nsId '_id' in created nslcmop and database nsr",
+            )
+            self.assertTrue(
+                created_nslcmop["lcmOperationType"] == "migrate",
+                "Database record must contain 'lcmOperationType=migrate'",
+            )
+        with self.subTest(i=3, t="Migration failure - vduId not provided in vdu "):
+            indata = {
+                "lcmOperationType": "migrate",
+                "nsInstanceId": self.nsr_id,
+                "migrateToHost":"sample02",
+                "vdu": {
+                    "vduCountIndex": 0
+                },
+                "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f"
+            }
+
+            with self.assertRaises(Exception) as e:
+                nslcmop_id, _ = self.nslcmop_topic.new(
+                rollback, session, indata, kwargs=None, headers=headers
+            )
+            self.assertTrue("Format error at 'vdu' ''vduId' is a required property'" in str(e.exception))
+
 
 class TestNsLcmOpTopicWithMock(unittest.TestCase):
     def setUp(self):