group overrides from dbt 1.9.4 and above

Author: arbiv

integration_tests/tests/test_dbt_artifacts/test_groups.py

@@ -3,7 +3,6 @@
 Covers models, tests, seeds, and snapshots group assignment and artifact table correctness.
 """
 import contextlib
-import subprocess
 import uuid
 
 import pytest
@@ -46,26 +45,6 @@ def _normalize_empty(val):
     return val if val not in (None, "") else None
 
 
-def print_dbt_log_tail(n=40):
-    try:
-        output = subprocess.check_output(["tail", f"-n{n}", "logs/dbt.log"], text=True)
-        print("\n--- dbt log tail ---\n", output)
-    except Exception as e:
-        print(f"Could not read dbt log: {e}")
-
-
-def run_dbt_and_check(cmd, *args, **kwargs):
-    result = cmd(*args, **kwargs)
-    print_dbt_log_tail()  # Always print log tail
-    # If result is a dict or object, check for success
-    if hasattr(result, "success") and not result.success:
-        assert False, f"dbt command failed: {result}"
-    elif isinstance(result, dict) and not result.get("success", True):
-        assert False, f"dbt command failed: {result}"
-    # If result is None, rely on downstream assertions
-    return result
-
-
 @pytest.mark.parametrize(
     "group_config, expected_groups, test_name",
     [
@@ -161,7 +140,7 @@ def test_dbt_groups_artifact_parametrized(
         name=f"groups_test_{test_name}_{unique_id}.yml",
     ):
         dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-        run_dbt_and_check(dbt_project.dbt_runner.run)
+        dbt_project.dbt_runner.run()
         dbt_project.assert_table_exists("dbt_groups")
     for group_name, owner_name, owner_email in patched_expected_groups:
         group = _get_group_from_table(dbt_project, group_name)
@@ -220,7 +199,7 @@ def test_model_group_attribute(dbt_project: DbtProject, tmp_path):
         dbt_model_path.write_text(model_sql)
         try:
             dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-            run_dbt_and_check(dbt_project.dbt_runner.run, select=model_name)
+            dbt_project.dbt_runner.run(select=model_name)
             dbt_project.assert_table_exists("dbt_models")
             models = dbt_project.read_table(
                 "dbt_models", where=f"name = '{model_name}'", raise_if_empty=True
@@ -280,7 +259,7 @@ def test_test_group_attribute(dbt_project: DbtProject, tmp_path):
         dbt_model_path.write_text(model_sql)
         try:
             dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-            run_dbt_and_check(dbt_project.dbt_runner.run, select=model_name)
+            dbt_project.dbt_runner.run(select=model_name)
             dbt_project.assert_table_exists("dbt_tests")
             tests = dbt_project.read_table(
                 "dbt_tests", where="name LIKE 'unique%'", raise_if_empty=True
@@ -295,6 +274,7 @@ def test_test_group_attribute(dbt_project: DbtProject, tmp_path):
                 dbt_model_path.unlink()
 
 
+@pytest.mark.requires_dbt_version("1.9.4")
 def test_test_override_group(dbt_project: DbtProject, tmp_path):
     """
     Test that a singular test defined in schema.yml, which belongs to a model with a group, but also has a config: section with another group,
@@ -344,7 +324,7 @@ def test_test_override_group(dbt_project: DbtProject, tmp_path):
         dbt_model_path.write_text(model_sql)
         try:
             dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-            run_dbt_and_check(dbt_project.dbt_runner.run, select=model_name)
+            dbt_project.dbt_runner.run(select=model_name)
             dbt_project.assert_table_exists("dbt_tests")
             tests = dbt_project.read_table(
                 "dbt_tests", where="name LIKE 'unique%'", raise_if_empty=True
@@ -411,7 +391,7 @@ def test_seed_group_attribute(dbt_project: DbtProject, tmp_path):
             dbt_seed_path.write_text(seed_csv)
             # Run dbt seed
             dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-            run_dbt_and_check(dbt_project.dbt_runner.seed, select=seed_name)
+            dbt_project.dbt_runner.seed(select=seed_name)
             dbt_project.assert_table_exists("dbt_seeds")
             seeds = dbt_project.read_table(
                 "dbt_seeds", where=f"name = '{seed_name}'", raise_if_empty=True
@@ -481,7 +461,7 @@ def test_snapshot_group_attribute(dbt_project: DbtProject, tmp_path):
             dbt_snapshot_path.write_text(snapshot_sql)
             # Run dbt snapshot (runs all snapshots, as selecting is not supported by the runner)
             dbt_project.dbt_runner.vars["disable_dbt_artifacts_autoupload"] = False
-            run_dbt_and_check(dbt_project.dbt_runner.run)
+            dbt_project.dbt_runner.run()
             dbt_project.assert_table_exists("dbt_snapshots")
             snapshots = dbt_project.read_table(
                 "dbt_snapshots", where=f"name = '{snapshot_name}'", raise_if_empty=False