[VVP] udpating scripts for casablanca

[vvp/validation-scripts.git] / ice_validator / tests / cached_yaml.py

diff --git a/ice_validator/tests/test_env_files_provided.py b/ice_validator/tests/cached_yaml.py
similarity index 79%
rename from ice_validator/tests/test_env_files_provided.py
rename to ice_validator/tests/cached_yaml.py
index d784d20..99fbd2c 100644 (file)
--- a/ice_validator/tests/test_env_files_provided.py
+++ b/ice_validator/tests/cached_yaml.py
@@ -34,19 +34,27 @@
 # limitations under the License.
 #
 # ============LICENSE_END============================================
-#
-# ECOMP is a trademark and service mark of AT&T Intellectual Property.
-#
 
-'''
-test_env_files_provided
-'''
+import os
+
+import yaml
+
+YAML_CACHE = {}
+resolver = yaml.resolver
+YAMLError = yaml.YAMLError
+constructor = yaml.constructor
+
 
-VERSION = '1.0.0'
+def add_constructor(tag, constructor):
+    yaml.add_constructor(
+        tag,
+        constructor,
+    )
 
 
-def test_env_files_provided(env_files):
-    '''
-    Make sure environment files have been provided
-    '''
-    assert len(env_files) > 0, '"*.env" not found'
+def load(fp):
+    """Provides cached loading of yaml files"""
+    abs_path = os.path.abspath(fp.name)
+    if abs_path not in YAML_CACHE:
+        YAML_CACHE[abs_path] = yaml.safe_load(fp)
+    return YAML_CACHE[abs_path]