Update create vl logic
[vfc/nfvo/lcm.git] / 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     '''
61     @mock.patch.object(restcall, "call_req")
62     @mock.patch.object(vimadaptor.VimAdaptor, "create_network")
63     def test_create_vl(self, mock_create_network, mock_req_by_rest):
64         network_id = str(uuid.uuid4())
65         subnetwork_id = str(uuid.uuid4())
66         mock_create_network.return_value = [0,
67                                             {"status": "ACTIVE", "id": network_id, "name": "net1",
68                                              "provider:segmentation_id": 204, "provider:network_type": "vlan",
69                                              "res_type": 1,
70                                              "subnet_list": [
71                                                  {"id": subnetwork_id, "name": "subnet1", "res_type": 1}]}]
72         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
73
74         self.create_vl(self.vl_id_1)
75         self.create_vl(self.vl_id_2)
76         vl_from_vl_1 = VLInstModel.objects.filter(vldid=self.vl_id_1, ownerid=self.ns_inst_id)
77         self.assertEqual(network_id, vl_from_vl_1[0].relatednetworkid)
78         self.assertEqual(subnetwork_id, vl_from_vl_1[0].relatedsubnetworkid)
79         self.assertEqual(self.tenant, vl_from_vl_1[0].tenant)
80         vl_from_vl_2 = VLInstModel.objects.filter(vldid=self.vl_id_2, ownerid=self.ns_inst_id)
81         self.assertEqual(VNFFGInstModel.objects.filter(vnffgdid=self.vnffg_id, nsinstid=self.ns_inst_id)[0].vllist,
82                          vl_from_vl_2[0].vlinstanceid + "," + vl_from_vl_1[0].vlinstanceid)
83     '''
84
85     def create_vl(self, vl_id):
86         req_data = {
87             "nsInstanceId": self.ns_inst_id,
88             "context": json.JSONEncoder().encode(self.context),
89             "vlIndex": vl_id}
90         response = self.client.post("/api/nslcm/v1/ns/vls", data=req_data)
91         self.assertEqual(status.HTTP_201_CREATED, response.status_code)
92         self.assertEqual(0, response.data["result"], response.data)
93
94     '''
95     @mock.patch.object(restcall, "call_req")
96     @mock.patch.object(vimadaptor.VimAdaptor, "create_network")
97     @mock.patch.object(uuid, "uuid4")
98     def test_create_network_fail_when_send_to_vim(self, mock_uuid4, mock_create_network, mock_req_by_rest):
99         req_data = {
100             "nsInstanceId": self.ns_inst_id,
101             "context": json.JSONEncoder().encode(self.context),
102             "vlIndex": self.vl_id_1}
103         mock_uuid4.return_value = '999'
104         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
105         mock_create_network.return_value = [1, (1)]
106         response = self.client.post("/api/nslcm/v1/ns/vls", data=req_data)
107         retinfo = {"detail": "vl instantiation failed, detail message: Send post vl request to vim failed."}
108         self.assertEqual(retinfo["detail"], response.data["detail"])
109     '''
110
111
112 class TestVlDetailViews(TestCase):
113     def setUp(self):
114         self.client = Client()
115         self.vl_inst_id = str(uuid.uuid4())
116         self.vl_name = str(uuid.uuid4())
117         self.ns_inst_id = str(uuid.uuid4())
118         VLInstModel(vlinstanceid=self.vl_inst_id, vldid="", vlinstancename=self.vl_name, ownertype=1,
119                     ownerid=self.ns_inst_id, relatednetworkid="network1", relatedsubnetworkid="subnet1,subnet2",
120                     vimid="",
121                     tenant="").save()
122         VNFFGInstModel(vnffgdid="", vnffginstid="", nsinstid=self.ns_inst_id,
123                        vllist="test1," + self.vl_inst_id + ",test2,test3", endpointnumber=0, cplist="", vnflist="",
124                        fplist="", status="").save()
125
126     def tearDown(self):
127         VLInstModel.objects.all().delete()
128         VNFFGInstModel.objects.all().delete()
129
130     @mock.patch.object(restcall, "call_req")
131     @mock.patch.object(vimadaptor.VimAdaptor, "delete_network")
132     @mock.patch.object(vimadaptor.VimAdaptor, "delete_subnet")
133     def test_delete_vl(self, mock_delete_subnet, mock_delete_network, mock_req_by_rest):
134         mock_req_by_rest.return_value = [0, json.JSONEncoder().encode(vim_info), '200']
135         response = self.client.delete("/api/nslcm/v1/ns/vls/%s" % self.vl_inst_id)
136         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
137         expect_resp_data = {"result": 0, "detail": "delete vl success"}
138         self.assertEqual(expect_resp_data, response.data)
139
140         for vnffg_info in VNFFGInstModel.objects.filter(nsinstid=self.ns_inst_id):
141             self.assertEqual(vnffg_info.vllist, "test1,test2,test3")
142         if VLInstModel.objects.filter(vlinstanceid=self.vl_inst_id):
143             self.fail()
144
145         response = self.client.delete("/api/nslcm/v1/ns/vls/%s" % "notExist")
146         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
147         expect_resp_data = {"result": 0, "detail": "vl is not exist or has been already deleted"}
148         self.assertEqual(expect_resp_data, response.data)
149
150     def test_query_vl(self):
151         response = self.client.get("/api/nslcm/v1/ns/vls/%s" % self.vl_inst_id)
152         self.assertEqual(status.HTTP_200_OK, response.status_code)
153         expect_resp_data = {'vlId': self.vl_inst_id, 'vlName': self.vl_name, 'vlStatus': "active"}
154         self.assertEqual(expect_resp_data, response.data)
155
156         response = self.client.get("/api/nslcm/v1/ns/vls/%s" % "notExist")
157         self.assertEqual(status.HTTP_404_NOT_FOUND, response.status_code)