Fix: setting project on standalone audits (#4411)

Author: vholmer

sqlmesh/core/audit/definition.py

@@ -383,6 +383,7 @@ def load_audit(
     dialect: t.Optional[str] = None,
     default_catalog: t.Optional[str] = None,
     variables: t.Optional[t.Dict[str, t.Any]] = None,
+    project: t.Optional[str] = None,
 ) -> Audit:
     """Load an audit from a parsed SQLMesh audit file.
 
@@ -455,6 +456,8 @@ def load_audit(
             used_variables=used_variables,
         )
         extra_kwargs["default_catalog"] = default_catalog
+        if project is not None:
+            extra_kwargs["project"] = project
 
     dialect = meta_fields.pop("dialect", dialect)
     try:
@@ -482,6 +485,7 @@ def load_multiple_audits(
     dialect: t.Optional[str] = None,
     default_catalog: t.Optional[str] = None,
     variables: t.Optional[t.Dict[str, t.Any]] = None,
+    project: t.Optional[str] = None,
 ) -> t.Generator[Audit, None, None]:
     audit_block: t.List[exp.Expression] = []
     for expression in expressions:
@@ -496,6 +500,7 @@ def load_multiple_audits(
                     dialect=dialect,
                     default_catalog=default_catalog,
                     variables=variables,
+                    project=project,
                 )
                 audit_block.clear()
         audit_block.append(expression)
@@ -505,6 +510,7 @@ def load_multiple_audits(
         dialect=dialect,
         default_catalog=default_catalog,
         variables=variables,
+        project=project,
     )
 
 

sqlmesh/core/loader.py

@@ -619,6 +619,7 @@ def _load_audits(
                     dialect=self.config.model_defaults.dialect,
                     default_catalog=self.context.default_catalog,
                     variables=variables,
+                    project=self.config.project,
                 )
                 for audit in audits:
                     audits_by_name[audit.name] = audit

tests/core/test_model.py

@@ -472,6 +472,40 @@ def test_model_missing_audits(tmp_path: Path):
         )
 
 
+def test_project_is_set_in_standalone_audit(tmp_path: Path) -> None:
+    init_example_project(tmp_path, dialect="duckdb", template=ProjectTemplate.EMPTY)
+
+    db_path = str(tmp_path / "db.db")
+    db_connection = DuckDBConnectionConfig(database=db_path)
+
+    config = Config(
+        project="test",
+        gateways={"gw": GatewayConfig(connection=db_connection)},
+        model_defaults=ModelDefaultsConfig(dialect="duckdb"),
+    )
+
+    model = tmp_path / "models" / "some_model.sql"
+    model.parent.mkdir(parents=True, exist_ok=True)
+    model.write_text("MODEL (name m); SELECT 1 AS c")
+
+    audit = tmp_path / "audits" / "a_standalone_audit.sql"
+    audit.parent.mkdir(parents=True, exist_ok=True)
+    audit.write_text("AUDIT (name a, standalone true); SELECT * FROM m WHERE c <= 0")
+
+    context = Context(paths=tmp_path, config=config)
+    context.plan(no_prompts=True, auto_apply=True)
+
+    model = tmp_path / "models" / "some_model.sql"
+    model.parent.mkdir(parents=True, exist_ok=True)
+    model.write_text("MODEL (name m); SELECT 2 AS c")
+
+    assert context.fetchdf(
+        "select snapshot -> 'node' -> 'project' AS standalone_audit_project "
+        """from sqlmesh._snapshots where (snapshot -> 'node' -> 'source_type')::text = '"audit"'"""
+    ).to_dict()["standalone_audit_project"] == {0: '"test"'}
+    assert context.load().standalone_audits["a"].project == "test"
+
+
 @pytest.mark.parametrize(
     "partition_by_input, partition_by_output, output_dialect, expected_exception",
     [