Skip to content

Commit e2a50e9

Browse files
NO-SNOW: Fix proxies test
1 parent 918f8ae commit e2a50e9

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

test/unit/test_proxies.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,11 @@ def _setup_backend_storage_mappings(
236236
):
237237
"""Setup backend, storage, and proxy mappings for large queries."""
238238
_set_mappings_for_common_backend(target_wm, wiremock_generic_mappings_dir)
239-
_set_mappings_for_query_and_chunks(target_wm, storage_wm, wiremock_mapping_dir)
239+
_set_mappings_for_query_and_chunks(
240+
target_wm,
241+
wiremock_mapping_dir,
242+
storage_or_target_wm=storage_wm,
243+
)
240244

241245
proxy_wm.add_mapping(
242246
{
@@ -789,7 +793,10 @@ def test_connection_params_proxy_take_precedence_over_env_vars(
789793

790794
# Setup backend mappings for large query with multiple chunks
791795
_set_mappings_for_common_backend(target_wm, wiremock_generic_mappings_dir)
792-
_set_mappings_for_query_and_chunks(target_wm, wiremock_mapping_dir)
796+
_set_mappings_for_query_and_chunks(
797+
target_wm,
798+
wiremock_mapping_dir,
799+
)
793800

794801
# Set HTTP_PROXY env var to point to proxy2
795802
set_proxy_env_vars, _ = proxy_env_vars

0 commit comments

Comments
 (0)