Expand governed Dataflow editor and node library

tests/test_service.py

@@ -220,12 +220,49 @@ class DataflowServiceTests(unittest.TestCase):
         self.assertEqual(2, run.output_row_count)
         self.assertEqual(3, run.input_row_count)
         self.assertEqual(1, len(run.source_fingerprints))
+        self.assertEqual(3, response.input_row_count)
+        self.assertEqual(run.source_fingerprints, response.source_fingerprints)
         self.assertFalse(hasattr(run, "result_rows"))
         self.assertEqual(
             1,
             self.session.scalar(select(func.count()).select_from(DataflowRun)),
         )
 
+    def test_failed_preview_keeps_upstream_lineage_and_failed_node_diagnostic(self) -> None:
+        graph = sample_graph()
+        graph.nodes[0].config["rows"] = [{"id": 1, "amount": "not-a-number"}]
+        pipeline = create_pipeline(
+            self.session,
+            tenant_id="tenant-1",
+            actor_id="user-1",
+            payload=PipelineCreateRequest(
+                name="Invalid runtime value",
+                graph=graph,
+                editor_mode="graph",
+            ),
+        )
+        self.session.commit()
+
+        response = preview_pipeline(
+            self.session,
+            tenant_id="tenant-1",
+            actor_id="user-1",
+            payload=PipelinePreviewRequest(pipeline_id=pipeline.id),
+        )
+        self.session.commit()
+
+        run = self.session.scalar(select(DataflowRun).where(DataflowRun.id == response.run_id))
+        self.assertEqual("failed", response.status)
+        self.assertEqual(
+            [("source", "succeeded"), ("filter", "failed")],
+            [(item.node_id, item.status) for item in response.node_diagnostics],
+        )
+        self.assertEqual(1, run.input_row_count)
+        self.assertEqual(1, len(run.source_fingerprints))
+        self.assertEqual(1, response.input_row_count)
+        self.assertEqual(run.source_fingerprints, response.source_fingerprints)
+        self.assertEqual("preview.execution", run.diagnostics[-1]["code"])
+
 
 if __name__ == "__main__":
     unittest.main()