Merge "dovetail tool: check push_to_db success or fail"

1 files changed, 22 insertions, 9 deletions

diff --git a/dovetail/run.py b/dovetail/run.py
index f6ce4929..aae35e12 100755
--- a/dovetail/run.py
+++ b/dovetail/run.py
@@ -44,6 +44,7 @@ def run_test(testsuite, testarea, logger):
             testarea_list.append(value)
 
     duration = 0
+    start_time = time.time()
     for testcase_name in testarea_list:
         logger.info('>>[testcase]: %s', testcase_name)
         testcase = Testcase.get(testcase_name)
@@ -60,21 +61,33 @@ def run_test(testsuite, testarea, logger):
             run_testcase = False
 
         if run_testcase:
-            start_time = time.time()
             testcase.run()
-            end_time = time.time()
-            duration = end_time - start_time
 
-        if dt_cfg.dovetail_config['report_dest'].startswith("http"):
-            logger.info("Results has been pushed to database.")
-        if dt_cfg.dovetail_config['report_dest'] == "file":
-            logger.info("Results has been stored with files.")
-            result = Report.get_result(testcase)
-            Report.check_result(testcase, result)
+        check_tc_result(testcase, logger)
 
+    end_time = time.time()
+    duration = end_time - start_time
     return duration
 
 
+def check_tc_result(testcase, logger):
+    if dt_cfg.dovetail_config['report_dest'].startswith("http"):
+        if testcase.validate_type() == 'yardstick':
+            logger.info("Results have been stored with files.")
+        else:
+            if dt_utils.check_db_results(dt_cfg.dovetail_config['report_dest'],
+                                         dt_cfg.dovetail_config['build_tag'],
+                                         testcase.validate_testcase(),
+                                         logger):
+                logger.info("Results have been pushed to database.")
+            else:
+                logger.error("Fail to push results to database.")
+    if dt_cfg.dovetail_config['report_dest'] == "file":
+        logger.info("Results have been stored with files.")
+        result = Report.get_result(testcase)
+        Report.check_result(testcase, result)
+
+
 def validate_input(input_dict, check_dict, logger):
     # for 'func_tag' and 'yard_tag' options
     func_tag = input_dict['func_tag']