Use utc time in mocked data

MetaphorData · Jul 30, 2024 · c4552ca · c4552ca
1 parent 7cf4da1
commit c4552ca
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/tests/oracle/expected_query_logs.json b/tests/oracle/expected_query_logs.json
@@ -4,7 +4,7 @@
     "platform": "ORACLE",
     "queryId": "sql-id",
     "sql": "SELECT...",
-    "startTime": "2024-07-30T23:31:33+00:00",
+    "startTime": "2024-07-30T15:31:33+00:00",
     "userId": "DEV"
   }
 ]
diff --git a/tests/oracle/test_extractor.py b/tests/oracle/test_extractor.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timezone
 from unittest.mock import MagicMock, patch
 
 import pytest
@@ -123,7 +123,15 @@ def mock_connection():
     )
 
     cursor.__iter__.return_value = iter(
-        [("DEV", "SELECT...", datetime.fromtimestamp(1722353493), 10.0, "sql-id")]
+        [
+            (
+                "DEV",
+                "SELECT...",
+                datetime.fromtimestamp(1722353493, tz=timezone.utc),
+                10.0,
+                "sql-id",
+            )
+        ]
     )
 
     return connection