Fix some tests

Author: dgarros

backend/tests/unit/graphql/diff/test_diff_tree_query.py

@@ -1,3 +1,4 @@
+from datetime import datetime, timezone
 from unittest.mock import AsyncMock
 
 import pytest
@@ -20,6 +21,8 @@
 from infrahub.graphql.initialization import prepare_graphql_params
 from tests.helpers.graphql import graphql
 
+UTC = timezone.utc  # Required for older versions of Python
+
 ADDED_ACTION = "ADDED"
 UPDATED_ACTION = "UPDATED"
 REMOVED_ACTION = "REMOVED"
@@ -779,12 +782,24 @@ async def test_diff_tree_summary_no_changes(
     [
         pytest.param(
             {},
-            DiffSummaryCounters(num_added=2, num_updated=5, num_removed=2, from_time=Timestamp(), to_time=Timestamp()),
+            DiffSummaryCounters(
+                num_added=2,
+                num_updated=5,
+                num_removed=2,
+                from_time=Timestamp(datetime.now(UTC).isoformat()),
+                to_time=Timestamp(datetime.now(UTC).isoformat()),
+            ),
             id="no-filters",
         ),
         pytest.param(
             {"kind": {"includes": ["TestThing"]}},
-            DiffSummaryCounters(num_added=2, num_updated=1, num_removed=2, from_time=Timestamp(), to_time=Timestamp()),
+            DiffSummaryCounters(
+                num_added=2,
+                num_updated=1,
+                num_removed=2,
+                from_time=Timestamp(datetime.now(UTC).isoformat()),
+                to_time=Timestamp(datetime.now(UTC).isoformat()),
+            ),
             id="kind-includes",
         ),
     ],

backend/tests/unit/graphql/test_graphql_query.py

@@ -2048,10 +2048,9 @@ async def test_query_node_updated_at(db: InfrahubDatabase, default_branch: Branc
     assert result2.errors is None
     assert result2.data["TestPerson"]["edges"][0]["node"]["_updated_at"]
     assert result2.data["TestPerson"]["edges"][1]["node"]["_updated_at"]
-    assert (
+    assert result2.data["TestPerson"]["edges"][1]["node"]["_updated_at"] == Timestamp(
         result2.data["TestPerson"]["edges"][1]["node"]["_updated_at"]
-        == Timestamp(result2.data["TestPerson"]["edges"][1]["node"]["_updated_at"]).to_string()
-    )
+    ).to_string(with_z=False)
     assert (
         result2.data["TestPerson"]["edges"][0]["node"]["_updated_at"]
         != result2.data["TestPerson"]["edges"][1]["node"]["_updated_at"]
@@ -2125,12 +2124,9 @@ async def test_query_relationship_updated_at(db: InfrahubDatabase, default_branc
         result2.data["TestPerson"]["edges"][0]["node"]["tags"]["edges"][0]["node"]["_updated_at"]
         != result2.data["TestPerson"]["edges"][0]["node"]["tags"]["edges"][0]["properties"]["updated_at"]
     )
-    assert (
+    assert result2.data["TestPerson"]["edges"][0]["node"]["tags"]["edges"][0]["node"]["_updated_at"] == Timestamp(
         result2.data["TestPerson"]["edges"][0]["node"]["tags"]["edges"][0]["node"]["_updated_at"]
-        == Timestamp(
-            result2.data["TestPerson"]["edges"][0]["node"]["tags"]["edges"][0]["node"]["_updated_at"]
-        ).to_string()
-    )
+    ).to_string(with_z=False)
 
 
 async def test_query_attribute_node_property_source(