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 def test_heal_vnf_conflict(self):
\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
64 @mock.patch.object(HealVnf, 'run')
\r
65 def test_heal_vnf_success(self, mock_run):
\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
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
79 NfInstModel.objects.create(nfinstid='1111',
\r
86 vnfd_model=json.dumps(vnfd_model_dict),
\r
87 status='INSTANTIATED',
\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
97 VmInstModel.objects.create(vmid="1",
\r
102 vmname="vduinstname",
\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
114 "additionalParams": {
\r
115 "action": "vmReset",
\r
119 "vmname": "vduinstname"
\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
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
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
138 NfInstModel.objects.create(nfinstid='1111',
\r
145 vnfd_model=json.dumps(vnfd_model_dict),
\r
146 status='INSTANTIATED',
\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
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
163 "additionalParams": {
\r
164 "action": "vmCreate",
\r
168 "vmname": "vduinstname"
\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