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