fix: errors on first-time setup

Author: hrideshmg

install/sample_db.py

@@ -11,12 +11,15 @@
 def run():
     from database import create_session
     from mod_auth.models import User
+    from mod_customized.models import CustomizedTest
     from mod_home.models import CCExtractorVersion, GeneralData
     from mod_regression.models import (Category, InputType, OutputType,
                                        RegressionTest, RegressionTestOutput)
     from mod_sample.models import Sample
+    from mod_test.models import Test
     from mod_upload.models import Upload
 
+
     db = create_session(sys.argv[1])
 
     entries = []

mod_ci/controllers.py

@@ -1348,8 +1348,8 @@ def progress_type_request(log, test, test_id, request) -> bool:
 
             for p in times:
                 parts = p.time.split(',')
-                start = datetime.datetime.strptime(parts[0], '%Y-%m-%d %H:%M:%S.%f')
-                end = datetime.datetime.strptime(parts[-1], '%Y-%m-%d %H:%M:%S.%f')
+                start = datetime.datetime.strptime(parts[0], '%Y-%m-%d %H:%M:%S')
+                end = datetime.datetime.strptime(parts[-1], '%Y-%m-%d %H:%M:%S')
                 total_time += int((end - start).total_seconds())
 
             if len(times) != 0:
@@ -1377,7 +1377,7 @@ def progress_type_request(log, test, test_id, request) -> bool:
 
             last_running_test = end_time - start_time
             updated_average = updated_average + last_running_test.total_seconds()
-            current_average.value = updated_average // number_test
+            current_average.value = 0 if number_test == 0 else updated_average // number_test
             g.db.commit()
             log.info(f'average time updated to {str(current_average.value)}')
 

mod_test/controllers.py

@@ -150,7 +150,7 @@ def get_data_for_test(test, title=None) -> Dict[str, Any]:
         prep_average_key = 'avg_prep_time_' + test.platform.value
         average_prep_time = int(float(GeneralData.query.filter(GeneralData.key == prep_average_key).first().value))
 
-        test_progress_last_entry = g.db.query(func.max(TestProgress.test_id)).first()
+        test_progress_last_entry = g.db.query(func.max(TestProgress.test_id)).first() or 0
         queued_gcp_instance = g.db.query(GcpInstance.test_id).filter(GcpInstance.test_id < test.id).subquery()
         queued_gcp_instance_entries = g.db.query(Test.id).filter(
             and_(Test.id.in_(queued_gcp_instance), Test.platform == test.platform)