PR Feedback 2

Author: VaggelisD

sqlmesh/core/console.py

@@ -3204,10 +3204,12 @@ def update_snapshot_evaluation_progress(
         num_audits_failed: int,
         audit_only: bool = False,
     ) -> None:
+        view_name, loaded_batches = self.evaluation_batch_progress[snapshot.snapshot_id]
+
         if audit_only:
+            print(f"Completed Auditing {view_name}")
             return
 
-        view_name, loaded_batches = self.evaluation_batch_progress[snapshot.snapshot_id]
         total_batches = self.evaluation_model_batch_sizes[snapshot]
 
         loaded_batches += 1

tests/core/test_context.py

@@ -5,7 +5,6 @@
 from datetime import date, timedelta
 from tempfile import TemporaryDirectory
 from unittest.mock import PropertyMock, call, patch
-from IPython.utils.capture import capture_output
 
 import time_machine
 import pytest
@@ -2072,42 +2071,3 @@ def test_audit():
     context.plan(no_prompts=True, auto_apply=True)
 
     assert context.audit(models=["dummy"], start="2020-01-01", end="2020-01-01") is True
-
-
-@use_terminal_console
-def test_audits_running_on_metadata_changes(tmp_path: pathlib.Path):
-    def setup_senario(model_before: str, model_after: str):
-        models_dir = pathlib.Path("models")
-        create_temp_file(tmp_path, pathlib.Path(models_dir, "test.sql"), model_before)
-
-        # Create first snapshot
-        context = Context(paths=tmp_path, config=Config())
-        context.plan("prod", no_prompts=True, auto_apply=True)
-
-        # Create second (metadata) snapshot
-        create_temp_file(tmp_path, pathlib.Path(models_dir, "test.sql"), model_after)
-        context.load()
-
-        with capture_output() as output:
-            with pytest.raises(PlanError):
-                context.plan("prod", no_prompts=True, auto_apply=True)
-
-        assert 'Failed models\n\n  "model"' in output.stdout
-
-        return output
-
-    # Ensure incorrect audits (bad data, incorrect definition etc) are evaluated immediately
-    output = setup_senario(
-        "MODEL (name model); SELECT NULL AS col",
-        "MODEL (name model, audits (not_null(columns=[col]))); SELECT NULL AS col",
-    )
-    assert "'not_null' audit error: 1 row failed" in output.stdout
-
-    output = setup_senario(
-        "MODEL (name model); SELECT NULL AS col",
-        "MODEL (name model, audits (not_null(columns=[this_col_does_not_exist]))); SELECT NULL AS col",
-    )
-    assert (
-        'Binder Error: Referenced column "this_col_does_not_exist" not found in \nFROM clause!'
-        in output.stdout
-    )

tests/core/test_integration.py

@@ -19,6 +19,8 @@
 from sqlglot import exp
 from sqlglot.expressions import DataType
 import re
+from IPython.utils.capture import capture_output
+
 
 from sqlmesh import CustomMaterialization
 from sqlmesh.cli.example_project import init_example_project
@@ -62,10 +64,11 @@
     SnapshotTableInfo,
 )
 from sqlmesh.utils.date import TimeLike, now, to_date, to_datetime, to_timestamp
-from sqlmesh.utils.errors import NoChangesPlanError, SQLMeshError
+from sqlmesh.utils.errors import NoChangesPlanError, SQLMeshError, PlanError
 from sqlmesh.utils.pydantic import validate_string
 from tests.conftest import DuckDBMetadata, SushiDataValidator
 from tests.utils.test_helpers import use_terminal_console
+from tests.utils.test_filesystem import create_temp_file
 
 if t.TYPE_CHECKING:
     from sqlmesh import QueryOrDF
@@ -5922,3 +5925,42 @@ def setup_scenario():
 
     # - Check that the environment record does not exist in the state sync anymore
     assert not ctx.state_sync.get_environment("dev")
+
+
+@use_terminal_console
+def test_audits_running_on_metadata_changes(tmp_path: Path):
+    def setup_senario(model_before: str, model_after: str):
+        models_dir = Path("models")
+        create_temp_file(tmp_path, models_dir / "test.sql", model_before)
+
+        # Create first snapshot
+        context = Context(paths=tmp_path, config=Config())
+        context.plan("prod", no_prompts=True, auto_apply=True)
+
+        # Create second (metadata) snapshot
+        create_temp_file(tmp_path, models_dir / "test.sql", model_after)
+        context.load()
+
+        with capture_output() as output:
+            with pytest.raises(PlanError):
+                context.plan("prod", no_prompts=True, auto_apply=True)
+
+        assert 'Failed models\n\n  "model"' in output.stdout
+
+        return output
+
+    # Ensure incorrect audits (bad data, incorrect definition etc) are evaluated immediately
+    output = setup_senario(
+        "MODEL (name model); SELECT NULL AS col",
+        "MODEL (name model, audits (not_null(columns=[col]))); SELECT NULL AS col",
+    )
+    assert "'not_null' audit error: 1 row failed" in output.stdout
+
+    output = setup_senario(
+        "MODEL (name model); SELECT NULL AS col",
+        "MODEL (name model, audits (not_null(columns=[this_col_does_not_exist]))); SELECT NULL AS col",
+    )
+    assert (
+        'Binder Error: Referenced column "this_col_does_not_exist" not found in \nFROM clause!'
+        in output.stdout
+    )