fix debug task for bigquery connections

dbt-labs · May 12, 2021 · c370d79 · c370d79
1 parent d8c261f
commit c370d79
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/plugins/bigquery/dbt/adapters/bigquery/connections.py b/plugins/bigquery/dbt/adapters/bigquery/connections.py
@@ -309,7 +309,7 @@ def raw_execute(self, sql, fetch=False, *, use_legacy_sql=False):
 
         logger.debug('On {}: {}', conn.name, sql)
 
-        if self.profile.query_comment.job_label:
+        if self.profile.query_comment and self.profile.query_comment.job_label:
             query_comment = self.query_header.comment.query_comment
             labels = self._labels_from_query_comment(query_comment)
         else:

diff --git a/test/integration/049_dbt_debug_test/test_debug.py b/test/integration/049_dbt_debug_test/test_debug.py
@@ -64,6 +64,11 @@ def test_postgres_ok(self):
         self.run_dbt(['debug'])
         self.assertNotIn('ERROR', self.capsys.readouterr().out)
 
+    @use_profile('bigquery')
+    def test_bigquery_ok(self):
+        self.run_dbt(['debug'])
+        self.assertNotIn('ERROR', self.capsys.readouterr().out)
+
     @use_profile('postgres')
     def test_postgres_nopass(self):
         self.run_dbt(['debug', '--target', 'nopass'], expect_pass=False)