fix(copy tests): bugs when copying multiple and table tests

Author: aarthy-dk

testgen/commands/run_profiling_bridge.py

@@ -214,7 +214,7 @@ def run_profiling_in_background(table_group_id):
         empty_cache()
         background_thread = threading.Thread(
             target=run_profiling_queries,
-            args=(table_group_id, session.auth.user_display),
+            args=(table_group_id, session.auth.user_display if session.auth else None),
         )
         background_thread.start()
     else:

testgen/common/models/test_definition.py

@@ -147,7 +147,7 @@ class TestType(Entity):
 class TestDefinition(Entity):
     __tablename__ = "test_definitions"
 
-    id: UUID = Column(postgresql.UUID(as_uuid=True), default=uuid4)
+    id: UUID = Column(postgresql.UUID(as_uuid=True), server_default=text("gen_random_uuid()"))
     cat_test_id: int = Column(BigInteger, Identity(), primary_key=True)
     table_groups_id: UUID = Column(postgresql.UUID(as_uuid=True))
     profile_run_id: UUID = Column(postgresql.UUID(as_uuid=True))

testgen/ui/views/test_definitions.py

@@ -6,7 +6,7 @@
 
 import pandas as pd
 import streamlit as st
-from sqlalchemy import asc, func, tuple_
+from sqlalchemy import and_, asc, func, or_, tuple_
 from streamlit.delta_generator import DeltaGenerator
 from streamlit_extras.no_default_selectbox import selectbox
 
@@ -1184,13 +1184,16 @@ def get_test_definitions_collision(
     target_table_group_id: str,
     target_test_suite_id: str,
 ) -> pd.DataFrame:
+    table_tests = [(item["table_name"], item["test_type"]) for item in test_definitions if item["column_name"] is None]
+    column_tests = [(item["table_name"], item["column_name"], item["test_type"]) for item in test_definitions if item["column_name"] is not None]
     results = TestDefinition.select_minimal_where(
         TestDefinition.table_groups_id == target_table_group_id,
         TestDefinition.test_suite_id == target_test_suite_id,
         TestDefinition.last_auto_gen_date.isnot(None),
-        tuple_(TestDefinition.table_name, TestDefinition.column_name, TestDefinition.test_type).in_(
-            [(item["table_name"], item["column_name"], item["test_type"]) for item in test_definitions]
-        ),
+        or_(
+            tuple_(TestDefinition.table_name, TestDefinition.column_name, TestDefinition.test_type).in_(column_tests),
+            and_(tuple_(TestDefinition.table_name, TestDefinition.test_type).in_(table_tests), TestDefinition.column_name.is_(None)),
+        )
     )
     return to_dataframe(results, TestDefinitionMinimal.columns())