File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed
Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments