add error handling for rendering

Author: themisvaltinos

sqlmesh/core/environment.py

@@ -262,30 +262,35 @@ def execute_environment_statements(
     end: t.Optional[TimeLike] = None,
     execution_time: t.Optional[TimeLike] = None,
 ) -> None:
-    rendered_expressions = [
-        expr
-        for statements in environment_statements
-        for expr in render_statements(
-            statements=getattr(statements, runtime_stage.value),
-            dialect=adapter.dialect,
-            default_catalog=default_catalog,
-            python_env=statements.python_env,
-            jinja_macros=statements.jinja_macros,
-            snapshots=snapshots,
-            start=start,
-            end=end,
-            execution_time=execution_time,
-            environment_naming_info=environment_naming_info,
-            runtime_stage=runtime_stage,
-            engine_adapter=adapter,
+    try:
+        rendered_expressions = [
+            expr
+            for statements in environment_statements
+            for expr in render_statements(
+                statements=getattr(statements, runtime_stage.value),
+                dialect=adapter.dialect,
+                default_catalog=default_catalog,
+                python_env=statements.python_env,
+                jinja_macros=statements.jinja_macros,
+                snapshots=snapshots,
+                start=start,
+                end=end,
+                execution_time=execution_time,
+                environment_naming_info=environment_naming_info,
+                runtime_stage=runtime_stage,
+                engine_adapter=adapter,
+            )
+        ]
+    except Exception as e:
+        raise SQLMeshError(
+            f"An error occurred during rendering of the '{runtime_stage.value}' statements:\n\n{e}"
         )
-    ]
     if rendered_expressions:
         with adapter.transaction():
             for expr in rendered_expressions:
                 try:
                     adapter.execute(expr)
                 except Exception as e:
                     raise SQLMeshError(
-                        f"An error occurred during execution of the following `{runtime_stage.value}` statement:\n\n{expr}\n\n{e}"
+                        f"An error occurred during execution of the following '{runtime_stage.value}' statement:\n\n{expr}\n\n{e}"
                     )

tests/core/test_integration.py

@@ -5531,7 +5531,7 @@ def test_environment_statements_error_handling(tmp_path: Path):
     ctx = Context(paths=[tmp_path], config=config)
 
     expected_error_message = re.escape(
-        """An error occurred during execution of the following `before_all` statement:
+        """An error occurred during execution of the following 'before_all' statement:
 
 CREATE TABLE identical_table (physical_schema_name TEXT)
 
@@ -5541,6 +5541,29 @@ def test_environment_statements_error_handling(tmp_path: Path):
     with pytest.raises(SQLMeshError, match=expected_error_message):
         ctx.plan(auto_apply=True, no_prompts=True)
 
+    after_all = [
+        "@bad_macro()",
+    ]
+
+    config = Config(
+        model_defaults=ModelDefaultsConfig(dialect="duckdb"),
+        after_all=after_all,
+    )
+    ctx = Context(paths=[tmp_path], config=config)
+
+    expected_error_message = re.escape(
+        """An error occurred during rendering of the 'after_all' statements:
+
+Failed to resolve macros for
+
+@bad_macro()
+
+Macro 'bad_macro' does not exist."""
+    )
+
+    with pytest.raises(SQLMeshError, match=expected_error_message):
+        ctx.plan(auto_apply=True, no_prompts=True)
+
 
 @time_machine.travel("2025-03-08 00:00:00 UTC")
 def test_tz(init_and_plan_context):