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             "locationConstraints": [{
  58                 "vnfProfileId": "vnfd1",
  59                 "locationConstraints": {
  60                     "vimId": "3"
  61                 }
  62             }]
  63         }
  64         self.nsd_model = nsd_model
  65         self.updated_nsd_model = {
  66             "ns_vnfs": [{
  67                 "dependencies": [{
  68                     "vl_id": "5"
  69                 }],
  70                 "vnf_id": "vnf1",
  71                 "properties": {
  72                     "nf_type": "xgw",
  73                     "id": "vnfd1"
  74                 }
  75             }],
  76             "ns_vls": [{
  77                 "vl_id": "5",
  78                 "properties": {
  79                     "location_info": {
  80                         "vimid": "3"
  81                     }
  82                 }
  83             }]
  84         }
  85         self.vnfms = json.dumps({
  86             "esr-vnfm": [{
  87                 "vnfm-id": "4"
  88             }]
  89         })
  90         self.vnfm = json.dumps({
  91             "type": "xgw",
  92             "vim-id": "3",
  93             "vnfm-id": "4",
  94             "certificate-url": "http://127.0.0.0/ztevnfm/v1/auth",
  95             "esr-system-info-list": {
  96                 "esr-system-info": [{
  97                     "type": "xgw",
  98                     "vendor": "zte",
  99                     "version": "1.0",
 100                     "service-url": "http://127.0.0.0/ztevnfm/v1",
 101                     "user-name": "admin",
 102                     "password": "admin123"
 103                 }]
 104             }
 105         })
 106         NSInstModel(id="2", nspackage_id="7", nsd_id="2", status="active").save()
 107
 108     def tearDown(self):
 109         pass
 110
 111     @mock.patch.object(restcall, 'call_req')
 112     @mock.patch('lcm.pub.msapi.sdc_run_catalog.parse_nsd', MagicMock(return_value=nsd_model))
 113     @mock.patch.object(BuildInWorkflowThread, 'run')
 114     def test_ns_instantiate_when_succeed_to_enter_workflow(self, mock_run, mock_call_req):
 115         config.WORKFLOW_OPTION = "buildin"
 116         mock_call_req.side_effect = [
 117             [0, self.nsd_model, '200'],
 118             [0, self.vnfms, '200'],
 119             [0, self.vnfm, '200']
 120         ]
 121         resp = self.client.post(self.url, data=self.req_data, format='json')
 122         self.assertEqual(status.HTTP_200_OK, resp.status_code)
 123         self.assertIn("jobId", resp.data)
 124
 125     @mock.patch.object(InstantNSService, 'do_biz')
 126     def test_ns_instantiate_normal(self, mock_do_biz):
 127         mock_do_biz.return_value = dict(data={'jobId': "1"}, status=status.HTTP_200_OK)
 128         resp = self.client.post(self.url, data=self.req_data, format='json')
 129         self.failUnlessEqual(status.HTTP_200_OK, resp.status_code)
 130         self.assertEqual({'jobId': "1"}, resp.data)
 131
 132     @mock.patch.object(restcall, 'call_req')
 133     def test_ns_instantiate_when_fail_to_parse_nsd(self, mock_call_req):
 134         mock_call_req.return_value = [1, "Failed to parse nsd", '500']
 135         resp = self.client.post(self.url, data=self.req_data, format='json')
 136         self.assertEqual(resp.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
 137         self.assertIn("error", resp.data)
 138
 139     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"}})
 140     vnfminfo = {"vnfmId": "1"}
 141
 142     @mock.patch('lcm.ns.biz.ns_instantiate_flow.post_deal')
 143     @mock.patch.object(restcall, 'call_req')
 144     @mock.patch('lcm.ns.biz.ns_instantiate_flow.update_job')
 145     @mock.patch('lcm.pub.msapi.sdc_run_catalog.parse_nsd', MagicMock(return_value=nsd))
 146     @mock.patch('lcm.pub.msapi.extsys.select_vnfm', MagicMock(return_value=vnfminfo))
 147     def test_ns_instantiate_with_pnf(self, mock_updata_job, mock_call_req, mock_post_deal):
 148         config.WORKFLOW_OPTION = "grapflow"
 149         NSInstModel(id="1", name="test_ns", nspackage_id="1", status="created").save()
 150         ret = [0, json.JSONEncoder().encode({'jobId': "1", "responseDescriptor": {"progress": 100}}), '200']
 151         mock_call_req.side_effect = [ret for i in range(1, 20)]
 152         data = {
 153             "additionalParamForNs": {
 154                 "sdnControllerId": "2"
 155             },
 156             "locationConstraints": [{
 157                 "vnfProfileId": "zte_ran_cucp_0001",
 158                 "locationConstraints": {"vimId": "3"}
 159             },
 160                 {
 161                     "vnfProfileId": "zte_ran_cuup_0001",
 162                     "locationConstraints": {"vimId": "3"}
 163             }
 164             ],
 165             "addpnfData": [{
 166                 "pnfId": 1,
 167                 "pnfName": "test_pnf",
 168                 "pnfdId": "zte_ran_du_0001",
 169                 "pnfProfileId": "du"
 170             }]
 171         }
 172         # response = self.client.post("/api/nslcm/v1/ns/1/instantiate", data=data, format='json')
 173         ack = InstantNSService(1, data).do_biz()
 174         self.assertEqual(ack['status'], status.HTTP_200_OK)