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
  15 import json
  16 import uuid
  17
  18 import mock
  19 from django.test import TestCase, Client
  20 from rest_framework import status
  21
  22 from lcm.nf.vnfs.const import vnfd_rawdata, 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
  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
  31
  32
  33 class TestNFInstantiate(TestCase):
  34     def setUp(self):
  35         self.client = Client()
  36
  37     def tearDown(self):
  38         pass
  39
  40     def assert_job_result(self, job_id, job_progress, job_detail):
  41         jobs = JobStatusModel.objects.filter(jobid=job_id,
  42                                              progress=job_progress,
  43                                              descp=job_detail)
  44         self.assertEqual(1, len(jobs))
  45
  46     @mock.patch.object(restcall, 'call_req')
  47     def test_create_vnf_identifier(self, mock_call_req):
  48         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode(
  49             {
  50                 'csars': [
  51                     {
  52                         'package_id': '222',
  53                         'csarId': '2222',
  54                         'vnfdId': '111'
  55                     }
  56                 ]
  57             }), '200']
  58         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
  59         r3_create_vnf_to_aai = [0, json.JSONEncoder().encode({}), '200']
  60         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog, r3_create_vnf_to_aai]
  61         data = {
  62             "vnfdId": "111",
  63             "vnfInstanceName": "vFW_01",
  64             "vnfInstanceDescription": "vFW in Nanjing TIC Edge"
  65         }
  66         response = self.client.post("/api/vnflcm/v1/vnf_instances", data=data, format='json')
  67         self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
  68         context = json.loads(response.content)
  69         self.assertTrue(NfInstModel.objects.filter(nfinstid=context['vnfInstanceId']).exists())
  70
  71     @mock.patch.object(InstVnf, 'run')
  72     def test_instantiate_vnf(self, mock_run):
  73         mock_run.re.return_value = None
  74         response = self.client.post("/api/vnflcm/v1/vnf_instances/12/instantiate", data={}, format='json')
  75         self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code)
  76
  77     def test_instantiate_vnf_when_inst_id_not_exist(self):
  78         self.nf_inst_id = str(uuid.uuid4())
  79         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
  80         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
  81         jobs = JobStatusModel.objects.filter(jobid=self.job_id,
  82                                              progress=0,
  83                                              descp="INST_VNF_READY")
  84         self.assertEqual(1, len(jobs))
  85         data = inst_req_data
  86         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
  87         self.assert_job_result(self.job_id, 255, "VNF nf_inst_id is not exist.")
  88
  89     @mock.patch.object(restcall, 'call_req')
  90     def test_instantiate_vnf_when_get_package_info_by_vnfdid_failed(self, mock_call_req):
  91         NfInstModel.objects.create(nfinstid='1111',
  92                                    nf_name='vFW_01',
  93                                    package_id='222',
  94                                    version='',
  95                                    vendor='',
  96                                    netype='',
  97                                    vnfd_model='',
  98                                    status='NOT_INSTANTIATED',
  99                                    nf_desc='vFW in Nanjing TIC Edge',
 100                                    vnfdid='111',
 101                                    create_time=now_time())
 102         r1_get_csarid_by_vnfdid = [1, json.JSONEncoder().encode(
 103             {
 104                 'csars': [
 105                     {
 106                         'package_id': '222',
 107                         'csarId': '2222',
 108                         'vnfdId': '111'
 109                     }
 110                 ]
 111             }), '200']
 112         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid]
 113         self.nf_inst_id = '1111'
 114         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 115         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 116         data = inst_req_data
 117         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 118         self.assert_job_result(self.job_id, 255, "Failed to query package_info of vnfdid(111) from nslcm.")
 119
 120     @mock.patch.object(restcall, 'call_req')
 121     def test_instantiate_vnf_when_get_rawdata_by_csarid_failed(self, mock_call_req):
 122         NfInstModel.objects.create(nfinstid='1111',
 123                                    nf_name='vFW_01',
 124                                    package_id='222',
 125                                    version='',
 126                                    vendor='',
 127                                    netype='',
 128                                    vnfd_model='',
 129                                    status='NOT_INSTANTIATED',
 130                                    nf_desc='vFW in Nanjing TIC Edge',
 131                                    vnfdid='111',
 132                                    create_time=now_time())
 133         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode(
 134             {
 135                 'csars': [
 136                     {
 137                         'package_id': '222',
 138                         'csarId': '2222',
 139                         'vnfdId': '111'
 140                     }
 141                 ]
 142             }), '200']
 143         r2_get_rawdata_from_catalog = [1, json.JSONEncoder().encode(vnfd_rawdata), '200']
 144         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog]
 145         self.nf_inst_id = '1111'
 146         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 147         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 148         data = inst_req_data
 149         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 150         self.assert_job_result(self.job_id, 255, "Failed to query rawdata of CSAR(2222) from catalog.")
 151
 152     @mock.patch.object(restcall, 'call_req')
 153     def test_instantiate_vnf_when_applay_grant_failed(self, mock_call_req):
 154         NfInstModel.objects.create(nfinstid='1111',
 155                                    nf_name='vFW_01',
 156                                    package_id='222',
 157                                    version='',
 158                                    vendor='',
 159                                    netype='',
 160                                    vnfd_model='',
 161                                    status='NOT_INSTANTIATED',
 162                                    nf_desc='vFW in Nanjing TIC Edge',
 163                                    vnfdid='111',
 164                                    create_time=now_time())
 165         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode(
 166             {
 167                 'csars': [
 168                     {
 169                         'package_id': '222',
 170                         'csarId': '2222',
 171                         'vnfdId': '111'
 172                     }
 173                 ]
 174             }), '200']
 175         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 176         r3_apply_grant_result = [1, json.JSONEncoder().encode(
 177             {
 178                 "vim": {
 179                     "vimid": 'vimid_1',
 180                     "accessinfo": {
 181                         "tenant": 'tenantname_1'
 182                     }
 183                 }
 184             }), '200']
 185         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog, r3_apply_grant_result]
 186         self.nf_inst_id = '1111'
 187         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 188         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 189         data = inst_req_data
 190         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 191         self.assert_job_result(self.job_id, 255, "Nf instancing apply grant exception")
 192
 193     @mock.patch.object(restcall, 'call_req')
 194     @mock.patch.object(api, 'call')
 195     def test_instantiate_vnf_when_unexpected_exception(self, mock_call, mock_call_req):
 196         NfInstModel.objects.create(nfinstid='1111',
 197                                    nf_name='vFW_01',
 198                                    package_id='222',
 199                                    version='',
 200                                    vendor='',
 201                                    netype='',
 202                                    vnfd_model='',
 203                                    status='NOT_INSTANTIATED',
 204                                    nf_desc='vFW in Nanjing TIC Edge',
 205                                    vnfdid='111',
 206                                    create_time=now_time())
 207         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode(
 208             {
 209                 'csars': [
 210                     {
 211                         'package_id': '222',
 212                         'csarId': '2222',
 213                         'vnfdId': '111'
 214                     }
 215                 ]
 216             }), '200']
 217         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 218         r3_apply_grant_result = [0, json.JSONEncoder().encode(
 219             {
 220                 "vim": {
 221                     "vimid": 'vimid_1',
 222                     "accessinfo": {
 223                         "tenant": 'tenantname_1'
 224                     }
 225                 }
 226             }), '200']
 227         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog, r3_apply_grant_result]
 228         mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume]
 229         self.nf_inst_id = '1111'
 230         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 231         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 232         data = inst_req_data
 233         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 234         self.assert_job_result(self.job_id, 255, "unexpected exception")
 235
 236     @mock.patch.object(restcall, 'call_req')
 237     @mock.patch.object(api, 'call')
 238     def test_instantiate_vnf_success(self, mock_call, mock_call_req):
 239         NfInstModel.objects.create(nfinstid='1111',
 240                                    nf_name='vFW_01',
 241                                    package_id='222',
 242                                    version='',
 243                                    vendor='',
 244                                    netype='',
 245                                    vnfd_model='',
 246                                    status='NOT_INSTANTIATED',
 247                                    nf_desc='vFW in Nanjing TIC Edge',
 248                                    vnfdid='111',
 249                                    create_time=now_time())
 250         r1_get_csarid_by_vnfdid = [0, json.JSONEncoder().encode(
 251             {
 252                 'csars': [
 253                     {
 254                         'package_id': '222',
 255                         'csarId': '2222',
 256                         'vnfdId': '111'
 257                     }
 258                 ]
 259             }), '200']
 260         r2_get_rawdata_from_catalog = [0, json.JSONEncoder().encode(vnfd_rawdata), '200']
 261         r3_apply_grant_result = [0, json.JSONEncoder().encode(
 262             {
 263                 "vim": {
 264                     "vimid": 'vimid_1',
 265                     "accessinfo": {
 266                         "tenant": 'tenantname_1'
 267                     }
 268                 }
 269             }), '200']
 270         r4_lcm_notify_result = [0, json.JSONEncoder().encode(''), '200']
 271         mock_call_req.side_effect = [r1_get_csarid_by_vnfdid, r2_get_rawdata_from_catalog,
 272                                      r3_apply_grant_result, r4_lcm_notify_result]
 273         mock_call.side_effect = [c1_data_get_tenant_id, c2_data_create_volume, c3_data_get_volume,
 274                                  c4_data_create_network, c5_data_create_subnet, c6_data_create_port,
 275                                  c7_data_create_flavor, c8_data_list_image, c9_data_create_vm, c10_data_get_vm]
 276         self.nf_inst_id = '1111'
 277         self.job_id = JobUtil.create_job('NF', 'CREATE', self.nf_inst_id)
 278         JobUtil.add_job_status(self.job_id, 0, "INST_VNF_READY")
 279         data = inst_req_data
 280         InstVnf(data, nf_inst_id=self.nf_inst_id, job_id=self.job_id).run()
 281         self.assert_job_result(self.job_id, 100, "Instantiate Vnf success.")