1 files changed, 125 insertions, 0 deletions
diff --git a/robotframework-onap/listeners/OVPListener.py b/robotframework-onap/listeners/OVPListener.py
new file mode 100644
index 0000000..508568b
--- /dev/null
+++ b/robotframework-onap/listeners/OVPListener.py
@@ -0,0 +1,125 @@
+# Copyright 2019 AT&T Intellectual Property. All rights reserved.
+#
+# 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 datetime
+import hashlib
+import json
+import os
+from copy import deepcopy
+from robot.libraries.BuiltIn import BuiltIn
+from zipfile import ZipFile
+
+OUTPUT_DATA = {
+    "vnf_checksum": "",
+    "build_tag": "",
+    "version": "2019.09",
+    "test_date": "",
+    "duration": "",
+    "vnf_type": "heat",
+    "testcases_list": [
+        {
+            "mandatory": "true",
+            "name": "onap-vvp.validate.heat",
+            "result": "NOT_STARTED",
+            "objective": "onap heat template validation",
+            "sub_testcase": [],
+            "portal_key_file": "report.json",
+        },
+        {
+            "mandatory": "true",
+            "name": "onap-vvp.lifecycle_validate.heat",
+            "result": "NOT_STARTED",
+            "objective": "onap vnf lifecycle validation",
+            "sub_testcase": [
+                {"name": "model-and-distribute", "result": "NOT_STARTED"},
+                {"name": "instantiation", "result": "NOT_STARTED"},
+            ],
+            "portal_key_file": "log.html",
+        },
+        {
+            "mandatory": "true",
+            "name": "stack_validation",
+            "result": "NOT_STARTED",
+            "objective": "onap vnf openstack validation",
+            "sub_testcase": [],
+            "portal_key_file": "stack_report.json",
+        },
+    ],
+}
+
+
+class OVPListener:
+    ROBOT_LISTENER_API_VERSION = 2
+
+    def __init__(self):
+        self.report = deepcopy(OUTPUT_DATA)
+
+        self.build_number = ""
+        self.build_directory = ""
+        self.output_directory = ""
+        self.template_directory = ""
+
+    def initialize(self):
+        self.build_number = BuiltIn().get_variable_value("${GLOBAL_BUILD_NUMBER}")
+        self.build_directory = BuiltIn().get_variable_value("${BUILD_DIR}")
+        self.output_directory = BuiltIn().get_variable_value("${OUTPUTDIR}")
+
+        self.template_directory = "{}/templates".format(self.build_directory)
+        self.report["build_tag"] = "vnf-validation-{}".format(self.build_number)
+        self.report["vnf_checksum"] = sha256(self.template_directory)
+
+    def start_test(self, name, attrs):
+        self.initialize()
+        date = datetime.datetime.strptime(attrs["starttime"], '%Y%m%d %H:%M:%S.%f').strftime('%Y-%m-%d %H:%M:%S')
+        self.report["test_date"] = date
+
+    def end_keyword(self, name, attrs):
+        kwname = attrs["kwname"]
+        status = attrs["status"]
+
+        if kwname == "Run VVP Validation Scripts":
+            self.report["testcases_list"][0]["result"] = status
+        elif kwname == "Model Distribution For Directory":
+            self.report["testcases_list"][1]["sub_testcase"][0]["result"] = status
+        elif kwname == "Instantiate VNF":
+            self.report["testcases_list"][1]["sub_testcase"][1]["result"] = status
+            self.report["testcases_list"][1]["result"] = status
+        elif kwname == "Run VNF Instantiation Report":
+            self.report["testcases_list"][2]["result"] = status
+
+    def end_test(self, name, attrs):
+        self.report["duration"] = attrs["elapsedtime"] / 1000
+
+    def close(self):
+        with open("{}/summary/results.json".format(self.output_directory), "w") as f:
+            json.dump(self.report, f, indent=4)
+
+
+def sha256(template_directory):
+    heat_sha = None
+
+    if os.path.exists(template_directory):
+        zip_file = "{}/tmp_heat.zip".format(template_directory)
+        with ZipFile(zip_file, "w") as zip_obj:
+            for folder_name, subfolders, filenames in os.walk(template_directory):
+                for filename in filenames:
+                    file_path = os.path.join(folder_name, filename)
+                    zip_obj.write(file_path)
+
+        with open(zip_file, "rb") as f:
+            bytes = f.read()
+            heat_sha = hashlib.sha256(bytes).hexdigest()
+
+        os.remove(zip_file)
+
+    return heat_sha