fixed broken tests

Author: qcdyx

functions-python/tasks_executor/src/tasks/licenses/populate_licenses.py

@@ -18,8 +18,10 @@
 # 4. Supports a 'dry_run' mode, which simulates the process and logs intended
 #    actions without committing any changes to the database.
 # 5. Includes error handling for network issues and database transactions.
+from datetime import datetime
 import logging
 
+from pytz import timezone
 import requests
 from shared.database.database import with_db_session
 from shared.database_gen.sqlacodegen_models import License, Rule
@@ -86,8 +88,11 @@ def populate_licenses_task(dry_run, db_session):
                 license_object = db_session.get(License, license_id)
                 if not license_object:
                     license_object = License(id=license_id)
+                    license_object.created_at = datetime.now(timezone.utc)
+
                 license_object.is_spdx = is_spdx
                 license_object.name = spdx_data.get("name")
+                license_object.updated_at = datetime.now(timezone.utc)
                 cross_ref_list = spdx_data.get("crossRef")
                 if (
                     cross_ref_list

functions-python/tasks_executor/tests/tasks/populate_licenses_and_rules/test_populate_licenses.py

@@ -4,6 +4,7 @@
 import requests
 from sqlalchemy.dialects.postgresql import JSONB
 from sqlalchemy.ext.compiler import compiles
+from sqlalchemy.sql.functions import now
 from shared.database.database import Session
 from shared.database_gen.sqlacodegen_models import License, Rule
 from tasks.licenses.populate_licenses import populate_licenses_task
@@ -17,6 +18,13 @@ def compile_jsonb_for_sqlite(element, compiler, **kw):
     return "TEXT"
 
 
+# This compilation rule translates the PostgreSQL-specific `now()` function into
+# the SQLite-compatible `CURRENT_TIMESTAMP` function during test schema creation.
+@compiles(now, "sqlite")
+def compile_now_for_sqlite(element, compiler, **kw):
+    return "CURRENT_TIMESTAMP"
+
+
 # Mock data for GitHub API responses
 MOCK_LICENSE_LIST = [
     {