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