1 # Copyright (C) 2018 Verizon. All Rights Reserved.
\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
7 # http://www.apache.org/licenses/LICENSE-2.0
\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
18 from django.test import TestCase
\r
19 from rest_framework import status
\r
20 from rest_framework.test import APIClient
\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
31 class TestNFInstantiate(TestCase):
\r
33 self.client = APIClient()
\r
34 self.grant_result = {
\r
36 "tenant": 'tenantname_1',
\r
38 self.getvmResult = {
\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
50 self.assertEqual(1, len(jobs))
\r
52 def test_heal_vnf_not_found(self):
\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
57 @mock.patch.object(HealVnf, 'run')
\r
58 def test_heal_vnf_success(self, mock_run):
\r
60 NfInstModel(nfinstid='12',
\r
66 vnfSoftwareVersion="V1",
\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
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
81 NfInstModel.objects.create(nfinstid='1111',
\r
88 vnfd_model=json.dumps(vnfd_model_dict),
\r
89 status='INSTANTIATED',
\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
99 VmInstModel.objects.create(vmid="1",
\r
104 vmname="vduinstname",
\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
116 "additionalParams": {
\r
117 "action": "vmReset",
\r
121 "vmname": "vduinstname"
\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
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
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
140 NfInstModel.objects.create(nfinstid='1111',
\r
147 vnfd_model=json.dumps(vnfd_model_dict),
\r
148 status='INSTANTIATED',
\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
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
165 "additionalParams": {
\r
166 "action": "vmCreate",
\r
170 "vmname": "vduinstname"
\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