Standardize the adapter for core debug changes. (#754)

* Standardize the adapter for core debug changes. * add changelog * Tweak fields --------- Co-authored-by: Mila Page <versusfacit@users.noreply.github.com>
dbt-labs · Jun 7, 2023 · aa7bea8 · aa7bea8
1 parent 389acee
commit aa7bea8
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/.changes/unreleased/Features-20230604-034603.yaml b/.changes/unreleased/Features-20230604-034603.yaml
@@ -0,0 +1,6 @@
+kind: Features
+body: Standardize the _connection_keys and debug_query for `dbt debug`.
+time: 2023-06-04T03:46:03.065575-07:00
+custom:
+  Author: versusfacit
+  Issue: PR754
diff --git a/dbt/adapters/bigquery/connections.py b/dbt/adapters/bigquery/connections.py
@@ -169,17 +169,28 @@ def _connection_keys(self):
         return (
             "method",
             "database",
+            "execution_project",
             "schema",
             "location",
             "priority",
-            "timeout_seconds",
             "maximum_bytes_billed",
-            "execution_project",
+            "impersonate_service_account",
             "job_retry_deadline_seconds",
             "job_retries",
             "job_creation_timeout_seconds",
             "job_execution_timeout_seconds",
+            "keyfile",
+            "keyfile_json",
+            "timeout_seconds",
+            "token",
+            "refresh_token",
+            "client_id",
+            "client_secret",
+            "token_uri",
+            "dataproc_region",
+            "dataproc_cluster_name",
             "gcs_bucket",
+            "dataproc_batch",
         )
 
     @classmethod

diff --git a/dbt/adapters/bigquery/impl.py b/dbt/adapters/bigquery/impl.py
@@ -978,3 +978,7 @@ def render_model_constraint(cls, constraint: ModelLevelConstraint) -> Optional[s
             return f"{c} not enforced" if c else None
 
         return c
+
+    def debug_query(self):
+        """Override for DebugTask method"""
+        self.execute("select 1 as id")