From 83eb00e2158a025af2ce6fb43682575ac0eb8433 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1s=20Far=C3=ADas=20Santana?= Date: Tue, 26 Mar 2024 14:04:07 +0100 Subject: [PATCH] fix: Correct assertions in snowflake test --- .../batch_exports/test_snowflake_batch_export_workflow.py | 5 ++++- 1 file changed, 4 insertions(+), 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 index d4328768e211c..fffbb50534530 100644 --- 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):