fix: Correct assertions in snowflake test

PostHog · Mar 26, 2024 · 83eb00e · 83eb00e
1 parent fafa0eb
commit 83eb00e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/posthog/temporal/tests/batch_exports/test_snowflake_batch_export_workflow.py b/posthog/temporal/tests/batch_exports/test_snowflake_batch_export_workflow.py
@@ -697,7 +697,8 @@ async def insert_into_snowflake_activity_mocked(_: SnowflakeInsertInputs) -> str
     run = runs[0]
     assert run.status == "FailedRetryable"
     assert run.latest_error == "ValueError: A useful error message"
-    assert run.records_completed == 0
+    assert run.records_completed is None
+    assert run.records_total_count == 1
 
 
 async def test_snowflake_export_workflow_handles_insert_activity_non_retryable_errors(ateam, snowflake_batch_export):
@@ -744,6 +745,8 @@ class ForbiddenError(Exception):
     run = runs[0]
     assert run.status == "Failed"
     assert run.latest_error == "ForbiddenError: A useful error message"
+    assert run.records_completed is None
+    assert run.records_total_count == 1
 
 
 async def test_snowflake_export_workflow_handles_cancellation_mocked(ateam, snowflake_batch_export):