X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Ftests%2Funit%2Futils%2Ftest_vnfd_utils.py;h=1474a1c9f21271a0ee763ca9bf0b90a626af4cf1;hb=4584f8e86a492d67d120bfea1195eff1475c0a65;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..1474a1c 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,28 @@
 ##
 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]
-            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]
+    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')
+            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)