Notification stuffs.

[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / tests / test_instantiate_vnf.py

diff --git a/lcm/lcm/nf/tests/test_instantiate_vnf.py b/lcm/lcm/nf/tests/test_instantiate_vnf.py
index e698c4d..4784fa2 100644 (file)
--- a/lcm/lcm/nf/tests/test_instantiate_vnf.py
+++ b/lcm/lcm/nf/tests/test_instantiate_vnf.py
@@ -24,10 +24,11 @@ from lcm.nf.biz.instantiate_vnf import InstantiateVnf
 from lcm.nf.const import c1_data_get_tenant_id, c4_data_create_network, c2_data_create_volume, \\r
     c5_data_create_subnet, c3_data_get_volume, c6_data_create_port, c7_data_create_flavor, c8_data_list_image, \\r
     c9_data_create_vm, c10_data_get_vm, inst_req_data, vnfpackage_info\r
-from lcm.pub.database.models import NfInstModel, JobStatusModel\r
+from lcm.pub.database.models import NfInstModel, JobStatusModel, SubscriptionModel\r
 from lcm.pub.utils import restcall\r
 from lcm.pub.utils.jobutil import JobUtil\r
 from lcm.pub.utils.timeutil import now_time\r
+from lcm.pub.utils.notificationsutil import NotificationsUtil\r
 from lcm.pub.vimapi import api\r
 \r
 \r
@@ -68,22 +69,22 @@ class TestNFInstantiate(TestCase):
 \r
     @mock.patch.object(InstantiateVnf, 'run')\r
     def test_instantiate_vnf(self, mock_run):\r
-        NfInstModel(nfinstid='12', nf_name='VNF1', status="UN_INSTANTIATED").save()\r
+        NfInstModel(nfinstid='12', nf_name='VNF1', status='UN_INSTANTIATED').save()\r
         mock_run.re.return_value = None\r
-        response = self.client.post("/api/vnflcm/v1/vnf_instances/12/instantiate", data=inst_req_data, format='json')\r
+        response = self.client.post('/api/vnflcm/v1/vnf_instances/12/instantiate', data=inst_req_data, format='json')\r
         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)\r
 \r
     def test_instantiate_vnf_when_inst_id_not_exist(self):\r
         self.nf_inst_id = str(uuid.uuid4())\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         jobs = JobStatusModel.objects.filter(jobid=self.job_id,\r
                                              progress=0,\r
-                                             descp="INST_VNF_READY")\r
+                                             descp='INST_VNF_READY')\r
         self.assertEqual(1, len(jobs))\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 255, "VNF nf_inst_id is not exist.")\r
+        self.assert_job_result(self.job_id, 255, 'VNF nf_inst_id is not exist.')\r
 \r
     def test_instantiate_vnf_when_already_instantiated(self):\r
         NfInstModel.objects.create(nfinstid='1111',\r
@@ -99,14 +100,14 @@ class TestNFInstantiate(TestCase):
                                    create_time=now_time())\r
         self.nf_inst_id = '1111'\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         jobs = JobStatusModel.objects.filter(jobid=self.job_id,\r
                                              progress=0,\r
-                                             descp="INST_VNF_READY")\r
+                                             descp='INST_VNF_READY')\r
         self.assertEqual(1, len(jobs))\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 255, "VNF instantiationState is not NOT_INSTANTIATED.")\r
+        self.assert_job_result(self.job_id, 255, 'VNF instantiationState is not NOT_INSTANTIATED.')\r
 \r
     @mock.patch.object(restcall, 'call_req')\r
     def test_instantiate_vnf_when_get_packageinfo_by_csarid_failed(self, mock_call_req):\r
@@ -125,10 +126,10 @@ class TestNFInstantiate(TestCase):
         mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid]\r
         self.nf_inst_id = '1111'\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 255, "Failed to query vnf CSAR(111) from catalog.")\r
+        self.assert_job_result(self.job_id, 255, 'Failed to query vnf CSAR(111) from catalog.')\r
 \r
     @mock.patch.object(restcall, 'call_req')\r
     def test_instantiate_vnf_when_applay_grant_failed(self, mock_call_req):\r
@@ -148,10 +149,10 @@ class TestNFInstantiate(TestCase):
         mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r2_apply_grant_result]\r
         self.nf_inst_id = '1111'\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 255, "Nf instancing apply grant exception")\r
+        self.assert_job_result(self.job_id, 255, 'Nf instancing apply grant exception')\r
 \r
     @mock.patch.object(restcall, 'call_req')\r
     @mock.patch.object(api, 'call')\r
@@ -173,14 +174,15 @@ class TestNFInstantiate(TestCase):
         mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume]\r
         self.nf_inst_id = '1111'\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 255, "unexpected exception")\r
+        self.assert_job_result(self.job_id, 255, 'unexpected exception')\r
 \r
     @mock.patch.object(restcall, 'call_req')\r
     @mock.patch.object(api, 'call')\r
-    def test_instantiate_vnf_success(self, mock_call, mock_call_req):\r
+    @mock.patch.object(NotificationsUtil, 'post_notification')\r
+    def test_instantiate_vnf_success(self, mock_post_notification, mock_call, mock_call_req):\r
         NfInstModel.objects.create(nfinstid='1111',\r
                                    nf_name='vFW_01',\r
                                    package_id='222',\r
@@ -192,11 +194,36 @@ class TestNFInstantiate(TestCase):
                                    nf_desc='vFW in Nanjing TIC Edge',\r
                                    vnfdid='111',\r
                                    create_time=now_time())\r
+        SubscriptionModel.objects.create(\r
+            subscription_id=str(uuid.uuid4()),\r
+            callback_uri='api/gvnfmdriver/v1/vnfs/lifecyclechangesnotification',\r
+            auth_info=json.JSONEncoder().encode({\r
+                'authType': ['BASIC'],\r
+                'paramsBasic': {\r
+                    'userName': 'username',\r
+                    'password': 'password'\r
+                }\r
+            }),\r
+            notification_types=str([\r
+                'VnfLcmOperationOccurrenceNotification',\r
+                'VnfIdentifierCreationNotification',\r
+                'VnfIdentifierDeletionNotification'\r
+            ]),\r
+            operation_types=str(['INSTANTIATE']),\r
+            operation_states=str(['COMPLETED']),\r
+            vnf_instance_filter=json.JSONEncoder().encode({\r
+                'vnfdIds': ['111'],\r
+                'vnfProductsFromProviders': [],\r
+                'vnfInstanceIds': ['1111'],\r
+                'vnfInstanceNames': [],\r
+            })\r
+        )\r
         r1_get_vnfpackage_by_vnfdid = [0, json.JSONEncoder().encode(vnfpackage_info), '200']\r
         r2_apply_grant_result = [0, json.JSONEncoder().encode(self.grant_result), '200']\r
         r3_all_aai_result = [1, json.JSONEncoder().encode(''), '404']\r
-        r4_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
-        mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r2_apply_grant_result, r3_all_aai_result, r4_lcm_notify_result]\r
+        # r4_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
+        # mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r2_apply_grant_result, r3_all_aai_result, r4_lcm_notify_result]\r
+        mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r2_apply_grant_result, r3_all_aai_result]\r
         mock_call.side_effect = [c1_data_get_tenant_id,\r
                                  c2_data_create_volume, c3_data_get_volume,\r
                                  c4_data_create_network,\r
@@ -204,9 +231,10 @@ class TestNFInstantiate(TestCase):
                                  c6_data_create_port,\r
                                  c7_data_create_flavor,\r
                                  c8_data_list_image, c9_data_create_vm, c10_data_get_vm]\r
+        mock_post_notification.side_effect = None\r
         self.nf_inst_id = '1111'\r
         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)\r
-        JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")\r
+        JobUtil.add_job_status(self.job_id, 0, 'INST_VNF_READY')\r
         data = inst_req_data\r
         InstantiateVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
-        self.assert_job_result(self.job_id, 100, "Instantiate Vnf success.")\r
+        self.assert_job_result(self.job_id, 100, 'Instantiate Vnf success.')\r