Merge branch 'aarthy/test-definition-id' into 'enterprise'

fix(test definitions): id missing for user-defined tests

See merge request dkinternal/testgen/dataops-testgen!298

Author: ci bot

testgen/common/models/test_definition.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 from typing import Literal
-from uuid import UUID
+from uuid import UUID, uuid4
 
 import streamlit as st
 from sqlalchemy import (
@@ -146,7 +146,7 @@ class TestType(Entity):
 class TestDefinition(Entity):
     __tablename__ = "test_definitions"
 
-    id: UUID = Column(postgresql.UUID(as_uuid=True))
+    id: UUID = Column(postgresql.UUID(as_uuid=True), default=uuid4)
     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/template/dbupgrade/0148_incremental_upgrade.sql

@@ -0,0 +1,6 @@
+SET SEARCH_PATH TO {SCHEMA_NAME};
+
+UPDATE test_definitions
+   SET id = gen_random_uuid()
+   WHERE id IS NULL;
+