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     @mock.patch.object(HealVnf, 'run')\r
  58     def test_heal_vnf_success(self, mock_run):\r
  59         req_data = {}\r
  60         NfInstModel(nfinstid='12',\r
  61                     nf_name='VNF1',\r
  62                     nf_desc="VNF DESC",\r
  63                     vnfdid="1",\r
  64                     netype="XGW",\r
  65                     vendor="ZTE",\r
  66                     vnfSoftwareVersion="V1",\r
  67                     version="V1",\r
  68                     package_id="2",\r
  69                     status='INSTANTIATED').save()\r
  70         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/heal", data=req_data, format='json')\r
  71         mock_run.re.return_value = None\r
  72         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)\r
  73         NfInstModel(nfinstid='12').delete()\r
  74 \r
  75     @mock.patch.object(restcall, 'call_req')\r
  76     @mock.patch.object(api, 'call')\r
  77     def test_heal_vnf_success_reboot(self, mock_call, mock_call_req):\r
  78         vim_cache = {}\r
  79         res_cache = {}\r
  80 \r
  81         NfInstModel.objects.create(nfinstid='1111',\r
  82                                    nf_name='2222',\r
  83                                    vnfminstid='1',\r
  84                                    package_id='todo',\r
  85                                    version='',\r
  86                                    vendor='',\r
  87                                    netype='',\r
  88                                    vnfd_model=json.dumps(vnfd_model_dict),\r
  89                                    status='INSTANTIATED',\r
  90                                    nf_desc='',\r
  91                                    vnfdid='',\r
  92                                    vnfSoftwareVersion='',\r
  93                                    vnfConfigurableProperties='todo',\r
  94                                    localizationLanguage='EN_US',\r
  95                                    create_time=now_time(),\r
  96                                    resInfo=json.dumps(res_cache),\r
  97                                    vimInfo=json.dumps(vim_cache))\r
  98 \r
  99         VmInstModel.objects.create(vmid="1",\r
 100                                    vimid="vimid_1",\r
 101                                    resourceid="1",\r
 102                                    insttype=0,\r
 103                                    instid="1111",\r
 104                                    vmname="vduinstname",\r
 105                                    is_predefined=1,\r
 106                                    tenant="tenantname_1",\r
 107                                    operationalstate=1)\r
 108         t1_apply_grant_result = [0, json.JSONEncoder().encode(self.grant_result), '200']\r
 109         t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
 110         t3_action_get_vm = [0, json.JSONEncoder().encode(self.getvmResult), '202']\r
 111         t4_action_vm_start_reboot = [0, json.JSONEncoder().encode(''), '202']\r
 112         mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_get_vm, t4_action_vm_start_reboot]\r
 113         mock_call.side_effect = [self.getvmResult, None]\r
 114         req_data = {\r
 115             "cause": "Error",\r
 116             "additionalParams": {\r
 117                 "action": "vmReset",\r
 118                 "affectedvm": {\r
 119                     "vmid": "1",\r
 120                     "vduid": "vdu1Id",\r
 121                     "vmname": "vduinstname"\r
 122                 }\r
 123             }\r
 124         }\r
 125         self.nf_inst_id = '1111'\r
 126         self.job_id = JobUtil.create_job('NF', 'HEAL', self.nf_inst_id)\r
 127         JobUtil.add_job_status(self.job_id, 0, "HEAL_VNF_READY")\r
 128         HealVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
 129         self.assert_job_result(self.job_id, 100, "Heal Vnf success.")\r
 130 \r
 131     @mock.patch.object(restcall, 'call_req')\r
 132     @mock.patch.object(api, 'call')\r
 133     def test_heal_vnf_success_start(self, mock_call, mock_call_req):\r
 134         vim_cache = {}\r
 135         res_cache = {"volume": {}, "flavor": {}, "port": {}}\r
 136         res_cache["volume"]["volume_storage1"] = "vol1"\r
 137         res_cache["flavor"]["vdu1Id"] = "flavor1"\r
 138         res_cache["port"]["cpId1"] = "port1"\r
 139 \r
 140         NfInstModel.objects.create(nfinstid='1111',\r
 141                                    nf_name='2222',\r
 142                                    vnfminstid='1',\r
 143                                    package_id='todo',\r
 144                                    version='',\r
 145                                    vendor='',\r
 146                                    netype='',\r
 147                                    vnfd_model=json.dumps(vnfd_model_dict),\r
 148                                    status='INSTANTIATED',\r
 149                                    nf_desc='',\r
 150                                    vnfdid='',\r
 151                                    vnfSoftwareVersion='',\r
 152                                    vnfConfigurableProperties='todo',\r
 153                                    localizationLanguage='EN_US',\r
 154                                    resInfo=json.dumps(res_cache),\r
 155                                    vimInfo=json.dumps(vim_cache),\r
 156                                    create_time=now_time())\r
 157 \r
 158         t1_apply_grant_result = [0, json.JSONEncoder().encode(self.grant_result), '200']\r
 159         t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']\r
 160         t3_action_vm_start_create = [0, json.JSONEncoder().encode(''), '202']\r
 161         mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result, t3_action_vm_start_create]\r
 162         mock_call.side_effect = [c1_data_get_tenant_id_1, c9_data_create_vm, c10_data_get_vm]\r
 163         req_data = {\r
 164             "cause": "Error",\r
 165             "additionalParams": {\r
 166                 "action": "vmCreate",\r
 167                 "affectedvm": {\r
 168                     "vmid": "1",\r
 169                     "vduid": "vdu1Id",\r
 170                     "vmname": "vduinstname"\r
 171                 }\r
 172             }\r
 173         }\r
 174         self.nf_inst_id = '1111'\r
 175         self.job_id = JobUtil.create_job('NF', 'HEAL', self.nf_inst_id)\r
 176         JobUtil.add_job_status(self.job_id, 0, "HEAL_VNF_READY")\r
 177         HealVnf(req_data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()\r
 178         self.assert_job_result(self.job_id, 100, "Heal Vnf success.")\r