Additional Unit Tests for core modules

[functest.git] / functest / ci / generate_report.py

diff --git a/functest/ci/generate_report.py b/functest/ci/generate_report.py
index c934372..e400b1b 100644 (file)
--- a/functest/ci/generate_report.py
+++ b/functest/ci/generate_report.py
@@ -1,11 +1,17 @@
+#!/usr/bin/env python
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+#
 import json
-import os
+import logging
 import re
 import urllib2
 
-import functest.utils.functest_logger as ft_logger
 import functest.utils.functest_utils as ft_utils
-
+from functest.utils.constants import CONST
 
 COL_1_LEN = 25
 COL_2_LEN = 15
@@ -15,13 +21,12 @@ COL_5_LEN = 75
 
 # If we run from CI (Jenkins) we will push the results to the DB
 # and then we can print the url to the specific test result
-IS_CI_RUN = False
-BUILD_TAG = None
 
-logger = ft_logger.Logger("generate_report").getLogger()
+
+logger = logging.getLogger(__name__)
 
 
-def init(tiers_to_run):
+def init(tiers_to_run=[]):
     test_cases_arr = []
     for tier in tiers_to_run:
         for test in tier.get_tests():
@@ -34,7 +39,8 @@ def init(tiers_to_run):
 
 
 def get_results_from_db():
-    url = ft_utils.get_db_url() + '/results?build_tag=' + BUILD_TAG
+    url = "%s?build_tag=%s" % (ft_utils.get_db_url(),
+                               CONST.BUILD_TAG)
     logger.debug("Query to rest api: %s" % url)
     try:
         data = json.load(urllib2.urlopen(url))
@@ -50,7 +56,7 @@ def get_data(test, results):
     for test_db in results:
         if test['test_name'] in test_db['case_name']:
             id = test_db['_id']
-            url = ft_utils.get_db_url() + '/results/' + id
+            url = ft_utils.get_db_url() + '/' + id
             test_result = test_db['criteria']
 
     return {"url": url, "result": test_result}
@@ -61,7 +67,7 @@ def print_line(w1, w2='', w3='', w4='', w5=''):
            '| ' + w2.ljust(COL_2_LEN - 1) +
            '| ' + w3.ljust(COL_3_LEN - 1) +
            '| ' + w4.ljust(COL_4_LEN - 1))
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         str += ('| ' + w5.ljust(COL_5_LEN - 1))
     str += '|\n'
     return str
@@ -69,7 +75,7 @@ def print_line(w1, w2='', w3='', w4='', w5=''):
 
 def print_line_no_columns(str):
     TOTAL_LEN = COL_1_LEN + COL_2_LEN + COL_3_LEN + COL_4_LEN + 2
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         TOTAL_LEN += COL_5_LEN + 1
     return ('| ' + str.ljust(TOTAL_LEN) + "|\n")
 
@@ -79,21 +85,16 @@ def print_separator(char="=", delimiter="+"):
            delimiter + char * COL_2_LEN +
            delimiter + char * COL_3_LEN +
            delimiter + char * COL_4_LEN)
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         str += (delimiter + char * COL_5_LEN)
     str += '+\n'
     return str
 
 
-def main(args):
-    global BUILD_TAG, IS_CI_RUN
+def main(args=[]):
     executed_test_cases = args
 
-    BUILD_TAG = os.getenv("BUILD_TAG")
-    if BUILD_TAG is not None:
-        IS_CI_RUN = True
-
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         results = get_results_from_db()
         if results is not None:
             for test in executed_test_cases:
@@ -102,19 +103,15 @@ def main(args):
                              "result": data['result']})
 
     TOTAL_LEN = COL_1_LEN + COL_2_LEN + COL_3_LEN + COL_4_LEN
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         TOTAL_LEN += COL_5_LEN
     MID = TOTAL_LEN / 2
 
-    INSTALLER = os.getenv('INSTALLER_TYPE', 'unknown')
-    CI_LOOP = os.getenv('CI_LOOP')
-    SCENARIO = os.getenv('DEPLOY_SCENARIO')
-    CI_LOOP = None
-    if BUILD_TAG is not None:
-        if re.search("daily", BUILD_TAG) is not None:
-            CI_LOOP = "daily"
+    if CONST.__getattribute__('BUILD_TAG') is not None:
+        if re.search("daily", CONST.__getattribute__('BUILD_TAG')) is not None:
+            CONST.__setattr__('CI_LOOP', 'daily')
         else:
-            CI_LOOP = "weekly"
+            CONST.__setattr__('CI_LOOP', 'weekly')
 
     str = ''
     str += print_separator('=', delimiter="=")
@@ -122,16 +119,21 @@ def main(args):
     str += print_separator('=', delimiter="=")
     str += print_line_no_columns(' ')
     str += print_line_no_columns(" Deployment description:")
-    str += print_line_no_columns("   INSTALLER: %s" % INSTALLER)
-    if SCENARIO is not None:
-        str += print_line_no_columns("   SCENARIO:  %s" % SCENARIO)
-    if BUILD_TAG is not None:
-        str += print_line_no_columns("   BUILD TAG: %s" % BUILD_TAG)
-    if CI_LOOP is not None:
-        str += print_line_no_columns("   CI LOOP:   %s" % CI_LOOP)
+    str += print_line_no_columns("   INSTALLER: %s"
+                                 % CONST.__getattribute__('INSTALLER_TYPE'))
+    if CONST.__getattribute__('DEPLOY_SCENARIO') is not None:
+        str += print_line_no_columns("   SCENARIO:  %s"
+                                     % CONST.__getattribute__(
+                                         'DEPLOY_SCENARIO'))
+    if CONST.__getattribute__('BUILD_TAG') is not None:
+        str += print_line_no_columns("   BUILD TAG: %s"
+                                     % CONST.__getattribute__('BUILD_TAG'))
+    if CONST.__getattribute__('CI_LOOP') is not None:
+        str += print_line_no_columns("   CI LOOP:   %s"
+                                     % CONST.__getattribute__('CI_LOOP'))
     str += print_line_no_columns(' ')
     str += print_separator('=')
-    if IS_CI_RUN:
+    if CONST.__getattribute__('IS_CI_RUN'):
         str += print_line('TEST CASE', 'TIER', 'DURATION', 'RESULT', 'URL')
     else:
         str += print_line('TEST CASE', 'TIER', 'DURATION', 'RESULT')
@@ -145,8 +147,3 @@ def main(args):
         str += print_separator('-')
 
     logger.info("\n\n\n%s" % str)
-
-
-if __name__ == '__main__':
-    import sys
-    main(sys.argv[1:])