X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Ftests%2Funit%2Futils%2Ftest_vnfd_utils.py;h=049d2939803e00d1326bff6bcf3d89b6c43ab966;hb=refs%2Ftags%2Frelease-v9.0-start;hp=5b770dded9eec253597c5a93002556df5a8fb382;hpb=312c16596975a42d6294a1a2ca7af98b0ff2ffb5;p=osm%2FPOL.git

diff --git a/osm_policy_module/tests/unit/utils/test_vnfd_utils.py b/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
index 5b770dd..049d293 100644
--- a/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
+++ b/osm_policy_module/tests/unit/utils/test_vnfd_utils.py
@@ -23,23 +23,24 @@
 ##
 import os
 import unittest
-
 import yaml
 
 from osm_policy_module.utils.vnfd import VnfdUtils
+from osm_policy_module.core.exceptions import ManagementVduNotFound
 
 
 class VnfdUtilsTest(unittest.TestCase):
-    def test_get_mgmt_vdu_by_cp(self):
-        with open(
-                os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd_1.yaml'), 'r') as file:
-            vnfd = yaml.safe_load(file)['vnfd:vnfd-catalog']['vnfd'][0]
+    def test_get_mgmt_vdu_on_valid_descriptor(self):
+        example_file = os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd.yaml')
+        with open(example_file, 'r') as file:
+            vnfd = yaml.safe_load(file)['vnfd']
             vdu = VnfdUtils.get_mgmt_vdu(vnfd)
             self.assertEqual(vdu['id'], 'cirros_vnfd-VM')
 
-    def test_get_mgmt_vdu_by_id(self):
-        with open(
-                os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd_2.yaml'), 'r') as file:
-            vnfd = yaml.safe_load(file)['vnfd:vnfd-catalog']['vnfd'][0]
-            vdu = VnfdUtils.get_mgmt_vdu(vnfd)
-            self.assertEqual(vdu['id'], 'cirros_vnfd-VM')
+    def test_get_mgmt_vdu_on_invalid_descriptor(self):
+        example_file = os.path.join(os.path.dirname(__file__), 'examples/cirros_vdu_scaling_vnfd.yaml')
+        with open(example_file, 'r') as file:
+            vnfd = yaml.safe_load(file)['vnfd']
+            vnfd['mgmt-cp'] = 'invalid-cp'
+            with self.assertRaises(ManagementVduNotFound):
+                VnfdUtils.get_mgmt_vdu(vnfd)