Squash Odysseus development history

tests/test_update_plan_tool.py

@@ -0,0 +1,49 @@
+"""`update_plan` — the agent writes back to the active plan (tick done / revise).
+
+Pure UI-control marker: `execute_tool_block` returns a `plan_update` payload the
+agent loop turns into a `plan_update` SSE event; the frontend replaces the stored
+plan and refreshes the docked plan window. No I/O, does not end the turn.
+"""
+import asyncio
+import json
+
+from src.agent_tools import ToolBlock, TOOL_TAGS  # import first to avoid circular
+from src.tool_execution import NO_TOOL_SECURITY_CONTEXT, execute_tool_block
+from src.tool_index import ALWAYS_AVAILABLE, BUILTIN_TOOL_DESCRIPTIONS
+from src.tool_security import is_public_blocked_tool
+
+
+def _run(content):
+    return asyncio.run(execute_tool_block(
+        ToolBlock("update_plan", content),
+        security_context=NO_TOOL_SECURITY_CONTEXT,
+    ))
+
+
+def test_valid_plan_returns_marker_and_counts():
+    plan = "- [x] step one\n- [ ] step two\n- [ ] step three"
+    desc, result = _run(json.dumps({"plan": plan}))
+    assert result.get("exit_code") == 0
+    assert result["plan_update"]["plan"] == plan
+    assert "1/3" in result["output"]   # 1 done of 3
+
+
+def test_plain_string_accepted():
+    plan = "- [ ] a\n- [x] b"
+    _, result = _run(plan)
+    assert result["plan_update"]["plan"] == plan
+
+
+def test_empty_rejected():
+    _, result = _run(json.dumps({"plan": "   "}))
+    assert "error" in result and result.get("exit_code") == 1
+
+
+def test_registered_everywhere():
+    assert "update_plan" in TOOL_TAGS
+    assert "update_plan" in ALWAYS_AVAILABLE
+    assert "update_plan" in BUILTIN_TOOL_DESCRIPTIONS
+    from src.tool_schemas import FUNCTION_TOOL_SCHEMAS
+    assert "update_plan" in {s["function"]["name"] for s in FUNCTION_TOOL_SCHEMAS}
+    # Not admin/public-gated — any user can drive their own plan.
+    assert is_public_blocked_tool("update_plan") is False