X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=ice_validator%2Ftests%2Ftest_heat_template_structure.py;h=71571533224eeeafd3525ef56f3bc0f2c08218fa;hb=HEAD;hp=1e0b8c8c5fe0d92b9c45d5f2608b73a9cb764f8b;hpb=f5edc06be0d8bedeb0904b348ba5e3e67c74f186;p=vvp%2Fvalidation-scripts.git

diff --git a/ice_validator/tests/test_heat_template_structure.py b/ice_validator/tests/test_heat_template_structure.py
index 1e0b8c8..7157153 100644
--- a/ice_validator/tests/test_heat_template_structure.py
+++ b/ice_validator/tests/test_heat_template_structure.py
@@ -1,8 +1,8 @@
 # -*- coding: utf8 -*-
-# ============LICENSE_START=======================================================
+# ============LICENSE_START====================================================
 # org.onap.vvp/validation-scripts
 # ===================================================================
-# Copyright © 2018 AT&T Intellectual Property. All rights reserved.
+# Copyright © 2017 AT&T Intellectual Property. All rights reserved.
 # ===================================================================
 #
 # Unless otherwise specified, all software contained herein is licensed
@@ -35,89 +35,79 @@
 #
 # ============LICENSE_END============================================
 #
-# ECOMP is a trademark and service mark of AT&T Intellectual Property.
 #
-from .helpers import validates
-import yaml
 
+"""Test heat template structure
+"""
+import pytest
 
-def test_heat_template_structure(yaml_file):
-    '''
-    Check that all heat templates only have the allowed sections
-    '''
-    key_values = ["heat_template_version", "description",
-                  "parameter_groups", "parameters", "resources",
-                  "outputs", "conditions"]
+from tests import cached_yaml as yaml
+from .helpers import validates, is_base_module, load_yaml
 
-    with open(yaml_file) as fh:
-        yml = yaml.load(fh)
-    assert all([k in key_values for k in yml])
+VERSION = "1.2.0"
 
+# pylint: disable=invalid-name
 
-@validates('R-27078', 'R-39402', 'R-35414')
-def test_heat_template_structure_contains_required_sections(yaml_file):
-    '''
+
+@validates("R-27078")
+def test_heat_template_structure_contains_heat_template_version(yaml_file):
+    """
     Check that all heat templates have the required sections
-    '''
-    required_key_values = ["heat_template_version", "description",
-                           "parameters", "resources"]
+    """
+    template = load_yaml(yaml_file)
+    assert (
+        "heat_template_version" in template
+    ), "This template must contain a heat_template_version section"
 
-    with open(yaml_file) as fh:
-        yml = yaml.load(fh)
-    assert all([k in yml for k in required_key_values])
 
+@validates("R-39402")
+def test_heat_template_structure_contains_description(yaml_file):
+    """
+    Check that all heat templates have the required sections
+    """
+    template = load_yaml(yaml_file)
+    assert "description" in template, "This template must contain a description section"
 
-def test_heat_template_structure_sections_have_the_right_format(yaml_file):
-    '''
-    Check that all heat templates have sections of the right format.
-    Do note that it only tests for dicts or not dicts currently.
-    '''
-    key_values = ["heat_template_version", "description",
-                  "parameter_groups", "parameters", "resources",
-                  "outputs", "conditions"]
-    key_values_not_dicts = ["heat_template_version", "description"]
 
-    with open(yaml_file) as fh:
-        yml = yaml.load(fh)
+@validates("R-35414")
+def test_heat_template_structure_contains_parameters(yaml_file):
+    """
+    Check that all heat templates have the required sections
+    """
+    if is_base_module(yaml_file):
+        pytest.skip("Not applicable to base modules")
+    template = load_yaml(yaml_file)
+    assert "parameters" in template and len(
+        template["parameters"].keys()
+    ), "This template must contain a parameters section"
+
 
-    is_dict = 0
-    should_be_dict = 0
-    is_not_dict = 0
-    should_not_be_dict = 0
-    for key_value in key_values:
-        if key_value in yml:
-            if isinstance(yml[key_value], dict):
-                is_dict += 1
-                if key_value not in key_values_not_dicts:
-                    should_be_dict += 1
-            elif not isinstance(yml[key_value], list):
-                is_not_dict += 1
-                if key_value in key_values_not_dicts:
-                    should_not_be_dict += 1
-    assert (is_dict == should_be_dict and
-            is_not_dict == should_not_be_dict)
+@validates("R-23664")
+def test_heat_template_structure_contains_resources(heat_template):
+    """
+    Check that all heat templates have the required sections
+    """
+    if is_base_module(heat_template):
+        pytest.skip("Not applicable to base modules")
+    template = load_yaml(heat_template)
+    assert "resources" in template, "This template must contain a resources section"
 
 
-@validates('R-11441')
+@validates("R-11441")
 def test_parameter_type(yaml_file):
-    types = [
-        'string',
-        'number',
-        'json',
-        'comma_delimited_list',
-        'boolean',
-    ]
+    """A VNF's Heat Orchestration Template's parameter type **MUST**
+    be one of the following values:
+    """
+    types = ["string", "number", "json", "comma_delimited_list", "boolean"]
     with open(yaml_file) as fh:
         yml = yaml.load(fh)
-    for key, param in yml.get('parameters', {}).items():
-        assert isinstance(param, dict), '%s parameter %s is not dict' % (
-            yaml_file,
-            key)
-        assert 'type' in param, '%s parameter %s has no "type"' % (
-            yaml_file,
-            key)
-        typ = param['type']
+    for key, param in yml.get("parameters", {}).items():
+        assert isinstance(param, dict), "%s parameter %s is not dict" % (yaml_file, key)
+        if "type" not in param:
+            continue
+        typ = param["type"]
         assert typ in types, '%s parameter %s has invalid type "%s"' % (
             yaml_file,
             key,
-            typ)
+            typ,
+        )