fixed logging

Author: anna.yamkovaya

convert_trace_annos.py

@@ -16,7 +16,7 @@ def is_empty(value):
 def trace_test_cases_to_annos(db_path: Path, trace_file_path: Path):
     db = DbClient(db_path)
 
-    test_cases = set()
+    insertions = list()
     logger.info("Reading trace file and inserting annotations into table...")
     with open(trace_file_path, mode='r', newline='', encoding='utf-8') as trace_file:
         reader = csv.reader(trace_file)
@@ -29,27 +29,24 @@ def trace_test_cases_to_annos(db_path: Path, trace_file_path: Path):
                 current_tc = row[1]
                 test_script = EMPTY
                 concat_summary = EMPTY
-                if not is_empty(current_tc):
-                    test_cases.add(current_tc)
                 next(reader)
             elif row[0] == "Summary":
                 continue
             elif row[0] == "TestCaseEnd":
                 if not is_empty(current_tc) and not is_empty(concat_summary):
-                    case_id = db.test_cases.insert(test_script=test_script, test_case=current_tc)
-                    annotation_id = db.annotations.insert(summary=concat_summary)
-                    db.cases_to_annos.insert(case_id=case_id, annotation_id=annotation_id)
+                    case_id = db.test_cases.get_or_insert(test_script=test_script, test_case=current_tc)
+                    annotation_id = db.annotations.get_or_insert(summary=concat_summary)
+                    insertions.append(db.cases_to_annos.insert(case_id=case_id, annotation_id=annotation_id))
             else:
                 if not is_empty(row[global_columns.index("TestCase")]):
                     if current_tc != row[global_columns.index("TestCase")]:
                         current_tc = row[global_columns.index("TestCase")]
-                        test_cases.add(current_tc)
                     if is_empty(test_script) and not is_empty(row[global_columns.index("TestScript")]):
                         test_script = row[global_columns.index("TestScript")]
                     concat_summary += row[0]
 
     db.conn.commit()
-    logger.info(f"Inserted {len(test_cases)} testcase-annotations pairs to database.")
+    logger.info(f"Inserted {len(insertions)} testcase-annotations pairs to database. Successful: {sum(insertions)}")
 
 
 if __name__ == '__main__':