fix(runs): update Running runs to Canceled on startup

Author: aarthy-dk

testgen/__main__.py

@@ -43,6 +43,7 @@
     logs,
     version_service,
 )
+from testgen.ui.queries import profiling_run_queries, test_run_queries
 from testgen.utils import plugins
 
 LOG = logging.getLogger("testgen")
@@ -606,6 +607,11 @@ def run(debug: bool):
     use_ssl = os.path.isfile(settings.SSL_CERT_FILE) and os.path.isfile(settings.SSL_KEY_FILE)
 
     patch_streamlit.patch(force=True)
+    try:
+        profiling_run_queries.cancel_all_running()
+        test_run_queries.cancel_all_running()
+    except Exception:
+        LOG.warning("Failed to cancel 'Running' profiling/test runs")
 
     try:
         app_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), "ui/app.py")

testgen/ui/queries/profiling_run_queries.py

@@ -16,3 +16,12 @@ def update_status(profile_run_id: str, status: str) -> None:
     """
     db.execute_sql(sql)
     st.cache_data.clear()
+
+
+def cancel_all_running() -> None:
+    schema: str = db.get_schema()
+    db.execute_sql(f"""
+    UPDATE {schema}.profiling_runs
+        SET status = 'Cancelled'
+        WHERE status = 'Running';
+    """)

testgen/ui/queries/test_run_queries.py

@@ -39,3 +39,12 @@ def update_status(test_run_id: str, status: str) -> None:
     """
     db.execute_sql(sql)
     st.cache_data.clear()
+
+
+def cancel_all_running() -> None:
+    schema: str = db.get_schema()
+    db.execute_sql(f"""
+    UPDATE {schema}.test_runs
+        SET status = 'Cancelled'
+        WHERE status = 'Running';
+    """)