From 916f3ea18c8604af4538ab8bd7526943bfd215e8 Mon Sep 17 00:00:00 2001 From: Maxime Armstrong Date: Fri, 8 Nov 2024 14:23:42 -0500 Subject: [PATCH] Fix tests --- .../experimental/test_translator.py | 33 +++++++++---------- 1 file changed, 16 insertions(+), 17 deletions(-) diff --git a/python_modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py b/python_modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py index 3c35eb47def28..5c1cf535e4e5c 100644 --- a/python_modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py +++ b/python_modules/libraries/dagster-fivetran/dagster_fivetran_tests/experimental/test_translator.py @@ -5,26 +5,25 @@ def test_fivetran_workspace_data_to_fivetran_connector_table_props_data( - workspace_data_api_mocks_fn: Callable, + fetch_workspace_data_api_mocks: Callable, ) -> None: api_key = uuid.uuid4().hex api_secret = uuid.uuid4().hex resource = FivetranWorkspace(api_key=api_key, api_secret=api_secret) - with workspace_data_api_mocks_fn(include_sync_endpoints=False): - actual_workspace_data = resource.fetch_fivetran_workspace_data() - table_props_data = actual_workspace_data.to_fivetran_connector_table_props_data() - assert len(table_props_data) == 4 - assert ( - table_props_data[0].table == "schema_name_in_destination_1.table_name_in_destination_1" - ) - assert ( - table_props_data[1].table == "schema_name_in_destination_1.table_name_in_destination_2" - ) - assert ( - table_props_data[2].table == "schema_name_in_destination_2.table_name_in_destination_1" - ) - assert ( - table_props_data[3].table == "schema_name_in_destination_2.table_name_in_destination_2" - ) + actual_workspace_data = resource.fetch_fivetran_workspace_data() + table_props_data = actual_workspace_data.to_fivetran_connector_table_props_data() + assert len(table_props_data) == 4 + assert ( + table_props_data[0].table == "schema_name_in_destination_1.table_name_in_destination_1" + ) + assert ( + table_props_data[1].table == "schema_name_in_destination_1.table_name_in_destination_2" + ) + assert ( + table_props_data[2].table == "schema_name_in_destination_2.table_name_in_destination_1" + ) + assert ( + table_props_data[3].table == "schema_name_in_destination_2.table_name_in_destination_2" + )