1 files changed, 88 insertions, 0 deletions
diff --git a/tests/test_tosca_file_handler.py b/tests/test_tosca_file_handler.py
new file mode 100644
index 0000000..e288693
--- /dev/null
+++ b/tests/test_tosca_file_handler.py
@@ -0,0 +1,88 @@
+#   Copyright 2022 Orange, Deutsche Telekom AG
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+
+import logging
+import json
+import oyaml as yaml
+import os
+import os.path
+import unittest
+
+from onapsdk.exceptions import ValidationError
+import onapsdk.utils.tosca_file_handler as tosca_file_handler
+
+
+__author__ = "Morgan Richomme <morgan.richomme@orange.com>"
+
+
+class ToscaFileHandlerTestingBase(unittest.TestCase):
+
+    """The super class which testing classes could inherit."""
+
+    logging.disable(logging.CRITICAL)
+
+    _root_path = os.getcwd().rsplit('/onapsdk')[0]
+    _foo_path = _root_path +"/tests/data/service-Ubuntu16-template.yml"
+
+
+    def setUp(self):
+        pass
+
+    def test_get_parameter_from_yaml(self):
+        with open(self._foo_path) as f:
+            model = json.dumps(yaml.safe_load(f))
+        param = tosca_file_handler.get_parameter_from_yaml(
+            "metadata", model)
+        self.assertEqual(param['name'], "ubuntu16")
+
+    def test_get_wrong_parameter_from_yaml(self):
+        with open(self._foo_path) as f:
+            model = json.dumps(yaml.safe_load(f))
+        with self.assertRaises(ValidationError):
+            tosca_file_handler.get_parameter_from_yaml(
+                "wrong_parameter", model)
+
+    def test_get_parameter_from_wrong_yaml(self):
+        with self.assertRaises(FileNotFoundError):
+            with open("wrong_path") as f:
+                model = json.dumps(yaml.safe_load(f))
+                tosca_file_handler.get_parameter_from_yaml(
+                    "metadata", model)
+
+    def test_get_random_string_generator(self):
+        self.assertEqual(
+            len(tosca_file_handler.random_string_generator()), 6)
+
+    def test_get_vf_list_from_tosca_file(self):
+        with open(self._foo_path) as f:
+            model = json.dumps(yaml.safe_load(f))
+        vf_list = tosca_file_handler.get_vf_list_from_tosca_file(model)
+        self.assertEqual(vf_list[0], 'ubuntu16_VF')
+
+    def test_get_modules_list_from_tosca_file(self):
+        with open(self._foo_path) as f:
+            model = json.dumps(yaml.safe_load(f))
+        vf_modules = tosca_file_handler.get_modules_list_from_tosca_file(model)
+        self.assertEqual(len(vf_modules), 1)
+
+    # def get_vf_list_from_tosca_file_wrong_model(self):
+    #     with self.assertRaises(FileNotFoundError):
+    #         tosca_file_handler.get_vf_list_from_tosca_file(
+    #             self._root_path + "wrong_path")
+
+if __name__ == "__main__":
+    # logging must be disabled else it calls time.time()
+    # what will break these unit tests.
+    logging.disable(logging.CRITICAL)
+    unittest.main(verbosity=2)