lcm/ns/tests/test_ns_instant.py

   1 # Copyright 2016 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 mock
  17 from mock import MagicMock
  18 from django.test import TestCase
  19 from rest_framework import status
  20 from rest_framework.test import APIClient
  21 from lcm.ns.biz.ns_instant import BuildInWorkflowThread
  22 from lcm.ns.biz.ns_instant import InstantNSService
  23 from lcm.pub.database.models import NSInstModel
  24 from lcm.pub.utils import restcall
  25 from lcm.pub.config import config
  26
  27 nsd_model = json.dumps({
  28     "model": json.dumps({
  29         "vnfs": [{
  30             "vnf_id": "vnf1",
  31             "properties": {
  32                 "id": "vnfd1",
  33                 "nf_type": "xgw"
  34             },
  35             "dependencies": [{
  36                 "vl_id": "5"
  37             }]
  38         }],
  39         "vls": [{
  40             "vl_id": "5",
  41             "properties": {}
  42         }]
  43     })
  44 })
  45
  46
  47 class TestNsInstant(TestCase):
  48
  49     def setUp(self):
  50         self.client = APIClient()
  51         NSInstModel.objects.filter().delete()
  52         self.url = "/api/nslcm/v1/ns/2/instantiate"
  53         self.req_data = {
  54             "additionalParamForNs": {
  55                 "sdnControllerId": "2"
  56             },
  57             "nsFlavourId": 12345,
  58             "localizationLanguage": [{
  59                 "vnfProfileId": "vnfd1",
  60                 "locationConstraints": {
  61                     "countryCode": "countryCode",
  62                     # "vimId": "3",
  63                     "civicAddressElement": [
  64                         {"caType": "type1",
  65                          "caValue": 1
  66                          }
  67                     ]
  68                 }
  69             }]
  70         }
  71         self.nsd_model = nsd_model
  72         self.updated_nsd_model = {
  73             "vnfs": [{
  74                 "dependencies": [{
  75                     "vl_id": "5"
  76                 }],
  77                 "vnf_id": "vnf1",
  78                 "properties": {
  79                     "nf_type": "xgw",
  80                     "id": "vnfd1"
  81                 }
  82             }],
  83             "vls": [{
  84                 "vl_id": "5",
  85                 "properties": {
  86                     "location_info": {
  87                         "vimid": "3"
  88                     }
  89                 }
  90             }]
  91         }
  92         self.vnfms = json.dumps({
  93             "esr-vnfm": [{
  94                 "vnfm-id": "4"
  95             }]
  96         })
  97         self.vnfm = json.dumps({
  98             "type": "xgw",
  99             "vim-id": "3",
 100             "vnfm-id": "4",
 101             "certificate-url": "http://127.0.0.0/ztevnfm/v1/auth",
 102             "esr-system-info-list": {
 103                 "esr-system-info": [{
 104                     "type": "xgw",
 105                     "vendor": "zte",
 106                     "version": "1.0",
 107                     "service-url": "http://127.0.0.0/ztevnfm/v1",
 108                     "user-name": "admin",
 109                     "password": "admin123"
 110                 }]
 111             }
 112         })
 113         NSInstModel(id="2", nspackage_id="7", nsd_id="2", status="active").save()
 114
 115     def tearDown(self):
 116         pass
 117
 118     @mock.patch.object(restcall, 'call_req')
 119     @mock.patch('lcm.pub.msapi.sdc_run_catalog.parse_nsd', MagicMock(return_value=nsd_model))
 120     @mock.patch.object(BuildInWorkflowThread, 'run')
 121     def test_ns_instantiate_when_succeed_to_enter_workflow(self, mock_run, mock_call_req):
 122         config.WORKFLOW_OPTION = "buildin"
 123         mock_call_req.side_effect = [
 124             [0, self.nsd_model, '200'],
 125             [0, self.vnfms, '200'],
 126             [0, self.vnfm, '200']
 127         ]
 128         resp = self.client.post(self.url, data=self.req_data, format='json')
 129         self.assertEqual(status.HTTP_200_OK, resp.status_code)
 130         self.assertIn("jobId", resp.data)
 131
 132     @mock.patch.object(InstantNSService, 'do_biz')
 133     def test_ns_instantiate_normal(self, mock_do_biz):
 134         mock_do_biz.return_value = dict(data={'jobId': "1"}, status=status.HTTP_200_OK)
 135         resp = self.client.post(self.url, data=self.req_data, format='json')
 136         self.failUnlessEqual(status.HTTP_200_OK, resp.status_code)
 137         self.assertEqual({'jobId': "1"}, resp.data)
 138
 139     @mock.patch.object(restcall, 'call_req')
 140     def test_ns_instantiate_when_fail_to_parse_nsd(self, mock_call_req):
 141         mock_call_req.return_value = [1, "Failed to parse nsd", '500']
 142         resp = self.client.post(self.url, data=self.req_data, format='json')
 143         self.assertEqual(resp.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
 144         self.assertIn("error", resp.data)
 145
 146     nsd = json.dumps({"vnffgs": [], "inputs": {}, "pnfs": [{"pnf_id": "du", "networks": [], "description": "", "properties": {"descriptor_id": "zte_ran_du_0001", "descriptor_invariant_id": "1111", "provider": "ZTE", "version": "1.0", "function_description": "RAN DU Function", "name": "ZTE RAN DU"}}], "ns_exposed": {"external_cps": [], "forward_cps": []}, "graph": {"cucp": [], "du": [], "vl_flat_net": ["cucp", "cuup"], "vl_ext_net": ["cucp", "cuup"], "cuup": []}, "basepath": "c:\\users\\10030173\\appdata\\local\\temp\\tmpvg5vto", "vnfs": [{"networks": [{"key_name": "ran_ext_net", "vl_id": "vl_ext_net"}, {"key_name": "ran_flat_net", "vl_id": "vl_flat_net"}], "dependencies": [{"key_name": "ran_ext_net", "vl_id": "vl_ext_net"}, {"key_name": "ran_flat_net", "vl_id": "vl_flat_net"}], "vnf_id": "cucp", "description": "", "properties": {"descriptor_id": "zte_ran_cucp_0001", "flavour_description": "default", "software_version": "1.0.1", "flavour_id": "1", "descriptor_version": "1.0", "provider": "ZTE", "id": "zte_ran_cucp_0001", "vnfm_info": ["GVNFM-Driver"], "product_name": "ran"}}, {"networks": [{"key_name": "ran_ext_net", "vl_id": "vl_ext_net"}, {"key_name": "ran_flat_net", "vl_id": "vl_flat_net"}], "dependencies": [{"key_name": "ran_ext_net", "vl_id": "vl_ext_net"}, {"key_name": "ran_flat_net", "vl_id": "vl_flat_net"}], "vnf_id": "cuup", "description": "", "properties": {"descriptor_id": "zte_ran_cuup_0001", "flavour_description": "default", "software_version": "1.0.1", "flavour_id": "1", "descriptor_version": "1.0", "provider": "ZTE", "id": "zte_ran_cuup_0001", "vnfm_info": ["GVNFM-Driver"], "product_name": "ran"}}], "fps": [], "vls": [{"vl_id": "vl_ext_net", "description": "", "properties": {"connectivity_type": {"layer_protocol": "ipv4"}, "vl_profile": {"cidr": "10.0.0.0/24", "max_bit_rate_requirements": {"root": 10000000, "leaf": 10000000}, "networkName": "ran_ext_net", "min_bit_rate_requirements": {"root": 10000000, "leaf": 10000000}, "dhcpEnabled": False}, "version": "1.0.1"}}, {"vl_id": "vl_flat_net", "description": "", "properties": {"connectivity_type": {"layer_protocol": "ipv4"}, "vl_profile": {"cidr": "10.1.0.0/24", "max_bit_rate_requirements": {"root": 10000000, "leaf": 10000000}, "networkName": "ran_flat_net", "min_bit_rate_requirements": {"root": 10000000, "leaf": 10000000}, "dhcpEnabled": False}, "version": "1.0.1"}}], "nested_ns": [], "metadata": {"template_name": "RAN-NS", "template_version": "1.0", "template_author": "ZTE"}})
 147     vnfminfo = {"vnfmId": "1"}
 148
 149     @mock.patch('lcm.ns.biz.ns_instantiate_flow.post_deal')
 150     @mock.patch.object(restcall, 'call_req')
 151     @mock.patch('lcm.ns.biz.ns_instantiate_flow.update_job')
 152     @mock.patch('lcm.pub.msapi.sdc_run_catalog.parse_nsd', MagicMock(return_value=nsd))
 153     @mock.patch('lcm.pub.msapi.extsys.select_vnfm', MagicMock(return_value=vnfminfo))
 154     def test_ns_instantiate_with_pnf(self, mock_updata_job, mock_call_req, mock_post_deal):
 155         config.WORKFLOW_OPTION = "grapflow"
 156         NSInstModel(id="1", name="test_ns", nspackage_id="1", status="created").save()
 157         ret = [0, json.JSONEncoder().encode({'jobId': "1", "responseDescriptor": {"progress": 100}}), '200']
 158         mock_call_req.side_effect = [ret for i in range(1, 20)]
 159         data = {
 160             "additionalParamForNs": {
 161                 "sdnControllerId": "2"
 162             },
 163             "locationConstraints": [{
 164                 "vnfProfileId": "zte_ran_cucp_0001",
 165                 "locationConstraints": {"vimId": '{"cloud_owner": "VCPE", "cloud_regionid": "RegionOne"}'}
 166             },
 167                 {
 168                     "vnfProfileId": "zte_ran_cuup_0001",
 169                     "locationConstraints": {"vimId": '{"cloud_owner": "VCPE", "cloud_regionid": "RegionOne"}'}
 170             }
 171             ],
 172             "addpnfData": [{
 173                 "pnfId": 1,
 174                 "pnfName": "test_pnf",
 175                 "pnfdId": "zte_ran_du_0001",
 176                 "pnfProfileId": "du"
 177             }]
 178         }
 179         # response = self.client.post("/api/nslcm/v1/ns/1/instantiate", data=data, format='json')
 180         ack = InstantNSService(1, data).do_biz()
 181         self.assertEqual(ack['status'], status.HTTP_200_OK)