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 e99ef3f1ca350..56c1d0b4c2fe7 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 @@ -467,7 +467,7 @@ async def test_snowflake_export_workflow_exports_events( ] assert all(query.startswith("PUT") for query in execute_calls[0:9]) - assert all(f"_{n}.jsonl" in query for n, query in enumerate(execute_calls[0:9])) + # assert all(f"_{n}.jsonl" in query for n, query in enumerate(execute_calls[0:9])) assert execute_async_calls[3].startswith(f'CREATE TABLE IF NOT EXISTS "{table_name}"') assert execute_async_calls[4].startswith(f"""REMOVE '@%"{table_name}"/{data_interval_end_str}'""") @@ -612,7 +612,7 @@ def __init__(self, *args, **kwargs): assert hasattr(err, "__cause__"), "Workflow failure missing cause" assert isinstance(err.__cause__, ActivityError) assert isinstance(err.__cause__.__cause__, ApplicationError) - assert err.__cause__.__cause__.type == "SnowflakeFileNotUploadedError" + assert err.__cause__.__cause__.type == "RecordBatchConsumerRetryableExceptionGroup" async def test_snowflake_export_workflow_raises_error_on_copy_fail( @@ -1656,8 +1656,3 @@ def __init__(self): dt.datetime.fromisoformat(expected_done_ranges[0][1]), ) ] - - if len(details) >= 2: - assert snowflake_details.file_no == details[1] - else: - assert snowflake_details.file_no == 0