Bound RC10 persistence audit queries

2026-07-18 07:04:59 +02:00
parent 7b96037853
commit ba84b01692
2 changed files with 41 additions and 16 deletions
@@ -213,8 +213,13 @@ def test_source_family_report_covers_national_regional_and_maritime(tmp_path: Pa
            return self.values

    class Session:
-        def query(self, model):
-            return Query(rows[model])
+        def query(self, model, *_fields):
+            if model in rows:
+                return Query(rows[model])
+            owner = getattr(model, "class_", None)
+            if owner in rows:
+                return Query(rows[owner])
+            raise AssertionError(f"Unexpected query entity: {model!r}")

    state = audit.collect_database_state(Session(), tmp_path)