lcm/ns/tests/vls/tests.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 uuid
  17
  18 import mock
  19 from django.test import TestCase, Client
  20 from rest_framework import status
  21
  22 from lcm.ns.tests.vnfs.tests import vim_info
  23 from lcm.pub.database.models import VLInstModel, NSInstModel, VNFFGInstModel
  24 from lcm.pub.nfvi.vim import vimadaptor
  25 from lcm.pub.utils import restcall
  26
  27
  28 class TestVlViews(TestCase):
  29     def setUp(self):
  30         self.client = Client()
  31         self.ns_inst_id = str(uuid.uuid4())
  32         self.vnffg_id = str(uuid.uuid4())
  33         self.vl_id_1 = 1
  34         self.vl_id_2 = 1
  35         self.vim_id = str(uuid.uuid4())
  36         self.tenant = "tenantname"
  37         properties = {"network_type": "vlan", "name": "externalMNetworkName", "dhcp_enabled": False,
  38                       "location_info": {"host": True, "vimid": self.vim_id, "region": True, "tenant": self.tenant},
  39                       "end_ip": "190.168.100.100", "gateway_ip": "190.168.100.1", "start_ip": "190.168.100.2",
  40                       "cidr": "190.168.100.0/24", "mtu": 1500, "network_name": "sub_mnet", "ip_version": 4}
  41         self.context = {
  42             "vls": [{"vl_id": self.vl_id_1, "description": "", "properties": properties, "route_external": False},
  43                     {"vl_id": self.vl_id_2, "description": "", "properties": properties, "route_external": False}],
  44             "vnffgs": [{"vnffg_id": self.vnffg_id, "description": "",
  45                         "properties": {"vendor": "zte", "version": "1.1.2", "number_of_endpoints": 7,
  46                                        "dependent_virtual_link": [self.vl_id_2, self.vl_id_1],
  47                                        "connection_point": ["CP01", "CP02"],
  48                                        "constituent_vnfs": ["VNF1", "VNF2", "VNF3"],
  49                                        "constituent_pnfs": ["pnf1", "pnf2"]},
  50                         "members": ["forwarding_path1", "forwarding_path2"]}]}
  51         NSInstModel(id=self.ns_inst_id, name="ns_name").save()
  52         VNFFGInstModel(vnffgdid=self.vnffg_id, vnffginstid="", nsinstid=self.ns_inst_id, endpointnumber=0, vllist="",
  53                        cplist="", vnflist="", fplist="", status="").save()
  54
  55     def tearDown(self):
  56         VLInstModel.objects.all().delete()
  57         NSInstModel.objects.all().delete()
  58         VNFFGInstModel.objects.all().delete()
  59
  60     @mock.patch.object(restcall, "call_req")
  61     @mock.patch.object(vimadaptor.VimAdaptor, "create_network")
  62     def test_create_vl(self, mock_create_network, mock_req_by_rest):
  63         network_id = str(uuid.uuid4())
  64         subnetwork_id = str(uuid.uuid4())
  65         mock_create_network.return_value = [0,
  66                                             {"status": "ACTIVE", "id": network_id, "name": "net1",
  67                                              "provider:segmentation_id": 204, "provider:network_type": "vlan",
  68                                              "res_type": 1,
  69                                              "subnet_list": [
  70                                                  {"id": subnetwork_id, "name": "subnet1", "res_type": 1}]}]
  71         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
  72
  73         self.create_vl(self.vl_id_1)
  74         self.create_vl(self.vl_id_2)
  75         vl_from_vl_1 = VLInstModel.objects.filter(vldid=self.vl_id_1, ownerid=self.ns_inst_id)
  76         self.assertEqual(network_id, vl_from_vl_1[0].relatednetworkid)
  77         self.assertEqual(subnetwork_id, vl_from_vl_1[0].relatedsubnetworkid)
  78         self.assertEqual(self.tenant, vl_from_vl_1[0].tenant)
  79         vl_from_vl_2 = VLInstModel.objects.filter(vldid=self.vl_id_2, ownerid=self.ns_inst_id)
  80         self.assertEqual(VNFFGInstModel.objects.filter(vnffgdid=self.vnffg_id, nsinstid=self.ns_inst_id)[0].vllist,
  81                          vl_from_vl_2[0].vlinstanceid + "," + vl_from_vl_1[0].vlinstanceid)
  82
  83     def create_vl(self, vl_id):
  84         req_data = {
  85             "nsInstanceId": self.ns_inst_id,
  86             "context": json.JSONEncoder().encode(self.context),
  87             "vlIndex": vl_id}
  88         response = self.client.post("/api/nslcm/v1/ns/vls", data=req_data)
  89         self.assertEqual(status.HTTP_201_CREATED, response.status_code)
  90         self.assertEqual(0, response.data["result"], response.data)
  91
  92     @mock.patch.object(restcall, "call_req")
  93     @mock.patch.object(vimadaptor.VimAdaptor, "create_network")
  94     @mock.patch.object(uuid, "uuid4")
  95     def test_create_network_fail_when_send_to_vim(self, mock_uuid4, mock_create_network, mock_req_by_rest):
  96         req_data = {
  97             "nsInstanceId": self.ns_inst_id,
  98             "context": json.JSONEncoder().encode(self.context),
  99             "vlIndex": self.vl_id_1}
 100         mock_uuid4.return_value = '999'
 101         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
 102         mock_create_network.return_value = [1, (1)]
 103         response = self.client.post("/api/nslcm/v1/ns/vls", data=req_data)
 104         retinfo = {"detail": "vl instantiation failed, detail message: Send post vl request to vim failed."}
 105         self.assertEqual(retinfo["detail"], response.data["detail"])
 106
 107
 108 class TestVlDetailViews(TestCase):
 109     def setUp(self):
 110         self.client = Client()
 111         self.vl_inst_id = str(uuid.uuid4())
 112         self.vl_name = str(uuid.uuid4())
 113         self.ns_inst_id = str(uuid.uuid4())
 114         VLInstModel(vlinstanceid=self.vl_inst_id, vldid="", vlinstancename=self.vl_name, ownertype=1,
 115                     ownerid=self.ns_inst_id, relatednetworkid="network1", relatedsubnetworkid="subnet1,subnet2",
 116                     vimid="",
 117                     tenant="").save()
 118         VNFFGInstModel(vnffgdid="", vnffginstid="", nsinstid=self.ns_inst_id,
 119                        vllist="test1," + self.vl_inst_id + ",test2,test3", endpointnumber=0, cplist="", vnflist="",
 120                        fplist="", status="").save()
 121
 122     def tearDown(self):
 123         VLInstModel.objects.all().delete()
 124         VNFFGInstModel.objects.all().delete()
 125
 126     @mock.patch.object(restcall, "call_req")
 127     @mock.patch.object(vimadaptor.VimAdaptor, "delete_network")
 128     @mock.patch.object(vimadaptor.VimAdaptor, "delete_subnet")
 129     def test_delete_vl(self, mock_delete_subnet, mock_delete_network, mock_req_by_rest):
 130         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
 131         response = self.client.delete("/api/nslcm/v1/ns/vls/%s" % self.vl_inst_id)
 132         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
 133         expect_resp_data = {"result": 0, "detail": "delete vl success"}
 134         self.assertEqual(expect_resp_data, response.data)
 135
 136         for vnffg_info in VNFFGInstModel.objects.filter(nsinstid=self.ns_inst_id):
 137             self.assertEqual(vnffg_info.vllist, "test1,test2,test3")
 138         if VLInstModel.objects.filter(vlinstanceid=self.vl_inst_id):
 139             self.fail()
 140
 141         response = self.client.delete("/api/nslcm/v1/ns/vls/%s" % "notExist")
 142         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
 143         expect_resp_data = {"result": 0, "detail": "vl is not exist or has been already deleted"}
 144         self.assertEqual(expect_resp_data, response.data)
 145
 146     def test_query_vl(self):
 147         response = self.client.get("/api/nslcm/v1/ns/vls/%s" % self.vl_inst_id)
 148         self.assertEqual(status.HTTP_200_OK, response.status_code)
 149         expect_resp_data = {'vlId': self.vl_inst_id, 'vlName': self.vl_name, 'vlStatus': "active"}
 150         self.assertEqual(expect_resp_data, response.data)
 151
 152         response = self.client.get("/api/nslcm/v1/ns/vls/%s" % "notExist")
 153         self.assertEqual(status.HTTP_404_NOT_FOUND, response.status_code)