From de16f11ce8292c6dbf53d1afaddbe3f5bb54a77f Mon Sep 17 00:00:00 2001
From: Albrecht Degering <albrecht@degering.name>
Date: Wed, 29 Jul 2026 20:13:49 +0200
Subject: [PATCH] test(datasources): include immutable payload schema

---
 tools/checks/check-datasource-composition.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tools/checks/check-datasource-composition.py b/tools/checks/check-datasource-composition.py
index 0149de2..c30c17b 100644
--- a/tools/checks/check-datasource-composition.py
+++ b/tools/checks/check-datasource-composition.py
@@ -43,6 +43,8 @@ from govoplan_dataflow.backend.db.models import (
 from govoplan_dataflow.backend.service import create_pipeline, preview_pipeline
 from govoplan_datasources.backend.db.models import (
     DatasourceMaterializationRecord,
+    DatasourcePayloadRecord,
+    DatasourcePayloadRowRecord,
     DatasourcePublicationRecord,
     DatasourceRecord,
     DatasourceStageRecord,
@@ -62,6 +64,8 @@ def main() -> int:
         tables=[
             ConnectorTabularSource.__table__,
             DatasourceRecord.__table__,
+            DatasourcePayloadRecord.__table__,
+            DatasourcePayloadRowRecord.__table__,
             DatasourceMaterializationRecord.__table__,
             DatasourceStageRecord.__table__,
             DatasourcePublicationRecord.__table__,