Coverage for osm_policy_module/tests/unit/utils/test_vnfd_utils.py: 100%

1# -*- coding: utf-8 -*- 

2 

3# Copyright 2018 Whitestack, LLC 

4# ************************************************************* 

5 

6# This file is part of OSM Monitoring module 

7# All Rights Reserved to Whitestack, LLC 

8 

9# Licensed under the Apache License, Version 2.0 (the "License"); you may 

10# not use this file except in compliance with the License. You may obtain 

11# a copy of the License at 

12 

13# http://www.apache.org/licenses/LICENSE-2.0 

14 

15# Unless required by applicable law or agreed to in writing, software 

16# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

17# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

18# License for the specific language governing permissions and limitations 

19# under the License. 

20 

21# For those usages not covered by the Apache License, Version 2.0 please 

22# contact: bdiaz@whitestack.com or glavado@whitestack.com 

23## 

24import os 

25import unittest 

26import yaml 

27 

28from osm_policy_module.utils.vnfd import VnfdUtils 

29from osm_policy_module.core.exceptions import ManagementVduNotFound 

30 

31 

32class VnfdUtilsTest(unittest.TestCase): 

33 def test_get_mgmt_vdu_on_valid_descriptor(self): 

34 example_file = os.path.join( 

35 os.path.dirname(__file__), "examples/cirros_vdu_scaling_vnfd.yaml" 

36 ) 

37 with open(example_file, "r") as file: 

38 vnfd = yaml.safe_load(file)["vnfd"] 

39 vdu = VnfdUtils.get_mgmt_vdu(vnfd) 

40 self.assertEqual(vdu["id"], "cirros_vnfd-VM") 

41 

42 def test_get_mgmt_vdu_on_invalid_descriptor(self): 

43 example_file = os.path.join( 

44 os.path.dirname(__file__), "examples/cirros_vdu_scaling_vnfd.yaml" 

45 ) 

46 with open(example_file, "r") as file: 

47 vnfd = yaml.safe_load(file)["vnfd"] 

48 vnfd["mgmt-cp"] = "invalid-cp" 

49 with self.assertRaises(ManagementVduNotFound): 

50 VnfdUtils.get_mgmt_vdu(vnfd)