diff --git a/databricks/sdk/mixins/files.py b/databricks/sdk/mixins/files.py index 2cdaf3532..fdf08a839 100644 --- a/databricks/sdk/mixins/files.py +++ b/databricks/sdk/mixins/files.py @@ -1337,6 +1337,7 @@ def extended_is_retryable(e: BaseException) -> Optional[str]: # where we believe request didn't reach the server is_retryable=extended_is_retryable, before_retry=before_retry, + clock=self._config.clock, )(delegate)() def _open_download_stream( diff --git a/tests/conftest.py b/tests/conftest.py index 9dddc16a2..cb2efc0a2 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -8,6 +8,7 @@ from databricks.sdk.core import Config from databricks.sdk.credentials_provider import credentials_strategy +from .clock import FakeClock from .integration.conftest import restorable_env # type: ignore @@ -18,7 +19,7 @@ def noop_credentials(_: any): @pytest.fixture def config(): - return Config(host="http://localhost", credentials_strategy=noop_credentials) + return Config(host="http://localhost", credentials_strategy=noop_credentials, clock=FakeClock()) @pytest.fixture