1 # Copyright 2017 ZTE Corporation.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
19 from django.test import TestCase, Client
20 from rest_framework import status
22 from lcm.nf.vnfs.const import c1_data_get_tenant_id, c4_data_create_network, c2_data_create_volume, \
23 c5_data_create_subnet, c3_data_get_volume, c6_data_create_port, c7_data_create_flavor, c8_data_list_image, \
24 c9_data_create_vm, c10_data_get_vm, inst_req_data, vnfpackage_info
25 from lcm.nf.vnfs.vnf_create.inst_vnf import InstVnf
26 from lcm.pub.database.models import NfInstModel, JobStatusModel
27 from lcm.pub.utils import restcall
28 from lcm.pub.utils.jobutil import JobUtil
29 from lcm.pub.utils.timeutil import now_time
30 from lcm.pub.vimapi import api
33 class TestNFInstantiate(TestCase):
35 self.client = Client()
40 def assert_job_result(self, job_id, job_progress, job_detail):
41 jobs = JobStatusModel.objects.filter(jobid=job_id,
42 progress=job_progress,
44 self.assertEqual(1, len(jobs))
46 @mock.patch.object(restcall, 'call_req')
47 def test_create_vnf_identifier(self, mock_call_req):
48 r2_get_vnfpackage_from_catalog = [0, json.JSONEncoder().encode(vnfpackage_info), '200']
49 mock_call_req.side_effect = [r2_get_vnfpackage_from_catalog]
52 "vnfInstanceName": "vFW_01",
53 "vnfInstanceDescription": "vFW in Nanjing TIC Edge"
55 response = self.client.post("/api/vnflcm/v1/vnf_instances", data=data, format='json')
56 self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
57 context = json.loads(response.content)
58 self.assertTrue(NfInstModel.objects.filter(nfinstid=context['vnfInstanceId']).exists())
60 @mock.patch.object(InstVnf, 'run')
61 def test_instantiate_vnf(self, mock_run):
62 mock_run.re.return_value = None
63 response = self.client.post("/api/vnflcm/v1/vnf_instances/12/instantiate", data={}, format='json')
64 self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)
66 def test_instantiate_vnf_when_inst_id_not_exist(self):
67 self.nf_inst_id = str(uuid.uuid4())
68 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
69 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
70 jobs = JobStatusModel.objects.filter(jobid=self.job_id,
72 descp="INST_VNF_READY")
73 self.assertEqual(1, len(jobs))
75 InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
76 self.assert_job_result(self.job_id, 255, "VNF nf_inst_id is not exist.")
78 @mock.patch.object(restcall, 'call_req')
79 def test_instantiate_vnf_when_get_rawdata_by_csarid_failed(self, mock_call_req):
80 NfInstModel.objects.create(nfinstid='1111',
87 status='NOT_INSTANTIATED',
88 nf_desc='vFW in Nanjing TIC Edge',
90 create_time=now_time())
91 r1_get_vnfpackage_by_vnfdid = [1, json.JSONEncoder().encode(vnfpackage_info), '200']
92 mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid]
93 self.nf_inst_id = '1111'
94 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
95 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
97 InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
98 self.assert_job_result(self.job_id, 255, "Failed to query vnf CSAR(111) from catalog.")
100 @mock.patch.object(restcall, 'call_req')
101 def test_instantiate_vnf_when_applay_grant_failed(self, mock_call_req):
102 NfInstModel.objects.create(nfinstid='1111',
109 status='NOT_INSTANTIATED',
110 nf_desc='vFW in Nanjing TIC Edge',
112 create_time=now_time())
113 r1_get_vnfpackage_by_vnfdid = [0, json.JSONEncoder().encode(vnfpackage_info), '200']
114 r3_apply_grant_result = [1, json.JSONEncoder().encode(
119 "tenant": 'tenantname_1'
123 mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r3_apply_grant_result]
124 self.nf_inst_id = '1111'
125 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
126 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
128 InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
129 self.assert_job_result(self.job_id, 255, "Nf instancing apply grant exception")
131 @mock.patch.object(restcall, 'call_req')
132 @mock.patch.object(api, 'call')
133 def test_instantiate_vnf_when_unexpected_exception(self, mock_call, mock_call_req):
134 NfInstModel.objects.create(nfinstid='1111',
141 status='NOT_INSTANTIATED',
142 nf_desc='vFW in Nanjing TIC Edge',
144 create_time=now_time())
145 r1_get_vnfpackage_by_vnfdid = [0, json.JSONEncoder().encode(vnfpackage_info), '200']
146 r3_apply_grant_result = [0, json.JSONEncoder().encode(
151 "tenant": 'tenantname_1'
155 mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r3_apply_grant_result]
156 mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume]
157 self.nf_inst_id = '1111'
158 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
159 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
161 InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
162 self.assert_job_result(self.job_id, 255, "unexpected exception")
164 @mock.patch.object(restcall, 'call_req')
165 @mock.patch.object(api, 'call')
166 def test_instantiate_vnf_success(self, mock_call, mock_call_req):
167 NfInstModel.objects.create(nfinstid='1111',
174 status='NOT_INSTANTIATED',
175 nf_desc='vFW in Nanjing TIC Edge',
177 create_time=now_time())
178 r1_get_vnfpackage_by_vnfdid = [0, json.JSONEncoder().encode(vnfpackage_info), '200']
179 r3_apply_grant_result = [0, json.JSONEncoder().encode(
184 "tenant": 'tenantname_1'
188 r4_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']
189 mock_call_req.side_effect = [r1_get_vnfpackage_by_vnfdid, r3_apply_grant_result, r4_lcm_notify_result]
190 mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume,
191 c4_data_create_network, c5_data_create_subnet, c6_data_create_port,
192 c7_data_create_flavor, c8_data_list_image, c9_data_create_vm, c10_data_get_vm]
193 self.nf_inst_id = '1111'
194 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
195 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
197 InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
198 self.assert_job_result(self.job_id, 100, "Instantiate Vnf success.")