Skip to content

Commit

Permalink
fix: Update tests as we no longer have file_no
Browse files Browse the repository at this point in the history
  • Loading branch information
tomasfarias committed Dec 13, 2024
1 parent b9795bd commit 2bc6960
Showing 1 changed file with 2 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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}'""")
Expand Down Expand Up @@ -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(
Expand Down Expand Up @@ -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

0 comments on commit 2bc6960

Please sign in to comment.