run linter and update unit test mocks

dbt-labs · Nov 19, 2024 · 7c4388f · 7c4388f
1 parent 5f3a456
commit 7c4388f
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 7 deletions.
diff --git a/dbt/adapters/bigquery/clients.py b/dbt/adapters/bigquery/clients.py
@@ -45,7 +45,9 @@ def create_dataproc_job_controller_client(credentials: BigQueryCredentials) -> J
 
 
 @Retry()  # google decorator. retries on transient errors with exponential backoff
-def create_dataproc_batch_controller_client(credentials: BigQueryCredentials) -> BatchControllerClient:
+def create_dataproc_batch_controller_client(
+    credentials: BigQueryCredentials,
+) -> BatchControllerClient:
     return BatchControllerClient(
         credentials=create_google_credentials(credentials),
         client_options=ClientOptions(api_endpoint=_dataproc_endpoint(credentials)),

diff --git a/dbt/adapters/bigquery/retry.py b/dbt/adapters/bigquery/retry.py
@@ -63,7 +63,9 @@ def create_job_execution_timeout(self, fallback: Optional[float] = None) -> Opti
     def create_retry(
         self, timeout: Optional[float] = None, fallback_timeout: Optional[float] = None
     ) -> Retry:
-        return DEFAULT_RETRY.with_timeout(timeout or self.create_job_execution_timeout(fallback_timeout))
+        return DEFAULT_RETRY.with_timeout(
+            timeout or self.create_job_execution_timeout(fallback_timeout)
+        )
 
     def create_polling(
         self, timeout: Optional[float] = None, fallback_timeout: Optional[float] = None

diff --git a/tests/unit/test_bigquery_adapter.py b/tests/unit/test_bigquery_adapter.py
@@ -203,7 +203,7 @@ def get_adapter(self, target) -> BigQueryAdapter:
 
 class TestBigQueryAdapterAcquire(BaseTestBigQueryAdapter):
     @patch(
-        "dbt.adapters.bigquery.credentials._bigquery_defaults",
+        "dbt.adapters.bigquery.credentials._create_bigquery_defaults",
         return_value=("credentials", "project_id"),
     )
     @patch("dbt.adapters.bigquery.BigQueryConnectionManager.open", return_value=_bq_conn())
@@ -244,7 +244,7 @@ def test_acquire_connection_oauth_validations(self, mock_open_connection):
         mock_open_connection.assert_called_once()
 
     @patch(
-        "dbt.adapters.bigquery.credentials._bigquery_defaults",
+        "dbt.adapters.bigquery.credentials._create_bigquery_defaults",
         return_value=("credentials", "project_id"),
     )
     @patch(

diff --git a/tests/unit/test_bigquery_connection_manager.py b/tests/unit/test_bigquery_connection_manager.py
@@ -33,7 +33,7 @@ def setUp(self):
         self.connections.get_thread_connection = lambda: self.mock_connection
 
     @patch(
-        "dbt.adapters.bigquery.retry.bigquery_client",
+        "dbt.adapters.bigquery.retry.create_bigquery_client",
         return_value=Mock(google.cloud.bigquery.Client),
     )
     def test_retry_connection_reset(self, mock_client_factory):

diff --git a/tests/unit/test_configure_dataproc_batch.py b/tests/unit/test_configure_dataproc_batch.py
@@ -12,7 +12,7 @@
 # parsed credentials
 class TestConfigureDataprocBatch(BaseTestBigQueryAdapter):
     @patch(
-        "dbt.adapters.bigquery.credentials._bigquery_defaults",
+        "dbt.adapters.bigquery.credentials._create_bigquery_defaults",
         return_value=("credentials", "project_id"),
     )
     def test_update_dataproc_serverless_batch(self, mock_get_bigquery_defaults):
@@ -64,7 +64,7 @@ def to_str_values(d):
         )
 
     @patch(
-        "dbt.adapters.bigquery.credentials._bigquery_defaults",
+        "dbt.adapters.bigquery.credentials._create_bigquery_defaults",
         return_value=("credentials", "project_id"),
     )
     def test_default_dataproc_serverless_batch(self, mock_get_bigquery_defaults):