Fix: make sure builtin dbt macros can be referenced with the 'context.' prefix (#5308)

Author: treysp

sqlmesh/dbt/builtin.py

@@ -545,6 +545,7 @@ def create_builtin_globals(
 
     builtin_globals["run_started_at"] = jinja_globals.get("execution_dt") or now()
     builtin_globals["dbt"] = AttributeDict(builtin_globals)
+    builtin_globals["context"] = builtin_globals["dbt"]
 
     return {**builtin_globals, **jinja_globals}
 

tests/dbt/test_transformation.py

@@ -1435,6 +1435,13 @@ def test_flags(sushi_test_project: Project):
     assert context.render("{{ flags.WHICH }}") == "parse"
 
 
+@pytest.mark.xdist_group("dbt_manifest")
+def test_context_namespace(sushi_test_project: Project):
+    context = sushi_test_project.context
+
+    assert context.render("{{ context.flags.FULL_REFRESH }}") == "None"
+
+
 @pytest.mark.xdist_group("dbt_manifest")
 def test_relation(sushi_test_project: Project):
     context = sushi_test_project.context