sr_stage update

Change-Id: Ib7bad69f0d582f5319de0e5c5d5e45850203bc80
Signed-off-by: Junghyun Kim <jh0822.kim@samsung.com>

diff --git a/job_trbs_test_result_receiver.py b/job_trbs_test_result_receiver.py
index 9092856..908301e 100755 (executable)
--- a/job_trbs_test_result_receiver.py
+++ b/job_trbs_test_result_receiver.py
@@ -8,8 +8,42 @@ import MySQLdb.cursors
 import xml.etree.ElementTree as ET
 import urllib2
 import re
+import datetime
 from codebase import *
 
+def check_test_failure(data):
+    # tct test.
+    for suite in data['TCT']['SUITE'].keys():
+        case_tests = ['pass_case', 'fail_case', 'block_case', 'na_case']
+        for c in case_tests:
+            if data['TCT']['SUITE'][suite][c] != data['REF']['SUITE'][suite][c]:
+                return 0
+
+    # smoke test.
+    if data['SMOKES']['COUNT']['fail_case'] > 0:
+        return 0
+
+    # perf test
+    if data['PERFS']['COUNT']['fail_case'] > 0:
+        return 0
+
+    return 1
+
+def insert_test_stage(data, MYSQL, sr_id):
+    status = 'F';
+    if check_test_failure(data):
+        status = 'S';
+
+    cur_time = datetime.datetime.now()
+
+    sql_data = (sr_id, cur_time, cur_time, status, status)
+    sql = "INSERT INTO sr_stage (sr_status_id, info_stage_id, "\
+        " stage_start_time, stage_end_time, stage_status, build_project_id)"\
+        " VALUES (%s, 101, %s, %s, %s, 0) ON DUPLICATE KEY UPDATE stage_status=%s"
+    pprint.pprint(sql)
+    pprint.pprint(sql_data)
+    MYSQL['cursor'].execute(sql, sql_data)
+    MYSQL['connection'].commit()
 
 
 if __name__ == "__main__":
@@ -133,6 +167,7 @@ if __name__ == "__main__":
         exit(1)
 
 
+    insert_test_stage(CONFIG['RAW'], MYSQL, DINFO['sr_status_id'])
 
     SQL['REF'] = {}
     # select ref