workflow

2026-07-06 23:15:30 +08:00
parent c7ae2d86ec
commit 1fb62f9d45
15 changed files with 1865 additions and 44 deletions
@@ -0,0 +1,84 @@
+from __future__ import annotations
+
+import pytest
+
+from workflow.models import (
+    BranchStep,
+    ConditionSpec,
+    PlannedGoalStep,
+    SkillInvocationStep,
+    WaitForConditionStep,
+    WorkflowDefinition,
+)
+
+
+def test_workflow_definition_accepts_heterogeneous_steps() -> None:
+    definition = WorkflowDefinition(
+        name="chat workflow",
+        entry_step_id="start",
+        steps=[
+            PlannedGoalStep("start", "open chat", next_step_id="skill"),
+            SkillInvocationStep("skill", "skill-1", {"message": "hello"}, "wait"),
+            WaitForConditionStep(
+                "wait",
+                ConditionSpec("scene_contains_text", {"text": "Done"}),
+                timeout_seconds=1,
+                poll_interval_seconds=0,
+                next_step_id="branch",
+            ),
+            BranchStep(
+                "branch",
+                ConditionSpec("world_variable_equals", {"name": "ok", "value": True}),
+                on_true="start",
+                on_false="skill",
+            ),
+        ],
+    )
+
+    assert definition.step_by_id("skill").kind == "skill_invocation"
+
+
+def test_workflow_definition_rejects_duplicate_step_id() -> None:
+    with pytest.raises(ValueError, match="duplicate"):
+        WorkflowDefinition(
+            name="bad",
+            entry_step_id="same",
+            steps=[
+                PlannedGoalStep("same", "one"),
+                PlannedGoalStep("same", "two"),
+            ],
+        )
+
+
+@pytest.mark.parametrize(
+    "definition",
+    [
+        WorkflowDefinition,
+    ],
+)
+def test_workflow_definition_rejects_dangling_references(definition) -> None:
+    with pytest.raises(ValueError, match="unknown|entry"):
+        definition(
+            name="bad next",
+            entry_step_id="start",
+            steps=[PlannedGoalStep("start", "one", next_step_id="missing")],
+        )
+    with pytest.raises(ValueError, match="unknown|entry"):
+        definition(
+            name="bad branch",
+            entry_step_id="branch",
+            steps=[
+                BranchStep(
+                    "branch",
+                    ConditionSpec("elapsed_seconds", {"seconds": 0}),
+                    on_true="missing",
+                    on_false="branch",
+                )
+            ],
+        )
+    with pytest.raises(ValueError, match="entry"):
+        definition(
+            name="bad entry",
+            entry_step_id="missing",
+            steps=[PlannedGoalStep("start", "one")],
+        )