+# Copyright (C) 2018 Verizon. All Rights Reserved.\r
+#\r
+# Licensed under the Apache License, Version 2.0 (the "License");\r
+# you may not use this file except in compliance with the License.\r
+# You may obtain a copy of the License at\r
+#\r
+# http://www.apache.org/licenses/LICENSE-2.0\r
+#\r
+# Unless required by applicable law or agreed to in writing, software\r
+# distributed under the License is distributed on an "AS IS" BASIS,\r
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+# See the License for the specific language governing permissions and\r
+# limitations under the License.\r
+\r
+import json\r
+\r
+import mock\r
+from django.test import TestCase, Client\r
+from rest_framework import status\r
+\r
+from lcm.nf.biz.operate_vnf import OperateVnf\r
+from lcm.pub.database.models import NfInstModel, JobStatusModel, VmInstModel\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.vimapi import api\r
+\r
+\r
+class TestNFOperate(TestCase):\r
+ def setUp(self):\r
+ self.client = Client()\r
+\r
+ def tearDown(self):\r
+ VmInstModel.objects.all().delete()\r
+\r
+ def assert_job_result(self, job_id, job_progress, job_detail):\r
+ jobs = JobStatusModel.objects.filter(jobid=job_id,\r
+ progress=job_progress,\r
+ descp=job_detail)\r
+ self.assertEqual(1, len(jobs))\r
+\r
+ def test_operate_vnf_not_found(self):\r
+ req_data = {\r
+ "changeStateTo": "STARTED"\r
+ }\r
+ response = self.client.post("/api/vnflcm/v1/vnf_instances/12/operate", data=req_data, format='json')\r
+ self.failUnlessEqual(status.HTTP_404_NOT_FOUND, response.status_code)\r
+\r
+ def test_operate_vnf_conflict(self):\r
+ req_data = {\r
+ "changeStateTo": "STARTED"\r
+ }\r
+ NfInstModel(nfinstid='12', nf_name='VNF1', status='NOT_INSTANTIATED').save()\r
+ response = self.client.post("/api/vnflcm/v1/vnf_instances/12/operate", data=req_data, format='json')\r
+ self.failUnlessEqual(status.HTTP_409_CONFLICT, response.status_code)\r
+ NfInstModel(nfinstid='12', nf_name='VNF1', status='NOT_INSTANTIATED').delete()\r
+\r
+ @mock.patch.object(OperateVnf, 'run')\r
+ def test_operate_vnf_success(self, mock_run):\r
+ req_data = {\r
+ "changeStateTo": "STARTED"\r
+ }\r
+ NfInstModel(nfinstid='12', nf_name='VNF1', status='INSTANTIATED').save()\r
+ response = self.client.post("/api/vnflcm/v1/vnf_instances/12/operate", data=req_data, format='json')\r
+ mock_run.re.return_value = None\r
+ self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)\r
+ NfInstModel(nfinstid='12', nf_name='VNF1', status='INSTANTIATED').delete()\r
+\r
+ @mock.patch.object(restcall, 'call_req')\r
+ @mock.patch.object(api, 'call')\r
+ def test_operate_vnf_success_start(self, mock_call, mock_call_req):\r
+ NfInstModel.objects.create(nfinstid='1111',\r
+ nf_name='2222',\r
+ vnfminstid='1',\r
+ package_id='todo',\r
+ version='',\r
+ vendor='',\r
+ netype='',\r
+ vnfd_model='',\r
+ status='INSTANTIATED',\r
+ nf_desc='',\r
+ vnfdid='',\r
+ vnfSoftwareVersion='',\r
+ vnfConfigurableProperties='todo',\r
+ localizationLanguage='EN_US',\r
+ create_time=now_time())\r
+\r
+ VmInstModel.objects.create(vmid="1",\r
+ vimid="1",\r
+ resouceid="11",\r
+ insttype=0,\r
+ instid="1111",\r
+ vmname="test_01",\r
+ is_predefined=1,\r
+ operationalstate=1)\r
+ t1_apply_grant_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t3_action_vm_start_result = [0, json.JSONEncoder().encode(''), '202']\r
+ mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_start_result]\r
+ mock_call.return_value = None\r
+ req_data = {\r
+ "changeStateTo": "STARTED"\r
+ }\r
+ self.nf_inst_id = '1111'\r
+ self.job_id = JobUtil.create_job('NF', 'OPERATE', self.nf_inst_id)\r
+ JobUtil.add_job_status(self.job_id, 0, "OPERATE_VNF_READY")\r
+ OperateVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
+ vm = VmInstModel.objects.filter(vmid="1", vimid="1", resouceid="11")\r
+ self.assertEqual("ACTIVE", vm[0].operationalstate)\r
+ self.assert_job_result(self.job_id, 100, "Operate Vnf success.")\r
+\r
+ @mock.patch.object(restcall, 'call_req')\r
+ @mock.patch.object(api, 'call')\r
+ def test_operate_vnf_success_stop(self, mock_call, mock_call_req):\r
+ NfInstModel.objects.create(nfinstid='1111',\r
+ nf_name='2222',\r
+ vnfminstid='1',\r
+ package_id='todo',\r
+ version='',\r
+ vendor='',\r
+ netype='',\r
+ vnfd_model='',\r
+ status='INSTANTIATED',\r
+ nf_desc='',\r
+ vnfdid='',\r
+ vnfSoftwareVersion='',\r
+ vnfConfigurableProperties='todo',\r
+ localizationLanguage='EN_US',\r
+ create_time=now_time())\r
+\r
+ VmInstModel.objects.create(vmid="1",\r
+ vimid="1",\r
+ resouceid="11",\r
+ insttype=0,\r
+ instid="1111",\r
+ vmname="test_01",\r
+ is_predefined=1,\r
+ operationalstate=1)\r
+ t1_apply_grant_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t3_action_vm_stop_result = [0, json.JSONEncoder().encode(''), '202']\r
+ mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_stop_result]\r
+ mock_call.return_value = None\r
+ req_data = {\r
+ "changeStateTo": "STOPPED"\r
+ }\r
+ self.nf_inst_id = '1111'\r
+ self.job_id = JobUtil.create_job('NF', 'OPERATE', self.nf_inst_id)\r
+ JobUtil.add_job_status(self.job_id, 0, "OPERATE_VNF_READY")\r
+ OperateVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
+ vm = VmInstModel.objects.filter(vmid="1", vimid="1", resouceid="11")\r
+ self.assertEqual("INACTIVE", vm[0].operationalstate)\r
+ self.assert_job_result(self.job_id, 100, "Operate Vnf success.")\r
+\r
+ @mock.patch.object(restcall, 'call_req')\r
+ @mock.patch.object(api, 'call')\r
+ def test_operate_vnf_success_stop_graceful(self, mock_call, mock_call_req):\r
+ NfInstModel.objects.create(nfinstid='1111',\r
+ nf_name='2222',\r
+ vnfminstid='1',\r
+ package_id='todo',\r
+ version='',\r
+ vendor='',\r
+ netype='',\r
+ vnfd_model='',\r
+ status='INSTANTIATED',\r
+ nf_desc='',\r
+ vnfdid='',\r
+ vnfSoftwareVersion='',\r
+ vnfConfigurableProperties='todo',\r
+ localizationLanguage='EN_US',\r
+ create_time=now_time())\r
+\r
+ VmInstModel.objects.create(vmid="1",\r
+ vimid="1",\r
+ resouceid="11",\r
+ insttype=0,\r
+ instid="1111",\r
+ vmname="test_01",\r
+ is_predefined=1,\r
+ operationalstate=1)\r
+ t1_apply_grant_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t3_action_vm_stop_result = [0, json.JSONEncoder().encode(''), '202']\r
+ mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_stop_result]\r
+ mock_call.return_value = None\r
+ req_data = {\r
+ "changeStateTo": "STOPPED",\r
+ "stopType": "GRACEFUL",\r
+ "gracefulStopTimeout": 2\r
+ }\r
+ self.nf_inst_id = '1111'\r
+ self.job_id = JobUtil.create_job('NF', 'OPERATE', self.nf_inst_id)\r
+ JobUtil.add_job_status(self.job_id, 0, "OPERATE_VNF_READY")\r
+ OperateVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
+ vm = VmInstModel.objects.filter(vmid="1", vimid="1", resouceid="11")\r
+ self.assertEqual("INACTIVE", vm[0].operationalstate)\r
+ self.assert_job_result(self.job_id, 100, "Operate Vnf success.")\r
+\r
+ @mock.patch.object(restcall, 'call_req')\r
+ @mock.patch.object(api, 'call')\r
+ def test_operate_vnf_success_stop_forceful(self, mock_call, mock_call_req):\r
+ NfInstModel.objects.create(nfinstid='1111',\r
+ nf_name='2222',\r
+ vnfminstid='1',\r
+ package_id='todo',\r
+ version='',\r
+ vendor='',\r
+ netype='',\r
+ vnfd_model='',\r
+ status='INSTANTIATED',\r
+ nf_desc='',\r
+ vnfdid='',\r
+ vnfSoftwareVersion='',\r
+ vnfConfigurableProperties='todo',\r
+ localizationLanguage='EN_US',\r
+ create_time=now_time())\r
+\r
+ VmInstModel.objects.create(vmid="1",\r
+ vimid="1",\r
+ resouceid="11",\r
+ insttype=0,\r
+ instid="1111",\r
+ vmname="test_01",\r
+ is_predefined=1,\r
+ operationalstate=1)\r
+ t1_apply_grant_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
+ t3_action_vm_stop_result = [0, json.JSONEncoder().encode(''), '202']\r
+ mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_stop_result]\r
+ mock_call.return_value = None\r
+ req_data = {\r
+ "changeStateTo": "STOPPED",\r
+ "stopType": "FORCEFUL",\r
+ }\r
+ self.nf_inst_id = '1111'\r
+ self.job_id = JobUtil.create_job('NF', 'OPERATE', self.nf_inst_id)\r
+ JobUtil.add_job_status(self.job_id, 0, "OPERATE_VNF_READY")\r
+ OperateVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
+ vm = VmInstModel.objects.filter(vmid="1", vimid="1", resouceid="11")\r
+ self.assertEqual("INACTIVE", vm[0].operationalstate)\r
+ self.assert_job_result(self.job_id, 100, "Operate Vnf success.")\r