update the logic to distribute service from sdc.
[modeling/etsicatalog.git] / genericparser / packages / tests / test_servicepackage.py
1 # Copyright (c) 2019, CMCC Technologies. Co., Ltd.
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 import json
15
16 from django.test import TestCase, Client
17 from mock import mock
18 from rest_framework import status
19
20 from genericparser.packages.biz.sdc_service_package import ServicePackage
21 from genericparser.packages.const import PKG_STATUS
22 from genericparser.pub.database.models import ServicePackageModel, VnfPackageModel, PnfPackageModel
23 from genericparser.pub.exceptions import PackageNotFoundException, PackageHasExistsException, GenericparserException
24 from genericparser.pub.msapi import sdc
25 from genericparser.pub.utils import toscaparsers
26
27 PARSER_BASE_URL = "/api/parser/v1"
28
29
30 class TestServicePackage(TestCase):
31     """ Test case for Service Package operations"""
32
33     def setUp(self):
34         self.client = Client()
35         ServicePackageModel.objects.filter().delete()
36         self.sd_data = {
37             "inputs": {
38                 "sdwanvpnresource_list": [
39                     {
40                         "sdwanvpn_topology": "",
41                         "required": True,
42                         "type": "string"
43                     },
44                     {
45                         "sdwansitelan_list": [
46                             {
47                                 "deviceName": "",
48                                 "required": True,
49                                 "type": "string",
50                                 "description": "The device name in the site"
51                             }
52                         ]
53                     }
54                 ],
55                 "sdwansiteresource_list": [
56                     {
57                         "sdwansite_controlPoint": "",
58                         "required": False,
59                         "type": "string",
60                         "description": "The control point of the site,only for sd-wan-edge"
61                     },
62                     {
63                         "sdwandevice_list": [
64                             {
65                                 "systemIp": "",
66                                 "required": False,
67                                 "type": "string",
68                                 "description": "The system ip of the device"
69                             }
70                         ]
71                     }
72                 ]
73             },
74             "pnfs": [
75                 {
76                     "pnf_id": "m6000_s",
77                     "cps": [],
78                     "description": "",
79                     "properties": {
80                         "vendor": "zte",
81                         "request_reclassification": False,
82                         "pnf_type": "m6000s",
83                         "version": "1.0",
84                         "management_address": "111111",
85                         "id": "m6000_s",
86                         "nsh_aware": False
87                     }
88                 }
89             ],
90             "description": "",
91             "graph": {
92                 "sdwansiteresource": [
93                     "sdwanvpnresource"
94                 ],
95                 "sdwanvpnresource": []
96             },
97             "basepath": "c:\\users\\cmcc\\appdata\\local\\temp\\tmpn79jwc\\Definitions",
98             "vnfs": [
99                 {
100                     "vnf_id": "sdwansiteresource",
101                     "description": "",
102                     "properties": {
103                         "sdwandevice_type": "",
104                         "sdwandevice_class": "PNF",
105                         "multi_stage_design": "False",
106                         "min_instances": "1",
107                         "sdwansite_controlPoint": "",
108                         "id": "cd557883-ac4b-462d-aa01-421b5fa606b1",
109                         "sdwansite_longitude": "",
110                         "sdwansite_latitude": "",
111                         "sdwansite_postcode": "",
112                         "sdwansite_type": "",
113                         "nf_naming": {
114                             "ecomp_generated_naming": True
115                         },
116                         "sdwansite_emails": "",
117                         "sdwansite_role": "",
118                         "vnfm_info": "",
119                         "sdwansite_address": "",
120                         "sdwansite_description": "",
121                         "availability_zone_max_count": "1",
122                         "sdwansite_name": ""
123                     },
124                     "dependencies": [],
125                     "networks": [],
126                     "metadata": {
127                         "category": "Configuration",
128                         "subcategory": "Configuration",
129                         "UUID": "cd557883-ac4b-462d-aa01-421b5fa606b1",
130                         "invariantUUID": "c83b621e-e267-4910-a75a-a2a5957296e4",
131                         "name": "sdwansiteresource",
132                         "customizationUUID": "673dd6b3-3a06-4ef0-8ad0-8c26224b08f7",
133                         "resourceVendorRelease": "1.0",
134                         "version": "1.0",
135                         "resourceVendor": "onap",
136                         "resourceVendorModelNumber": "",
137                         "type": "VF",
138                         "description": "sdwansiteresource"
139                     }
140                 }
141             ],
142             "vls": [],
143             "service": {
144                 "type": "org.openecomp.service.EnhanceService",
145                 "requirements": {
146                     "sdwanvpnresource.sdwanvpn.dependency": [
147                         "sdwanvpnresource",
148                         "sdwanvpn.dependency"
149                     ],
150                     "sdwansiteresource.sdwansitewan.dependency": [
151                         "sdwansiteresource",
152                         "sdwansitewan.dependency"
153                     ],
154                     "sdwansiteresource.sdwandevice.dependency": [
155                         "sdwansiteresource",
156                         "sdwandevice.dependency"
157                     ],
158                     "sdwanvpnresource.sdwansitelan.dependency": [
159                         "sdwanvpnresource",
160                         "sdwansitelan.dependency"
161                     ],
162                     "sdwanvpnresource.sdwanvpn.device": [
163                         "sdwanvpnresource",
164                         "sdwanvpn.device"
165                     ],
166                     "sdwansiteresource.sdwansite.device": [
167                         "sdwansiteresource",
168                         "sdwansite.device"
169                     ],
170                     "sdwansiteresource.sdwansite.dependency": [
171                         "sdwansiteresource",
172                         "sdwansite.dependency"
173                     ],
174                     "sdwanvpnresource.sdwansitelan.device": [
175                         "sdwanvpnresource",
176                         "sdwansitelan.device"
177                     ],
178                     "sdwansiteresource.sdwansitewan.device": [
179                         "sdwansiteresource",
180                         "sdwansitewan.device"
181                     ],
182                     "sdwansiteresource.sdwandevice.device": [
183                         "sdwansiteresource",
184                         "sdwandevice.device"
185                     ]
186                 },
187                 "properties": {
188                     "descriptor_id": "49ee73f4-1e31-4054-b871-eb9b1c29999b",
189                     "designer": "",
190                     "invariant_id": "5de07996-7ff0-4ec1-b93c-e3a00bb3f207",
191                     "name": "Enhance_Service",
192                     "verison": ""
193                 },
194                 "capabilities": {
195                     "sdwansiteresource.sdwandevice.feature": [
196                         "sdwansiteresource",
197                         "sdwandevice.feature"
198                     ],
199                     "sdwanvpnresource.sdwanvpn.feature": [
200                         "sdwanvpnresource",
201                         "sdwanvpn.feature"
202                     ],
203                     "sdwanvpnresource.sdwanvpn.link": [
204                         "sdwanvpnresource",
205                         "sdwanvpn.link"
206                     ],
207                     "sdwansiteresource.sdwansite.feature": [
208                         "sdwansiteresource",
209                         "sdwansite.feature"
210                     ],
211                     "sdwansiteresource.sdwansitewan.feature": [
212                         "sdwansiteresource",
213                         "sdwansitewan.feature"
214                     ],
215                     "sdwanvpnresource.sdwansitelan.feature": [
216                         "sdwanvpnresource",
217                         "sdwansitelan.feature"
218                     ]
219                 },
220                 "metadata": {
221                     "category": "E2E Service",
222                     "serviceType": "",
223                     "description": "Enhance_Service",
224                     "instantiationType": "A-la-carte",
225                     "type": "Service",
226                     "environmentContext": "General_Revenue-Bearing",
227                     "serviceEcompNaming": True,
228                     "UUID": "49ee73f4-1e31-4054-b871-eb9b1c29999b",
229                     "ecompGeneratedNaming": True,
230                     "serviceRole": "",
231                     "invariantUUID": "5de07996-7ff0-4ec1-b93c-e3a00bb3f207",
232                     "namingPolicy": "",
233                     "name": "Enhance_Service"
234                 }
235             },
236             "metadata": {
237                 "category": "E2E Service",
238                 "serviceType": "",
239                 "description": "Enhance_Service",
240                 "instantiationType": "A-la-carte",
241                 "type": "Service",
242                 "environmentContext": "General_Revenue-Bearing",
243                 "serviceEcompNaming": True,
244                 "UUID": "49ee73f4-1e31-4054-b871-eb9b1c29999b",
245                 "ecompGeneratedNaming": True,
246                 "serviceRole": "",
247                 "invariantUUID": "5de07996-7ff0-4ec1-b93c-e3a00bb3f207",
248                 "namingPolicy": "",
249                 "name": "Enhance_Service"
250             }
251         }
252         self.asset_data = {
253             "uuid": "1",
254             "invariantUUID": "63eaec39-ffbe-411c-a838-448f2c73f7eb",
255             "name": "underlayvpn",
256             "version": "2.0",
257             "toscaModelURL": "/sdc/v1/catalog/resources/c94490a0-f7ef-48be-b3f8-8d8662a37236/toscaModel",
258             "category": "Volte",
259             "subCategory": "VolteVNF",
260             "resourceType": "VF",
261             "lifecycleState": "CERTIFIED",
262             "distributionStatus": "DISTRIBUTION_APPROVED",
263             "lastUpdaterUserId": "jh0003",
264             "resources": [
265                 {
266                     "resourceInstanceName": "contrailV2VLANSubInterface 0",
267                     "resourceName": "contrailV2VLANSubInterface",
268                     "resourceInvariantUUID": "4d31b775-af63-491d-89f1-254e218e7140",
269                     "resourceVersion": "1.0",
270                     "resoucreType": "CP",
271                     "resourceUUID": "cd557883-ac4b-462d-aa01-421b5fa606b1"
272                 },
273                 {
274                     "resourceInstanceName": "Network 0",
275                     "resourceName": "Network",
276                     "resourceInvariantUUID": "f90f567e-7d7d-4216-af38-6bca0637c59f",
277                     "resourceVersion": "1.0",
278                     "resoucreType": "VL",
279                     "resourceUUID": "m6000_s"
280                 }
281             ]
282         }
283
284     def tearDown(self):
285         pass
286
287     ###############################################################
288
289     def test_service_pkg_distribute_when_pkg_exists(self):
290         ServicePackageModel(servicePackageId="1", servicedId="2").save()
291         csar_id = "1"
292         try:
293             ServicePackage().on_distribute(csar_id)
294         except PackageHasExistsException as e:
295             self.assertEqual("Service CSAR(1) already exists.", e.args[0])
296
297     @mock.patch.object(sdc, 'get_asset')
298     def test_service_pkg_distribute_when_fail_get_artifacts(self, mock_get_asset):
299         mock_get_asset.side_effect = GenericparserException("Failed to query artifact(services,1) from sdc.")
300         csar_id = "1"
301         try:
302             ServicePackage().on_distribute(csar_id)
303         except Exception as e:
304             self.assertTrue(isinstance(e, GenericparserException))
305             self.assertEqual("Failed to query artifact(services,1) from sdc.", e.args[0])
306
307     @mock.patch.object(sdc, 'get_asset')
308     def test_service_pkg_distribute_when_resource_not_distribute(self, mock_get_asset):
309         mock_get_asset.return_value = self.asset_data
310         csar_id = "1"
311         try:
312             ServicePackage().on_distribute(csar_id)
313         except Exception as e:
314             self.assertTrue(isinstance(e, GenericparserException))
315             self.assertEqual("Resource (cd557883-ac4b-462d-aa01-421b5fa606b1) is not distributed.", e.args[0])
316
317     @mock.patch.object(sdc, 'get_asset')
318     @mock.patch.object(sdc, 'download_artifacts')
319     def test_service_pkg_distribute_when_fail_download_artifacts(self, mock_get_asset, mock_download_artifacts):
320         mock_get_asset.return_value = self.asset_data
321         mock_download_artifacts.side_effect = GenericparserException("Failed to download 1 from sdc.")
322         csar_id = "1"
323         VnfPackageModel(vnfPackageId="cd557883-ac4b-462d-aa01-421b5fa606b1",
324                         vnfdId="cd557883-ac4b-462d-aa01-421b5fa606b1").save()
325         PnfPackageModel(pnfPackageId="m6000_s", pnfdId="m6000_s").save()
326
327         try:
328             ServicePackage().on_distribute(csar_id)
329         except Exception as e:
330             self.assertTrue(isinstance(e, GenericparserException))
331             self.assertEqual("Failed to download 1 from sdc.", e.args[0])
332
333     @mock.patch.object(sdc, 'get_asset')
334     @mock.patch.object(sdc, 'download_artifacts')
335     @mock.patch.object(toscaparsers, 'parse_sd')
336     def test_service_pkg_distribute(self, mock_parse_sd, mock_download_artifacts, mock_get_asset):
337         mock_parse_sd.return_value = json.JSONEncoder().encode(self.sd_data)
338         mock_download_artifacts.return_value = "/test.csar"
339         mock_get_asset.return_value = self.asset_data
340         VnfPackageModel(vnfPackageId="cd557883-ac4b-462d-aa01-421b5fa606b1",
341                         vnfdId="cd557883-ac4b-462d-aa01-421b5fa606b1").save()
342         PnfPackageModel(pnfPackageId="m6000_s", pnfdId="m6000_s").save()
343         ServicePackage().on_distribute(csar_id="1")
344
345         service_package = ServicePackageModel.objects.filter(servicePackageId="1").first()
346         self.assertEqual("5de07996-7ff0-4ec1-b93c-e3a00bb3f207", service_package.invariantId)
347         self.assertEqual("Enhance_Service", service_package.servicedName)
348         self.assertEqual(PKG_STATUS.ONBOARDED, service_package.onboardingState)
349         self.assertEqual(PKG_STATUS.ENABLED, service_package.operationalState)
350         self.assertEqual(PKG_STATUS.NOT_IN_USE, service_package.usageState)
351
352     def test_api_service_pkg_distribute_when_pkg_exists(self):
353         ServicePackageModel(servicePackageId="1", servicedId="2").save()
354         resp = self.client.post(
355             PARSER_BASE_URL + "/service_packages", {"csarId": "1"}, format='json')
356         self.assertEqual(resp.status_code, status.HTTP_400_BAD_REQUEST)
357         self.assertEqual("Service CSAR(1) already exists.", resp.data["errorMessage"])
358
359     ###############################################################
360
361     def test_service_pkg_get_all(self):
362         ServicePackageModel(
363             servicePackageId="13",
364             servicedId="2",
365             servicedDesigner="2",
366             servicedVersion="2",
367             servicePackageUri="13.csar",
368             servicedModel="").save()
369         ServicePackageModel(
370             servicePackageId="14",
371             servicedId="3",
372             servicedDesigner="3",
373             servicedVersion="3",
374             servicePackageUri="14.csar",
375             servicedModel="").save()
376         csars = ServicePackage().get_csars()
377         self.assertEqual(2, len(csars))
378
379     def test_api_service_pkg_get_all(self):
380         ServicePackageModel(
381             servicePackageId="13",
382             servicedId="2",
383             servicedDesigner="2",
384             servicedVersion="2",
385             servicePackageUri="13.csar",
386             servicedModel="").save()
387         ServicePackageModel(
388             servicePackageId="14",
389             servicedId="3",
390             servicedDesigner="3",
391             servicedVersion="3",
392             servicePackageUri="14.csar",
393             servicedModel="").save()
394         resp = self.client.get(PARSER_BASE_URL + "/service_packages")
395         self.assertEqual(resp.status_code, status.HTTP_200_OK)
396
397     ###############################################################
398
399     def test_service_pkg_get_one(self):
400         ServicePackageModel(
401             servicePackageId="14",
402             servicedId="2",
403             servicedDesigner="3",
404             servicedVersion="4",
405             servicePackageUri="14.csar",
406             servicedModel="").save()
407         csar = ServicePackage().get_csar(14)
408         self.assertEqual(14, csar['csarId'])
409
410     def test_service_pkg_get_one_not_found(self):
411         try:
412             ServicePackage().get_csar(1000)
413         except PackageNotFoundException as e:
414             self.assertEqual("Service package[1000] not Found.", e.args[0])
415
416     def test_api_service_pkg_get_one(self):
417         ServicePackageModel(
418             servicePackageId="14",
419             servicedId="2",
420             servicedDesigner="3",
421             servicedVersion="4",
422             servicePackageUri="14.csar",
423             servicedModel="").save()
424         resp = self.client.get(PARSER_BASE_URL + "/service_packages/14")
425         self.assertEqual(resp.status_code, status.HTTP_200_OK)
426
427     def test_api_service_pkg_get_one_not_found(self):
428         resp = self.client.get(PARSER_BASE_URL + "/service_packages/22")
429         self.assertEqual(resp.status_code, status.HTTP_404_NOT_FOUND)
430         self.assertEqual(
431             {"errorMessage": "Service package[22] not Found.", 'error': 404},
432             resp.data)
433
434     ###############################################################
435
436     def test_service_pkg_normal_delete(self):
437         ServicePackageModel(servicePackageId="8", servicedId="2").save()
438         sp = ServicePackageModel.objects.filter(servicePackageId=8)
439         self.assertEqual(1, len(sp))
440         ServicePackage().delete_csar("8")
441         sp = ServicePackageModel.objects.filter(servicePackageId=8)
442         self.assertEqual(0, len(sp))
443
444     def test_service_pkg_normal_delete_not_found(self):
445         try:
446             ServicePackage().delete_csar("8000")
447         except PackageNotFoundException as e:
448             self.assertEqual("Service package[8000] not Found.", e.args[0])
449
450     def test_api_service_pkg_normal_delete(self):
451         ServicePackageModel(servicePackageId="8", servicedId="2").save()
452         resp = self.client.delete(PARSER_BASE_URL + "/service_packages/8")
453         self.assertEqual(resp.status_code, status.HTTP_204_NO_CONTENT)
454
455     ###############################################################
456
457     @mock.patch.object(toscaparsers, 'parse_sd')
458     def test_service_pkg_parser(self, mock_parse_sd):
459         ServicePackageModel(servicePackageId="8", servicedId="2").save()
460         mock_parse_sd.return_value = json.JSONEncoder().encode({"a": "b"})
461
462         inputs = []
463         ret = ServicePackage().parse_serviced("8", inputs)
464         self.assertTrue({"model": '{"c": "d"}'}, ret)
465
466     def test_service_pkg_parser_not_found(self):
467         try:
468             csar_id = "8000"
469             inputs = []
470             ServicePackage().parse_serviced(csar_id, inputs)
471         except PackageNotFoundException as e:
472             self.assertEqual("Service CSAR(8000) does not exist.", e.args[0])
473
474     def test_api_service_pkg_parser_not_found(self):
475         query_data = {
476             "csarId": "1",
477             "packageType": "Service",
478             "inputs": "string"
479         }
480         resp = self.client.post(PARSER_BASE_URL + "/parser", query_data, format='json')
481         self.assertEqual(resp.status_code, status.HTTP_404_NOT_FOUND)