X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=ice_validator%2Ftests%2Fhelpers.py;h=502d3aaedbd119839537b9d85ee4e4c60d25ff38;hb=refs%2Fchanges%2F83%2F66483%2F4;hp=5e4f3d38d772c8ecd64a9c5affcbf7fda71f81c2;hpb=8de26dd1cc9ed33c3ab85a5014ac949f174db932;p=vvp%2Fvalidation-scripts.git

diff --git a/ice_validator/tests/helpers.py b/ice_validator/tests/helpers.py
index 5e4f3d3..502d3aa 100644
--- a/ice_validator/tests/helpers.py
+++ b/ice_validator/tests/helpers.py
@@ -41,8 +41,9 @@
 """Helpers
 """
 
+import os
 from boltons import funcutils
-import yaml
+from tests import cached_yaml as yaml
 
 VERSION = '1.1.0'
 
@@ -106,3 +107,20 @@ def validates(*requirement_ids):
         return wrapper
     decorator.requirement_ids = requirement_ids
     return decorator
+
+
+def get_environment_pair(heat_template):
+    """Returns a yaml/env pair given a yaml file"""
+    base_dir, filename = os.path.split(heat_template)
+    basename = os.path.splitext(filename)[0]
+    env_template = os.path.join(base_dir, "{}.env".format(basename))
+    if os.path.exists(env_template):
+        with open(heat_template, "r") as fh:
+            yyml = yaml.load(fh)
+        with open(env_template, "r") as fh:
+            eyml = yaml.load(fh)
+
+        environment_pair = {"name": basename, "yyml": yyml, "eyml": eyml}
+        return environment_pair
+
+    return None