lcm/lcm/nf/vnfs/tests/test_vnf_create.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 json
  15 import uuid
  16
  17 import mock
  18 from django.test import TestCase, Client
  19 from rest_framework import status
  20
  21 from lcm.nf.vnfs.const import vnfd_rawdata, c1_data_get_tenant_id, c4_data_create_network, c2_data_create_volume, \
  22     c5_data_create_subnet, c3_data_get_volume, c6_data_create_port, c7_data_create_flavor, c8_data_list_image, \
  23     c9_data_create_vm, c10_data_get_vm, inst_req_data
  24 from lcm.nf.vnfs.vnf_create.inst_vnf import InstVnf
  25 from lcm.pub.database.models import NfInstModel, JobStatusModel
  26 from lcm.pub.utils import restcall
  27 from lcm.pub.utils.jobutil import JobUtil
  28 from lcm.pub.utils.timeutil import now_time
  29 from lcm.pub.vimapi import api
  30
  31
  32 class TestNFInstantiate(TestCase):
  33     def setUp(self):
  34         self.client = Client()
  35
  36     def tearDown(self):
  37         pass
  38
  39     def assert_job_result(self, job_id, job_progress, job_detail):
  40         jobs = JobStatusModel.objects.filter(
  41             jobid=job_id,
  42             progress=job_progress,
  43             descp=job_detail)
  44         self.assertEqual(1, len(jobs))
  45
  46     def test_swagger_ok(self):
  47         response = self.client.get("/api/vnflcm/v1/swagger.json", format='json')
  48         self.assertEqual(response.status_code, status.HTTP_200_OK)
  49
  50     @mock.patch.object(restcall, 'call_req')
  51     @mock.patch.object(restcall, 'call_req_aai')
  52     def test_create_vnf_identifier(self, mock_call_req_aai, mock_call_req):
  53         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode({'csars':[{'package_id': '222',
  54                                                                   'csarId': '2222',
  55                                                                   'vnfdId': '111'}]}), '200']
  56         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
  57         r3_create_vnf_to_aai = [0, json.JSONEncoder().encode({}), '200']
  58         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog]
  59         mock_call_req_aai.side_effect = [r3_create_vnf_to_aai]
  60         data = {
  61             "vnfdId": "111",
  62             "vnfInstanceName": "vFW_01",
  63             "vnfInstanceDescription": "vFW in Nanjing TIC Edge"}
  64         response = self.client.post("/api/vnflcm/v1/vnf_instances", data=data, format='json')
  65         self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
  66         context = json.loads(response.content)
  67         self.assertTrue(NfInstModel.objects.filter(nfinstid=context['vnfInstanceId']).exists())
  68
  69     @mock.patch.object(InstVnf, 'run')
  70     def test_instantiate_vnf(self, mock_run):
  71         mock_run.re.return_value = None
  72         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/instantiate", data={}, format='json')
  73         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)
  74
  75     def test_instantiate_vnf_when_inst_id_not_exist(self):
  76         self.nf_inst_id = str(uuid.uuid4())
  77         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
  78         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
  79         jobs = JobStatusModel.objects.filter(
  80             jobid=self.job_id,
  81             progress=0,
  82             descp="INST_VNF_READY")
  83         self.assertEqual(1, len(jobs))
  84         data = inst_req_data
  85         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
  86         self.assert_job_result(self.job_id, 255, "VNF nf_inst_id is not exist.")
  87
  88     @mock.patch.object(restcall, 'call_req')
  89     def test_instantiate_vnf_when_get_package_info_by_vnfdid_failed(self, mock_call_req):
  90         NfInstModel.objects.create(nfinstid='1111', nf_name='vFW_01', package_id='222',
  91                                    version='', vendor='', netype='', vnfd_model='', status='NOT_INSTANTIATED',
  92                                    nf_desc='vFW in Nanjing TIC Edge', vnfdid='111', create_time=now_time())
  93         r1_get_csarid_by_vnfdid = [1, json.JSONEncoder().encode({'csars':[{'package_id': '222',
  94                                                                   'csarId': '2222',
  95                                                                   'vnfdId': '111'}]}), '200']
  96         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid]
  97         self.nf_inst_id = '1111'
  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         data = inst_req_data
 101         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 102         self.assert_job_result(self.job_id, 255, "Failed to query package_info of vnfdid(111) from nslcm.")
 103
 104     @mock.patch.object(restcall, 'call_req')
 105     def test_instantiate_vnf_when_get_rawdata_by_csarid_failed(self, mock_call_req):
 106         NfInstModel.objects.create(nfinstid='1111', nf_name='vFW_01', package_id='222',
 107                                    version='', vendor='', netype='', vnfd_model='', status='NOT_INSTANTIATED',
 108                                    nf_desc='vFW in Nanjing TIC Edge', vnfdid='111', create_time=now_time())
 109         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode({'csars':[{'package_id': '222',
 110                                                                   'csarId': '2222',
 111                                                                   'vnfdId': '111'}]}), '200']
 112         r2_get_rawdata_from_catalog = [1, json.JSONEncoder().encode(vnfd_rawdata), '200']
 113         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog]
 114         self.nf_inst_id = '1111'
 115         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 116         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 117         data = inst_req_data
 118         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 119         self.assert_job_result(self.job_id, 255, "Failed to query rawdata of CSAR(2222) from catalog.")
 120
 121     @mock.patch.object(restcall, 'call_req')
 122     def test_instantiate_vnf_when_applay_grant_failed(self, mock_call_req):
 123         NfInstModel.objects.create(nfinstid='1111', nf_name='vFW_01', package_id='222',
 124                                    version='', vendor='', netype='', vnfd_model='', status='NOT_INSTANTIATED',
 125                                    nf_desc='vFW in Nanjing TIC Edge', vnfdid='111', create_time=now_time())
 126         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode({'csars':[{'package_id': '222',
 127                                                                   'csarId': '2222',
 128                                                                   'vnfdId': '111'}]}), '200']
 129         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 130         r3_apply_grant_result = [1, json.JSONEncoder().encode(
 131             {"vim": {"vimid": 'vimid_1', "accessinfo": {"tenant": 'tenantname_1'}}}), '200']
 132         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog, r3_apply_grant_result]
 133         self.nf_inst_id = '1111'
 134         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 135         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 136         data = inst_req_data
 137         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 138         self.assert_job_result(self.job_id, 255, "Nf instancing apply grant exception")
 139
 140     @mock.patch.object(restcall, 'call_req')
 141     @mock.patch.object(api, 'call')
 142     def test_instantiate_vnf_when_unexpected_exception(self, mock_call, mock_call_req):
 143         NfInstModel.objects.create(nfinstid='1111', nf_name='vFW_01', package_id='222',
 144                                    version='', vendor='', netype='', vnfd_model='', status='NOT_INSTANTIATED',
 145                                    nf_desc='vFW in Nanjing TIC Edge', vnfdid='111', create_time=now_time())
 146         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode({'csars':[{'package_id': '222',
 147                                                                   'csarId': '2222',
 148                                                                   'vnfdId': '111'}]}), '200']
 149         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 150         r3_apply_grant_result = [0, json.JSONEncoder().encode(
 151             {"vim": {"vimid": 'vimid_1', "accessinfo": {"tenant": 'tenantname_1'}}}), '200']
 152         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog, r3_apply_grant_result]
 153         mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume]
 154         self.nf_inst_id = '1111'
 155         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 156         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 157         data = inst_req_data
 158         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 159         self.assert_job_result(self.job_id, 255, "unexpected exception")
 160
 161     @mock.patch.object(restcall, 'call_req')
 162     @mock.patch.object(api, 'call')
 163     def test_instantiate_vnf_success(self, mock_call, mock_call_req):
 164         NfInstModel.objects.create(nfinstid='1111', nf_name='vFW_01', package_id='222',
 165                                    version='', vendor='', netype='', vnfd_model='', status='NOT_INSTANTIATED',
 166                                    nf_desc='vFW in Nanjing TIC Edge', vnfdid='111', create_time=now_time())
 167         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode({'csars':[{'package_id': '222',
 168                                                                   'csarId': '2222',
 169                                                                   'vnfdId': '111'}]}), '200']
 170         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 171         r3_apply_grant_result = [0, json.JSONEncoder().encode(
 172             {"vim": {"vimid": 'vimid_1', "accessinfo": {"tenant": 'tenantname_1'}}}), '200']
 173         r4_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']
 174         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog,
 175                                      r3_apply_grant_result, r4_lcm_notify_result]
 176         mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume,
 177                                  c4_data_create_network, c5_data_create_subnet, c6_data_create_port,
 178                                  c7_data_create_flavor, c8_data_list_image, c9_data_create_vm, c10_data_get_vm]
 179         self.nf_inst_id = '1111'
 180         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 181         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 182         data = inst_req_data
 183         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 184         self.assert_job_result(self.job_id, 100, "Instantiate Vnf success.")