fix: Add additional type checks in test

PostHog · Nov 21, 2024 · 975ff4f · 975ff4f
1 parent 9ef193f
commit 975ff4f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/posthog/temporal/tests/batch_exports/test_s3_batch_export_workflow.py b/posthog/temporal/tests/batch_exports/test_s3_batch_export_workflow.py
@@ -1393,7 +1393,7 @@ async def test_insert_into_s3_activity_heartbeats(
             inserted_at=part_inserted_at,
         )
 
-    heartbeat_details = []
+    heartbeat_details: list[S3HeartbeatDetails] = []
 
     def track_hearbeat_details(*details):
         """Record heartbeat details received."""
@@ -1423,6 +1423,7 @@ def track_hearbeat_details(*details):
 
     detail = heartbeat_details[-1]
 
+    assert detail.upload_state is not None
     assert len(detail.upload_state.parts) == 3
     assert len(detail.done_ranges) == 3