1 # Copyright (C) 2019 Verizon. All Rights Reserved
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.
17 from rest_framework.test import APIClient
18 from django.test import TestCase
19 from genericparser.pub.database.models import VnfPkgSubscriptionModel
22 class TestNfPackageSubscription(TestCase):
24 self.client = APIClient()
25 VnfPkgSubscriptionModel.objects.filter().delete()
26 self.vnf_subscription_data = {
28 "notificationTypes": [
29 "VnfPackageOnboardingNotification"
31 "vnfProductsFromProviders": {
32 "vnfProvider": "string",
34 "vnfProductName": "string",
36 "vnfSoftwareVersion": "string",
44 "3fa85f64-5717-4562-b3fc-2c963f66afa6"
47 "3fa85f64-5717-4562-b3fc-2c963f66afa6"
56 "callbackUri": "http://www.vnf1.com/notification",
71 @mock.patch("requests.get")
72 @mock.patch.object(uuid, 'uuid4')
73 def test_create_vnf_subscription(self, mock_uuid4, mock_requests):
74 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
75 mock_requests.return_value.status_code = 204
76 mock_requests.get.status_code = 204
77 mock_uuid4.return_value = temp_uuid
78 response = self.client.post("/api/vnfpkgm/v1/subscriptions", data=self.vnf_subscription_data, format='json')
79 self.assertEqual(201, response.status_code)
80 self.assertEqual(self.vnf_subscription_data["callbackUri"], response.data["callbackUri"])
81 self.assertEqual(temp_uuid, response.data["id"])
83 @mock.patch("requests.get")
84 @mock.patch.object(uuid, 'uuid4')
85 def test_duplicate_subscriptions(self, mock_uuid4, mock_requests):
86 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
87 temp1_uuid = "00342b18-a5c7-11e8-998c-bf1755941f12"
88 mock_requests.return_value.status_code = 204
89 mock_requests.get.status_code = 204
90 mock_uuid4.side_effect = [temp_uuid, temp1_uuid]
91 response = self.client.post("/api/vnfpkgm/v1/subscriptions", data=self.vnf_subscription_data, format='json')
92 self.assertEqual(201, response.status_code)
93 self.assertEqual(self.vnf_subscription_data["callbackUri"], response.data["callbackUri"])
94 self.assertEqual(temp_uuid, response.data["id"])
95 temp_uuid = "00442b18-a5c7-11e8-998c-bf1755941f12"
96 mock_requests.return_value.status_code = 204
97 mock_requests.get.status_code = 204
98 mock_uuid4.return_value = temp_uuid
99 response = self.client.post("/api/vnfpkgm/v1/subscriptions", data=self.vnf_subscription_data, format='json')
100 self.assertEqual(303, response.status_code)
102 @mock.patch("requests.get")
103 @mock.patch.object(uuid, 'uuid4')
104 def test_get_subscriptions(self, mock_uuid4, mock_requests):
105 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
106 mock_requests.return_value.status_code = 204
107 mock_requests.get.status_code = 204
108 mock_uuid4.return_value = temp_uuid
109 self.client.post("/api/vnfpkgm/v1/subscriptions",
110 data=self.vnf_subscription_data, format='json')
111 response = self.client.get("/api/vnfpkgm/v1/subscriptions?usageState=IN_USE",
113 self.assertEqual(200, response.status_code)
114 self.assertEqual(1, len(response.data))
116 @mock.patch("requests.get")
117 @mock.patch.object(uuid, 'uuid4')
118 def test_get_subscriptions_with_invalid_params(self, mock_uuid4, mock_requests):
119 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
120 mock_requests.return_value.status_code = 204
121 mock_requests.get.status_code = 204
122 mock_uuid4.return_value = temp_uuid
123 self.client.post("/api/vnfpkgm/v1/subscriptions",
124 data=self.vnf_subscription_data, format='json')
125 response = self.client.get("/api/vnfpkgm/v1/subscriptions?dummy=dummy",
127 self.assertEqual(400, response.status_code)
129 @mock.patch("requests.get")
130 @mock.patch.object(uuid, 'uuid4')
131 def test_get_subscription_with_id(self, mock_uuid4, mock_requests):
132 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
133 mock_requests.return_value.status_code = 204
134 mock_requests.get.status_code = 204
135 mock_uuid4.return_value = temp_uuid
136 self.client.post("/api/vnfpkgm/v1/subscriptions",
137 data=self.vnf_subscription_data, format='json')
138 response = self.client.get("/api/vnfpkgm/v1/subscriptions/" + temp_uuid,
140 self.assertEqual(200, response.status_code)
141 self.assertEqual(temp_uuid, response.data["id"])
143 @mock.patch("requests.get")
144 @mock.patch.object(uuid, 'uuid4')
145 def test_get_subscription_with_id_not_exists(self, mock_uuid4, mock_requests):
146 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
147 dummy_uuid = str(uuid.uuid4())
148 mock_requests.return_value.status_code = 204
149 mock_requests.get.status_code = 204
150 mock_uuid4.return_value = temp_uuid
151 self.client.post("/api/vnfpkgm/v1/subscriptions",
152 data=self.vnf_subscription_data, format='json')
153 response = self.client.get("/api/vnfpkgm/v1/subscriptions/" + dummy_uuid,
155 self.assertEqual(404, response.status_code)
157 @mock.patch("requests.get")
158 @mock.patch.object(uuid, 'uuid4')
159 def test_delete_subscription_with_id(self, mock_uuid4, mock_requests):
160 temp_uuid = "99442b18-a5c7-11e8-998c-bf1755941f13"
161 dummy_uuid = str(uuid.uuid4())
162 mock_requests.return_value.status_code = 204
163 mock_requests.get.status_code = 204
164 mock_uuid4.return_value = temp_uuid
165 self.client.post("/api/vnfpkgm/v1/subscriptions",
166 data=self.vnf_subscription_data, format='json')
167 self.client.get("/api/vnfpkgm/v1/subscriptions/" + dummy_uuid,
169 response = self.client.delete("/api/vnfpkgm/v1/subscriptions/" + temp_uuid)
170 self.assertEqual(204, response.status_code)
172 @mock.patch("requests.get")
173 @mock.patch.object(uuid, 'uuid4')
174 def test_delete_subscription_with_id_not_exists(self, mock_uuid4, mock_requests):
175 dummy_uuid = str(uuid.uuid4())
176 response = self.client.delete("/api/vnfpkgm/v1/subscriptions/" + dummy_uuid)
177 self.assertEqual(404, response.status_code)