pass extra_params into test_...unset...

varun-edachali-dbx · varun-edachali-dbx · commit 9cfa35074bc7 · 2025-07-28T06:13:45.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/e2e/common/retry_test_mixins.py b/tests/e2e/common/retry_test_mixins.py
@@ -511,7 +511,7 @@ def test_retry_max_redirects_raises_too_many_redirects_exception(
     )
     @patch("databricks.sql.telemetry.telemetry_client.TelemetryClient._send_telemetry")
     def test_retry_max_redirects_unset_doesnt_redirect_forever(
-        self, mock_send_telemetry
+        self, mock_send_telemetry, extra_params
     ):
         """GIVEN the connector is configured without a custom max_redirects
         WHEN the DatabricksRetryPolicy is used

Original file line number	Diff line number	Diff line change
`@@ -511,7 +511,7 @@ def test_retry_max_redirects_raises_too_many_redirects_exception(`
`511`	`511`	`)`
`512`	`512`	`@patch("databricks.sql.telemetry.telemetry_client.TelemetryClient._send_telemetry")`
`513`	`513`	`def test_retry_max_redirects_unset_doesnt_redirect_forever(`
`514`		`- self, mock_send_telemetry`
	`514`	`+ self, mock_send_telemetry, extra_params`
`515`	`515`	`):`
`516`	`516`	`"""GIVEN the connector is configured without a custom max_redirects`
`517`	`517`	`WHEN the DatabricksRetryPolicy is used`