workflow

tests/test_workflow_store.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+from workflow.models import PlannedGoalStep, WorkflowDefinition, WorkflowStepResult
+from workflow.store import WorkflowStore
+
+
+def _definition() -> WorkflowDefinition:
+    return WorkflowDefinition(
+        name="linear",
+        entry_step_id="first",
+        steps=[
+            PlannedGoalStep("first", "one", next_step_id="second"),
+            PlannedGoalStep("second", "two"),
+        ],
+    )
+
+
+def test_workflow_store_round_trips_definition_and_run(tmp_path) -> None:
+    store = WorkflowStore(tmp_path / "workflows.sqlite3")
+    definition = _definition()
+
+    store.save_definition(definition)
+    run = store.create_run(
+        definition.id,
+        {"contact": "Zhang San"},
+        device_id="phone",
+    )
+    loaded = store.get_run(run.id)
+
+    assert store.get_definition(definition.id) == definition
+    assert loaded is not None
+    assert loaded.status == "running"
+    assert loaded.current_step_id == "first"
+    assert loaded.variables == {"contact": "Zhang San"}
+    assert loaded.device_id == "phone"
+
+
+def test_workflow_store_appends_step_results_in_order(tmp_path) -> None:
+    store = WorkflowStore(tmp_path / "workflows.sqlite3")
+    definition = _definition()
+    store.save_definition(definition)
+    run = store.create_run(definition.id, {})
+
+    store.append_step_result(run.id, WorkflowStepResult("first", "planned_goal", True))
+    store.append_step_result(run.id, WorkflowStepResult("second", "planned_goal", True))
+
+    loaded = store.get_run(run.id)
+    assert loaded is not None
+    assert [result.step_id for result in loaded.step_results] == ["first", "second"]
+
+
+def test_workflow_store_survives_reopen(tmp_path) -> None:
+    db_path = tmp_path / "workflows.sqlite3"
+    first_store = WorkflowStore(db_path)
+    definition = _definition()
+    first_store.save_definition(definition)
+    run = first_store.create_run(definition.id, {"x": 1})
+    first_store.append_step_result(run.id, WorkflowStepResult("first", "planned_goal", True))
+    first_store.update_run(run.id, current_step_id="second", variables={"x": 2})
+
+    reopened = WorkflowStore(db_path)
+    loaded = reopened.get_run(run.id)
+
+    assert loaded is not None
+    assert loaded.current_step_id == "second"
+    assert loaded.variables == {"x": 2}
+    assert [result.step_id for result in loaded.step_results] == ["first"]