.

Author: sentrivana

tests/integrations/asyncpg/test_asyncpg.py

@@ -279,14 +279,24 @@ async def test_cursor(sentry_init, capture_events) -> None:
             "message": "INSERT INTO users(name, password, dob) VALUES($1, $2, $3)",
             "type": "default",
         },
-        {"category": "query", "data": {}, "message": "BEGIN;", "type": "default"},
+        {
+            "category": "query",
+            "data": {},
+            "message": "BEGIN;",
+            "type": "default",
+        },
         {
             "category": "query",
             "data": {},
             "message": "SELECT * FROM users WHERE dob > $1",
             "type": "default",
         },
-        {"category": "query", "data": {}, "message": "COMMIT;", "type": "default"},
+        {
+            "category": "query",
+            "data": {},
+            "message": "COMMIT;",
+            "type": "default",
+        },
     ]
 
 
@@ -337,14 +347,24 @@ async def test_cursor_manual(sentry_init, capture_events) -> None:
             "message": "INSERT INTO users(name, password, dob) VALUES($1, $2, $3)",
             "type": "default",
         },
-        {"category": "query", "data": {}, "message": "BEGIN;", "type": "default"},
+        {
+            "category": "query",
+            "data": {},
+            "message": "BEGIN;",
+            "type": "default",
+        },
         {
             "category": "query",
             "data": {},
             "message": "SELECT * FROM users WHERE dob > $1",
             "type": "default",
         },
-        {"category": "query", "data": {}, "message": "COMMIT;", "type": "default"},
+        {
+            "category": "query",
+            "data": {},
+            "message": "COMMIT;",
+            "type": "default",
+        },
     ]
 
 
@@ -654,9 +674,11 @@ async def test_no_query_source_if_duration_too_short(sentry_init, capture_events
 
         @contextmanager
         def fake_record_sql_queries(*args, **kwargs):
-            with freeze_time(datetime(2024, 1, 1, microsecond=0)):
+            with freeze_time(datetime.datetime(2024, 1, 1, microsecond=0)):
                 with record_sql_queries(*args, **kwargs) as span:
-                    freezer = freeze_time(datetime(2024, 1, 1, microsecond=99999))
+                    freezer = freeze_time(
+                        datetime.datetime(2024, 1, 1, microsecond=99999)
+                    )
                     freezer.start()
 
                 freezer.stop()
@@ -702,9 +724,11 @@ async def test_query_source_if_duration_over_threshold(sentry_init, capture_even
 
         @contextmanager
         def fake_record_sql_queries(*args, **kwargs):
-            with freeze_time(datetime(2024, 1, 1, microsecond=0)):
+            with freeze_time(datetime.datetime(2024, 1, 1, microsecond=0)):
                 with record_sql_queries(*args, **kwargs) as span:
-                    freezer = freeze_time(datetime(2024, 1, 1, microsecond=100001))
+                    freezer = freeze_time(
+                        datetime.datetime(2024, 1, 1, microsecond=100001)
+                    )
                     freezer.start()
 
                 freezer.stop()