1 # Copyright 2019 ZTE Corporation.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
18 from django.test import TestCase
19 from rest_framework import status
20 from rest_framework.test import APIClient
21 from lcm.pub.database.models import NSInstModel
22 from lcm.pub.utils import restcall
23 from lcm.ns.biz.ns_create import CreateNSService
24 from lcm.pub.exceptions import NSLCMException
25 from lcm.ns.tests import SOL_CREATE_NS_DICT, SOL_REST_HEADER_DICT, NS_PACKAGE_INFO_DICT, NS_INFO_AAI_DICT
28 class TestNsInstanceApi(TestCase):
31 self.apiClient = APIClient()
33 self.ns_instances_url = '/api/nslcm/v1/ns_instances'
34 self.nsd_id = "c9f0a95e-dea0-4698-96e5-5a79bc5a233d"
35 self.ns_package_id = "c9f0a95e-dea0-4698-96e5-5a79bc5a233d"
38 NSInstModel.objects.all().delete()
40 @mock.patch.object(restcall, 'call_req')
41 def test_create_ns(self, mock_call_req):
42 r1_query_nspackage_from_catalog = [0, json.JSONEncoder().encode(NS_PACKAGE_INFO_DICT), '201']
43 r2_create_ns_to_aai = [0, json.JSONEncoder().encode({}), '201']
44 mock_call_req.side_effect = [r1_query_nspackage_from_catalog, r2_create_ns_to_aai]
45 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
46 self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
47 return response.data['id']
49 @mock.patch.object(restcall, 'call_req')
50 def test_create_ns_cpe(self, mock_call_req):
51 r1_query_nspackage_from_catalog = [0, json.JSONEncoder().encode(NS_PACKAGE_INFO_DICT), '201']
52 r2_create_ns_to_aai = [0, json.JSONEncoder().encode({}), '201']
53 mock_call_req.side_effect = [r1_query_nspackage_from_catalog, r2_create_ns_to_aai]
54 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
55 self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
57 @mock.patch.object(restcall, 'call_req')
58 def test_create_ns_when_ns_name_exist(self, mock_call_req):
59 NSInstModel.objects.all().delete()
60 NSInstModel(id="1", name="ns").save()
61 nspackage_info = json.JSONEncoder().encode(NS_PACKAGE_INFO_DICT)
62 mock_call_req.return_value = [0, nspackage_info, '200']
63 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
64 self.assertEqual(response.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
66 @mock.patch.object(CreateNSService, "do_biz")
67 def test_create_ns_empty_data(self, mock_do_biz):
68 mock_do_biz.side_effect = Exception("Exception in CreateNS.")
69 response = self.apiClient.post(self.ns_instances_url, data={}, format=self.format, **SOL_REST_HEADER_DICT)
70 self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
72 @mock.patch.object(CreateNSService, "do_biz")
73 def test_create_ns_no_header(self, mock_do_biz):
74 mock_do_biz.side_effect = Exception("Exception in CreateNS.")
75 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format)
76 self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
78 @mock.patch.object(CreateNSService, "do_biz")
79 def test_create_ns_non_existing_nsd(self, mock_do_biz):
80 mock_do_biz.side_effect = NSLCMException("nsd not exists.")
81 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
82 self.assertEqual(response.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
84 @mock.patch.object(restcall, 'call_req')
85 def test_create_ns_when_fail_to_get_nsd(self, mock_call_req):
86 mock_call_req.return_value = [1, "Failed to get nsd.", '500']
87 response = self.apiClient.post(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
88 self.assertEqual(response.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
90 def test_ns_instances_method_not_allowed(self):
91 response = self.apiClient.delete(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
92 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
93 response = self.apiClient.put(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
94 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
95 response = self.apiClient.patch(self.ns_instances_url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
96 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
98 def test_invidual_ns_instance_method_not_allowed(self):
99 url = self.ns_instances_url + '/1'
100 response = self.apiClient.post(url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
101 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
102 response = self.apiClient.put(url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
103 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
104 response = self.apiClient.patch(url, data=SOL_CREATE_NS_DICT, format=self.format, **SOL_REST_HEADER_DICT)
105 self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
107 def test_query_ns(self):
108 NSInstModel.objects.all().delete()
109 self.test_create_ns()
110 response = self.apiClient.get(self.ns_instances_url)
111 self.failUnlessEqual(status.HTTP_200_OK, response.status_code, response.data)
112 self.assertIsNotNone(response.data)
113 self.assertEqual(1, len(response.data))
114 self.assertEquals(self.nsd_id, response.data[0]['nsdId'])
115 self.assertEquals('ns', response.data[0]['nsInstanceName'])
116 self.assertEquals('NOT_INSTANTIATED', response.data[0]['nsState'])
118 def test_query_one_ns(self):
119 NSInstModel.objects.all().delete()
120 id = self.test_create_ns()
121 url = self.ns_instances_url + '/' + id
122 response = self.apiClient.get(url)
123 self.failUnlessEqual(status.HTTP_200_OK, response.status_code, response.data)
124 self.assertIsNotNone(response.data)
125 self.assertEquals(self.nsd_id, response.data['nsdId'])
126 self.assertEquals('ns', response.data['nsInstanceName'])
127 self.assertEquals('NOT_INSTANTIATED', response.data['nsState'])
129 @mock.patch.object(restcall, 'call_req')
130 def test_delete_ns(self, mock_call_req):
131 NSInstModel(id="1", nspackage_id="7", nsd_id="2").save()
132 r1_query_ns_to_aai = [0, json.JSONEncoder().encode(NS_INFO_AAI_DICT), '200']
133 r2_delete_ns_to_aai = [0, json.JSONEncoder().encode({}), '200']
134 mock_call_req.side_effect = [r1_query_ns_to_aai, r2_delete_ns_to_aai]
135 url = self.ns_instances_url + '/1'
136 response = self.apiClient.delete(url)
137 self.failUnlessEqual(status.HTTP_204_NO_CONTENT, response.status_code)