move the job label test into its own module

tests/functional/adapter/query_comment_test/test_job_label.py

@@ -0,0 +1,46 @@
+import pytest
+
+from google.cloud.bigquery.client import Client
+
+from dbt.tests.util import run_dbt_and_capture
+
+
+_MACRO__BQ_LABELS = """
+{% macro bq_labels() %}{
+    "system": "{{ env_var('LABEL_SYSTEM', 'my_system') }}",
+    "env_type": "{{ env_var('LABEL_ENV', 'dev') }}",
+}{% endmacro %}
+"""
+_MODEL__MY_TABLE = """
+{{ config(materialized= "table") }}
+select 1 as id
+"""
+
+
+class TestQueryCommentJobLabel:
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {"my_table.sql": _MODEL__MY_TABLE}
+
+    @pytest.fixture(scope="class")
+    def macros(self):
+        return {"bq_labels.sql": _MACRO__BQ_LABELS}
+
+    @pytest.fixture(scope="class")
+    def project_config_update(self):
+        return {
+            "query-comment": {
+                "comment": "{{ bq_labels() }}",
+                "job-label": True,
+                "append": True,
+            }
+        }
+
+    def test_query_comments_displays_as_job_labels(self, project):
+        results, logs = run_dbt_and_capture(["--debug", "run"])
+        job_id = results.results[0].adapter_response.get("job_id")
+        with project.adapter.connection_named("_test"):
+            client: Client = project.adapter.connections.get_thread_connection().handle
+            job = client.get_job(job_id=job_id)
+        assert job.labels.get("system") == "my_system"
+        assert job.labels.get("env_type") == "dev"

tests/functional/adapter/query_comment_test/test_query_comment.py

@@ -1,7 +1,3 @@
-import pytest
-
-from google.cloud.bigquery.client import Client
-
 from dbt.tests.adapter.query_comment.test_query_comment import (
     BaseQueryComments,
     BaseMacroQueryComments,
@@ -10,7 +6,6 @@
     BaseNullQueryComments,
     BaseEmptyQueryComments,
 )
-from dbt.tests.util import run_dbt_and_capture
 
 
 class TestQueryCommentsBigQuery(BaseQueryComments):
@@ -35,44 +30,3 @@ class TestNullQueryCommentsBigQuery(BaseNullQueryComments):
 
 class TestEmptyQueryCommentsBigQuery(BaseEmptyQueryComments):
     pass
-
-
-_MACRO__BQ_LABELS = """
-{% macro bq_labels() %}{
-    "system": "{{ env_var('LABEL_SYSTEM', 'my_system') }}",
-    "env_type": "{{ env_var('LABEL_ENV', 'dev') }}",
-}{% endmacro %}
-"""
-_MODEL__MY_TABLE = """
-{{ config(materialized= "table") }}
-select 1 as id
-"""
-
-
-class TestQueryCommentJobLabel:
-    @pytest.fixture(scope="class")
-    def models(self):
-        return {"my_table.sql": _MODEL__MY_TABLE}
-
-    @pytest.fixture(scope="class")
-    def macros(self):
-        return {"bq_labels.sql": _MACRO__BQ_LABELS}
-
-    @pytest.fixture(scope="class")
-    def project_config_update(self):
-        return {
-            "query-comment": {
-                "comment": "{{ bq_labels() }}",
-                "job-label": True,
-                "append": True,
-            }
-        }
-
-    def test_query_comments_displays_as_job_labels(self, project):
-        results, logs = run_dbt_and_capture(["--debug", "run"])
-        job_id = results.results[0].adapter_response.get("job_id")
-        with project.adapter.connection_named("_test"):
-            client: Client = project.adapter.connections.get_thread_connection().handle
-            job = client.get_job(job_id=job_id)
-        assert job.labels.get("system") == "my_system"
-        assert job.labels.get("env_type") == "dev"