lcm/lcm/nf/tests/test_heal_vnf.py

   1 # Copyright (C) 2018 Verizon. All Rights Reserved.\r
   2 #\r
   3 # Licensed under the Apache License, Version 2.0 (the "License");\r
   4 # you may not use this file except in compliance with the License.\r
   5 # You may obtain a copy of the License at\r
   6 #\r
   7 #       http://www.apache.org/licenses/LICENSE-2.0\r
   8 #\r
   9 # Unless required by applicable law or agreed to in writing, software\r
  10 # distributed under the License is distributed on an "AS IS" BASIS,\r
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  12 # See the License for the specific language governing permissions and\r
  13 # limitations under the License.\r
  14 \r
  15 import json\r
  16 \r
  17 import mock\r
  18 from django.test import TestCase\r
  19 from rest_framework import status\r
  20 from rest_framework.test import APIClient\r
  21 \r
  22 from lcm.nf.biz.heal_vnf import HealVnf\r
  23 from lcm.nf.const import c1_data_get_tenant_id_1, c9_data_create_vm, c10_data_get_vm, vnfd_model_dict\r
  24 from lcm.pub.database.models import NfInstModel, JobStatusModel, VmInstModel\r
  25 from lcm.pub.utils import restcall\r
  26 from lcm.pub.utils.jobutil import JobUtil\r
  27 from lcm.pub.utils.timeutil import now_time\r
  28 from lcm.pub.vimapi import api\r
  29 \r
  30 \r
  31 class TestNFInstantiate(TestCase):\r
  32     def setUp(self):\r
  33         self.client = APIClient()\r
  34         self.grant_result = {\r
  35             "vimid": 'vimid_1',\r
  36             "tenant": 'tenantname_1',\r
  37         }\r
  38         self.getvmResult = {\r
  39             "status": "ACTIVE",\r
  40             "id": "11",\r
  41         }\r
  42 \r
  43     def tearDown(self):\r
  44         pass\r
  45 \r
  46     def assert_job_result(self, job_id, job_progress, job_detail):\r
  47         jobs = JobStatusModel.objects.filter(jobid=job_id,\r
  48                                              progress=job_progress,\r
  49                                              descp=job_detail)\r
  50         self.assertEqual(1, len(jobs))\r
  51 \r
  52     def test_heal_vnf_not_found(self):\r
  53         req_data = {}\r
  54         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/heal", data=req_data, format='json')\r
  55         self.failUnlessEqual(status.HTTP_404_NOT_FOUND, response.status_code)\r
  56 \r
  57     def test_heal_vnf_conflict(self):\r
  58         req_data = {}\r
  59         NfInstModel(nfinstid='12', nf_name='VNF1', status='NOT_INSTANTIATED').save()\r
  60         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/heal", data=req_data, format='json')\r
  61         self.failUnlessEqual(status.HTTP_409_CONFLICT, response.status_code)\r
  62         NfInstModel(nfinstid='12', nf_name='VNF1', status='NOT_INSTANTIATED').delete()\r
  63 \r
  64     @mock.patch.object(HealVnf, 'run')\r
  65     def test_heal_vnf_success(self, mock_run):\r
  66         req_data = {}\r
  67         NfInstModel(nfinstid='12', nf_name='VNF1', status='INSTANTIATED').save()\r
  68         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/heal", data=req_data, format='json')\r
  69         mock_run.re.return_value = None\r
  70         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)\r
  71         NfInstModel(nfinstid='12', nf_name='VNF1', status='INSTANTIATED').delete()\r
  72 \r
  73     @mock.patch.object(restcall, 'call_req')\r
  74     @mock.patch.object(api, 'call')\r
  75     def test_heal_vnf_success_reboot(self, mock_call, mock_call_req):\r
  76         vim_cache = {}\r
  77         res_cache = {}\r
  78 \r
  79         NfInstModel.objects.create(nfinstid='1111',\r
  80                                    nf_name='2222',\r
  81                                    vnfminstid='1',\r
  82                                    package_id='todo',\r
  83                                    version='',\r
  84                                    vendor='',\r
  85                                    netype='',\r
  86                                    vnfd_model=json.dumps(vnfd_model_dict),\r
  87                                    status='INSTANTIATED',\r
  88                                    nf_desc='',\r
  89                                    vnfdid='',\r
  90                                    vnfSoftwareVersion='',\r
  91                                    vnfConfigurableProperties='todo',\r
  92                                    localizationLanguage='EN_US',\r
  93                                    create_time=now_time(),\r
  94                                    resInfo=json.dumps(res_cache),\r
  95                                    vimInfo=json.dumps(vim_cache))\r
  96 \r
  97         VmInstModel.objects.create(vmid="1",\r
  98                                    vimid="vimid_1",\r
  99                                    resourceid="11",\r
 100                                    insttype=0,\r
 101                                    instid="1111",\r
 102                                    vmname="vduinstname",\r
 103                                    is_predefined=1,\r
 104                                    tenant="tenantname_1",\r
 105                                    operationalstate=1)\r
 106         t1_apply_grant_result = [0, json.JSONEncoder().encode(self.grant_result), '200']\r
 107         t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
 108         t3_action_get_vm = [0, json.JSONEncoder().encode(self.getvmResult), '202']\r
 109         t4_action_vm_start_reboot = [0, json.JSONEncoder().encode(''), '202']\r
 110         mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_get_vm, t4_action_vm_start_reboot]\r
 111         mock_call.side_effect = [self.getvmResult, None]\r
 112         req_data = {\r
 113             "cause": "Error",\r
 114             "additionalParams": {\r
 115                 "action": "vmReset",\r
 116                 "affectedvm": {\r
 117                     "vmid": "1",\r
 118                     "vduid": "vdu1Id",\r
 119                     "vmname": "vduinstname"\r
 120                 }\r
 121             }\r
 122         }\r
 123         self.nf_inst_id = '1111'\r
 124         self.job_id = JobUtil.create_job('NF', 'HEAL', self.nf_inst_id)\r
 125         JobUtil.add_job_status(self.job_id, 0, "HEAL_VNF_READY")\r
 126         HealVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
 127         self.assert_job_result(self.job_id, 100, "Heal Vnf success.")\r
 128 \r
 129     @mock.patch.object(restcall, 'call_req')\r
 130     @mock.patch.object(api, 'call')\r
 131     def test_heal_vnf_success_start(self, mock_call, mock_call_req):\r
 132         vim_cache = {}\r
 133         res_cache = {"volume": {}, "flavor": {}, "port": {}}\r
 134         res_cache["volume"]["volume_storage1"] = "vol1"\r
 135         res_cache["flavor"]["vdu1Id"] = "flavor1"\r
 136         res_cache["port"]["cpId1"] = "port1"\r
 137 \r
 138         NfInstModel.objects.create(nfinstid='1111',\r
 139                                    nf_name='2222',\r
 140                                    vnfminstid='1',\r
 141                                    package_id='todo',\r
 142                                    version='',\r
 143                                    vendor='',\r
 144                                    netype='',\r
 145                                    vnfd_model=json.dumps(vnfd_model_dict),\r
 146                                    status='INSTANTIATED',\r
 147                                    nf_desc='',\r
 148                                    vnfdid='',\r
 149                                    vnfSoftwareVersion='',\r
 150                                    vnfConfigurableProperties='todo',\r
 151                                    localizationLanguage='EN_US',\r
 152                                    resInfo=json.dumps(res_cache),\r
 153                                    vimInfo=json.dumps(vim_cache),\r
 154                                    create_time=now_time())\r
 155 \r
 156         t1_apply_grant_result = [0, json.JSONEncoder().encode(self.grant_result), '200']\r
 157         t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
 158         t3_action_vm_start_create = [0, json.JSONEncoder().encode(''), '202']\r
 159         mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_start_create]\r
 160         mock_call.side_effect = [c1_data_get_tenant_id_1, c9_data_create_vm, c10_data_get_vm]\r
 161         req_data = {\r
 162             "cause": "Error",\r
 163             "additionalParams": {\r
 164                 "action": "vmCreate",\r
 165                 "affectedvm": {\r
 166                     "vmid": "1",\r
 167                     "vduid": "vdu1Id",\r
 168                     "vmname": "vduinstname"\r
 169                 }\r
 170             }\r
 171         }\r
 172         self.nf_inst_id = '1111'\r
 173         self.job_id = JobUtil.create_job('NF', 'HEAL', self.nf_inst_id)\r
 174         JobUtil.add_job_status(self.job_id, 0, "HEAL_VNF_READY")\r
 175         HealVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
 176         self.assert_job_result(self.job_id, 100, "Heal Vnf success.")\r