add extra_params param

varun-edachali-dbx · varun-edachali-dbx · commit 9e086af932a2 · 2025-07-28T06:11:52.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
@@ -476,7 +476,7 @@ def test_retry_abort_close_operation_on_404(self, extra_params, caplog):
     )
     @patch("databricks.sql.telemetry.telemetry_client.TelemetryClient._send_telemetry")
     def test_retry_max_redirects_raises_too_many_redirects_exception(
-        self, mock_send_telemetry
+        self, mock_send_telemetry, extra_params
     ):
         """GIVEN the connector is configured with a custom max_redirects
         WHEN the DatabricksRetryPolicy is created

Original file line number	Diff line number	Diff line change
`@@ -476,7 +476,7 @@ def test_retry_abort_close_operation_on_404(self, extra_params, caplog):`
`476`	`476`	`)`
`477`	`477`	`@patch("databricks.sql.telemetry.telemetry_client.TelemetryClient._send_telemetry")`
`478`	`478`	`def test_retry_max_redirects_raises_too_many_redirects_exception(`
`479`		`- self, mock_send_telemetry`
	`479`	`+ self, mock_send_telemetry, extra_params`
`480`	`480`	`):`
`481`	`481`	`"""GIVEN the connector is configured with a custom max_redirects`
`482`	`482`	`WHEN the DatabricksRetryPolicy is created`