lcm/lcm/nf/vnfs/tests/test_vnf_cancel.py

   1 # # Copyright 2017 ZTE Corporation.
   2 # #
   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
   6 # #
   7 # #         http://www.apache.org/licenses/LICENSE-2.0
   8 # #
   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.
  14 import uuid
  15
  16 import mock
  17 from django.test import TestCase, Client
  18 from rest_framework import status
  19
  20 from lcm.nf.vnfs.vnf_cancel.term_vnf import TermVnf
  21 from lcm.pub.database.models import NfInstModel, JobStatusModel, VmInstModel, NetworkInstModel, SubNetworkInstModel, \
  22     PortInstModel, NfvoRegInfoModel
  23 from lcm.pub.utils.jobutil import JobUtil
  24 from lcm.pub.utils.timeutil import now_time
  25
  26
  27 class TestNFTerminate(TestCase):
  28     def setUp(self):
  29         self.client = Client()
  30         VmInstModel.objects.create(vmid="1", vimid="1", resouceid="11", insttype=0, instid="1111", vmname="test_01",
  31                                    is_predefined=1, operationalstate=1)
  32         VmInstModel.objects.create(vmid="2", vimid="2", resouceid="22", insttype=0, instid="1111",
  33                                    is_predefined=1, vmname="test_02", operationalstate=1)
  34         NetworkInstModel.objects.create(networkid='1', vimid='1', resouceid='1', name='pnet_network',
  35                                         is_predefined=1, tenant='admin', insttype=0, instid='1111')
  36         SubNetworkInstModel.objects.create(subnetworkid='1', vimid='1', resouceid='1', networkid='1',
  37                                            is_predefined=1, name='sub_pnet', tenant='admin', insttype=0, instid='1111')
  38         PortInstModel.objects.create(portid='1', networkid='1', subnetworkid='1', vimid='1', resouceid='1',
  39                                      is_predefined=1, name='aaa_pnet_cp', tenant='admin', insttype=0, instid='1111')
  40
  41     def tearDown(self):
  42         VmInstModel.objects.all().delete()
  43         NetworkInstModel.objects.all().delete()
  44         SubNetworkInstModel.objects.all().delete()
  45         PortInstModel.objects.all().delete()
  46
  47     def assert_job_result(self, job_id, job_progress, job_detail):
  48         jobs = JobStatusModel.objects.filter(
  49             jobid=job_id,
  50             progress=job_progress,
  51             descp=job_detail)
  52         self.assertEqual(1, len(jobs))
  53
  54     def test_delete_vnf_identifier(self):
  55         NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
  56                                    netype='', vnfd_model='', status='VNF_INSTANTIATED', nf_desc='', vnfdid='',
  57                                    vnfSoftwareVersion='', vnfConfigurableProperties='todo',
  58                                    localizationLanguage='EN_US', create_time=now_time())
  59         response = self.client.delete("/openoapi/vnflcm/v1/vnf_instances/1111")
  60         self.failUnlessEqual(status.HTTP_204_NO_CONTENT, response.status_code)
  61         self.assertEqual(None, response.data)
  62
  63     def test_delete_vnf_identifier_when_vnf_not_exist(self):
  64         response = self.client.delete("/openoapi/vnflcm/v1/vnf_instances/1111")
  65         self.failUnlessEqual(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code)
  66         self.assertEqual("VnfInst(1111) does not exist", response.data["error"])
  67
  68     def test_delete_vnf_identifier_when_status_check_failed(self):
  69         NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
  70                                    netype='', vnfd_model='', status='NOT_INSTANTIATED', nf_desc='', vnfdid='',
  71                                    vnfSoftwareVersion='', vnfConfigurableProperties='todo',
  72                                    localizationLanguage='EN_US', create_time=now_time())
  73         response = self.client.delete("/openoapi/vnflcm/v1/vnf_instances/1111")
  74         self.failUnlessEqual(status.HTTP_500_INTERNAL_SERVER_ERROR, response.status_code)
  75         self.assertEqual("Don't allow to delete vnf(status:[NOT_INSTANTIATED])", response.data["error"])
  76
  77     @mock.patch.object(TermVnf, 'run')
  78     def test_terminate_vnf(self, mock_run):
  79         mock_run.re.return_value = None
  80         response = self.client.post("/openoapi/vnflcm/v1/vnf_instances/12/terminate", data={}, format='json')
  81         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)
  82
  83     def test_terminate_vnf_when_inst_id_not_exist(self):
  84         data = {"terminationType": "GRACEFUL",
  85                 "gracefulTerminationTimeout": 120}
  86         self.nf_inst_id = str(uuid.uuid4())
  87         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
  88         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
  89         TermVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
  90         self.assert_job_result(self.job_id, 255, "VnfInst(%s) does not exist" % self.nf_inst_id)
  91
  92
  93
  94     def test_terminate_vnf_success(self):
  95         NfInstModel.objects.create(nfinstid='1111', nf_name='2222', package_id='todo', version='', vendor='',
  96                                    netype='', vnfd_model='', status='VNF_INSTANTIATED', nf_desc='', vnfdid='',
  97                                    vnfSoftwareVersion='', vnfConfigurableProperties='todo',
  98                                    localizationLanguage='EN_US', create_time=now_time())
  99         data = {"terminationType": "FORCEFUL",
 100                 "gracefulTerminationTimeout": 120}
 101         self.nf_inst_id = '1111'
 102         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 103         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 104         TermVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 105         self.assert_job_result(self.job_id, 100, "Terminate Vnf success.")