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.
18 from django.test import TestCase, Client
19 from rest_framework import status
21 from lcm.nf.vnfs.vnf_cancel.term_vnf import TermVnf
22 from lcm.pub.database.models import NfInstModel, JobStatusModel, VmInstModel, NetworkInstModel, SubNetworkInstModel, \
23 PortInstModel, FlavourInstModel, StorageInstModel, NfvoRegInfoModel
24 from lcm.pub.utils import restcall
25 from lcm.pub.utils.jobutil import JobUtil
26 from lcm.pub.utils.timeutil import now_time
27 from lcm.pub.vimapi import api
30 class TestNFTerminate(TestCase):
32 self.client = Client()
33 StorageInstModel.objects.create(storageid="1", vimid="1", resouceid="11", insttype=0, instid="1111",
35 NetworkInstModel.objects.create(networkid='1', vimid='1', resouceid='1', name='pnet_network',
36 is_predefined=1, tenant='admin', insttype=0, instid='1111')
37 SubNetworkInstModel.objects.create(subnetworkid='1', vimid='1', resouceid='1', networkid='1',
38 is_predefined=1, name='sub_pnet', tenant='admin', insttype=0, instid='1111')
39 PortInstModel.objects.create(portid='1', networkid='1', subnetworkid='1', vimid='1', resouceid='1',
40 is_predefined=1, name='aaa_pnet_cp', tenant='admin', insttype=0, instid='1111')
41 FlavourInstModel.objects.create(flavourid="1", vimid="1", resouceid="11", instid="1111", is_predefined=1)
42 VmInstModel.objects.create(vmid="1", vimid="1", resouceid="11", insttype=0, instid="1111", vmname="test_01",
43 is_predefined=1, operationalstate=1)
44 NfvoRegInfoModel.objects.create(nfvoid='1111', vnfminstid='11111', apiurl='1')
47 VmInstModel.objects.all().delete()
48 NetworkInstModel.objects.all().delete()
49 SubNetworkInstModel.objects.all().delete()
50 PortInstModel.objects.all().delete()
52 def assert_job_result(self, job_id, job_progress, job_detail):
53 jobs = JobStatusModel.objects.filter(
55 progress=job_progress,
57 self.assertEqual(1, len(jobs))
59 @mock.patch.object(restcall, 'call_req_aai')
60 def test_delete_vnf_identifier(self, mock_call_req_aai):
61 NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
62 netype='', vnfd_model='', status='NOT_INSTANTIATED', nf_desc='', vnfdid='',
63 vnfSoftwareVersion='', vnfConfigurableProperties='todo',
64 localizationLanguage='EN_US', create_time=now_time())
65 r1_create_vnf_to_aai = [0, json.JSONEncoder().encode({}), '200']
66 mock_call_req_aai.side_effect = [r1_create_vnf_to_aai]
67 response = self.client.delete("/api/vnflcm/v1/vnf_instances/1111")
68 self.failUnlessEqual(status.HTTP_204_NO_CONTENT, response.status_code)
69 self.assertEqual(None, response.data)
72 def test_delete_vnf_identifier_when_vnf_not_exist(self):
73 response = self.client.delete("/api/vnflcm/v1/vnf_instances/1111")
74 self.failUnlessEqual(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code)
75 self.assertEqual("VnfInst(1111) does not exist", response.data["error"])
77 def test_delete_vnf_identifier_when_status_check_failed(self):
78 NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
79 netype='', vnfd_model='', status='VNF_INSTANTIATED', nf_desc='', vnfdid='',
80 vnfSoftwareVersion='', vnfConfigurableProperties='todo',
81 localizationLanguage='EN_US', create_time=now_time())
82 response = self.client.delete("/api/vnflcm/v1/vnf_instances/1111")
83 self.failUnlessEqual(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code)
84 self.assertEqual("Don't allow to delete vnf(status:[VNF_INSTANTIATED])", response.data["error"])
87 @mock.patch.object(TermVnf, 'run')
88 def test_terminate_vnf(self, mock_run):
89 mock_run.re.return_value = None
90 response = self.client.post("/api/vnflcm/v1/vnf_instances/12/terminate", data={}, format='json')
91 self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)
94 def test_terminate_vnf_when_inst_id_not_exist(self):
95 data = {"terminationType": "GRACEFUL",
96 "gracefulTerminationTimeout": 120}
97 self.nf_inst_id = str(uuid.uuid4())
98 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
99 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
100 TermVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
101 self.assert_job_result(self.job_id, 255, "VnfInst(%s) does not exist" % self.nf_inst_id)
104 @mock.patch.object(restcall, 'call_req')
105 @mock.patch.object(api, 'call')
106 def test_terminate_vnf_success(self, mock_call, mock_call_req):
107 NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
108 netype='', vnfd_model='', status='VNF_INSTANTIATED', nf_desc='', vnfdid='',
109 vnfSoftwareVersion='', vnfConfigurableProperties='todo',
110 localizationLanguage='EN_US', create_time=now_time())
111 t1_apply_grant_result = [0, json.JSONEncoder().encode(
112 {"vim": {"vimid": 'vimid_1', "accessinfo": {"tenant": 'tenantname_1'}}}), '200']
113 t2_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']
114 mock_call_req.side_effect = [t1_apply_grant_result, t2_lcm_notify_result]
115 mock_call.return_value = None
116 data = {"terminationType": "FORCEFUL",
117 "gracefulTerminationTimeout": 120}
118 self.nf_inst_id = '1111'
119 self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
120 JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
121 TermVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
122 self.assert_job_result(self.job_id, 100, "Terminate Vnf success.")