mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
Completes the note(scope='handoff') write-path (WIP 23e6ee57): every role
with a dedicated note section is now obligated to populate it, the same way
journals are obligated.
Obligations (foundation.policy.tracing):
- DEV_NOTES / PR_REVIEWER_NOTES / QUICK_CONTEXT_MIN_CHARS requirements +
checkers, wired onto i_am_done (dev_notes), delegate (quick_context), and
pr_pass / pr_fail / post_pr_review (pr_reviewer_notes).
- Fixes a latent bug: the docs-notes checker read dev_notes instead of
doc_notes (the documenter's section); the i_documented shim now feeds
doc_notes to match.
Auditor: a session-scoped note obligation on i_am_idle — the auditor owns no
delivery task and has no delivery verb, so it must have recorded an
observation within the window before going idle (JournalService.has_recent_entry).
Write-then-gate: persisted sections (dev_notes / quick_context) are
pre-written by the agent's note(scope='handoff') before the gated verb;
argument-borne sections (doc_notes / pr_reviewer_notes) are checked through a
SimpleNamespace shim, the same pattern qa_notes already uses.
Config: dev/pr_reviewer/quick_context min-chars (40/40/30), panel-tunable.
Plus per-gap remediation hints and full coverage (write-path routing,
ownership, validation->remediation, each obligation, the doc_notes fix).
Full make-quality green: 9777 passed, 95.6% coverage.
328 lines
11 KiB
Python
328 lines
11 KiB
Python
"""End-to-end test: task pending -> awaiting_ceo_approval through the gateway.
|
|
|
|
Uses a minimal FastAPI app that includes all six v1 routers (no lifespan / no DB).
|
|
Choreographer and ContentActions are replaced by stateful async mocks that track
|
|
the simulated task lifecycle through every role hand-off.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
from roboco.api.deps import get_choreographer, get_content_actions
|
|
from roboco.api.routes.v1 import do as do_module
|
|
from roboco.api.routes.v1 import flow_cell_pm as flow_cell_pm_module
|
|
from roboco.api.routes.v1 import flow_dev as flow_dev_module
|
|
from roboco.api.routes.v1 import flow_doc as flow_doc_module
|
|
from roboco.api.routes.v1 import flow_main_pm as flow_main_pm_module
|
|
from roboco.api.routes.v1 import flow_qa as flow_qa_module
|
|
from roboco.services.gateway.envelope import Envelope
|
|
|
|
_HTTP_200 = 200
|
|
_PR_NUMBER = 8
|
|
|
|
# Shared mutable state threaded through all mock methods.
|
|
_State = dict[str, object]
|
|
|
|
|
|
def _build_app() -> FastAPI:
|
|
"""Minimal FastAPI app with all v1 routers — no lifespan, no DB."""
|
|
app = FastAPI()
|
|
app.include_router(flow_dev_module.router)
|
|
app.include_router(flow_qa_module.router)
|
|
app.include_router(flow_doc_module.router)
|
|
app.include_router(flow_cell_pm_module.router)
|
|
app.include_router(flow_main_pm_module.router)
|
|
app.include_router(do_module.router)
|
|
return app
|
|
|
|
|
|
class _MockChoreographer:
|
|
"""Stateful stub for Choreographer; methods advance shared lifecycle state."""
|
|
|
|
def __init__(self, state: _State) -> None:
|
|
self._state = state
|
|
|
|
async def give_me_work(self, _agent_id: object) -> Envelope:
|
|
return Envelope.ok(
|
|
status=str(self._state["task_status"]),
|
|
task_id=str(uuid4()),
|
|
next="claim it",
|
|
)
|
|
|
|
async def i_will_work_on(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
_plan: object = None,
|
|
**_kwargs: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "in_progress"
|
|
return Envelope.ok(
|
|
status="in_progress",
|
|
task_id=str(uuid4()),
|
|
next="commit + i_am_done",
|
|
)
|
|
|
|
async def i_am_done(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
_notes: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "awaiting_qa"
|
|
return Envelope.ok(
|
|
status="awaiting_qa",
|
|
task_id=str(uuid4()),
|
|
next="idle until QA",
|
|
evidence={"pr_url": "https://x/pr/8", "pr_number": _PR_NUMBER},
|
|
)
|
|
|
|
async def claim_review(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "claimed"
|
|
return Envelope.ok(
|
|
status="claimed",
|
|
task_id=str(uuid4()),
|
|
next="review then pass",
|
|
evidence={"pr_url": "https://x/pr/8", "pr_number": _PR_NUMBER},
|
|
)
|
|
|
|
async def pass_review(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
_notes: object,
|
|
_ac_verdicts: object = None,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "awaiting_documentation"
|
|
return Envelope.ok(
|
|
status="awaiting_documentation",
|
|
task_id=str(uuid4()),
|
|
next="idle",
|
|
)
|
|
|
|
async def claim_doc_task(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "claimed"
|
|
return Envelope.ok(
|
|
status="claimed",
|
|
task_id=str(uuid4()),
|
|
next="document then i_documented",
|
|
)
|
|
|
|
async def i_documented(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
_notes: object,
|
|
_files: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "awaiting_pm_review"
|
|
return Envelope.ok(
|
|
status="awaiting_pm_review",
|
|
task_id=str(uuid4()),
|
|
next="idle until PM completes",
|
|
)
|
|
|
|
async def complete(
|
|
self,
|
|
_agent_id: object,
|
|
_task_id: object,
|
|
_notes: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "completed"
|
|
return Envelope.ok(
|
|
status="completed",
|
|
task_id=str(uuid4()),
|
|
next="done",
|
|
)
|
|
|
|
async def main_pm_complete(
|
|
self,
|
|
_agent_id: object,
|
|
_root_task_id: object,
|
|
_notes: object,
|
|
) -> Envelope:
|
|
self._state["task_status"] = "awaiting_ceo_approval"
|
|
return Envelope.ok(
|
|
status="awaiting_ceo_approval",
|
|
task_id=str(uuid4()),
|
|
next="idle until CEO",
|
|
)
|
|
|
|
|
|
class _MockContentActions:
|
|
"""Stateful stub for ContentActions; only note() is exercised here."""
|
|
|
|
async def note(
|
|
self,
|
|
*,
|
|
agent_id: object,
|
|
text: object,
|
|
**_kwargs: object,
|
|
) -> Envelope:
|
|
# The route forwards scope / task_id / structured / section as keyword
|
|
# args; the stub absorbs them via **_kwargs (it asserts lifecycle
|
|
# transitions, not journal-entry / section rendering). ``section``
|
|
# mirrors the note(scope='handoff') write-path signature.
|
|
_ = agent_id
|
|
_ = text
|
|
return Envelope.ok(status="noted", task_id=None, next="continue")
|
|
|
|
|
|
@pytest.fixture
|
|
def stateful_app() -> Iterator[tuple[FastAPI, _State]]:
|
|
"""FastAPI app with stateful mocks (Choreographer + ContentActions)."""
|
|
app = _build_app()
|
|
state: _State = {"task_status": "pending"}
|
|
chor = _MockChoreographer(state)
|
|
actions = _MockContentActions()
|
|
app.dependency_overrides[get_choreographer] = lambda: chor
|
|
app.dependency_overrides[get_content_actions] = lambda: actions
|
|
yield app, state
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
async def test_pending_to_awaiting_ceo_approval(stateful_app: tuple) -> None:
|
|
"""Drive task pending -> awaiting_ceo_approval through all gateway roles."""
|
|
app, _state = stateful_app
|
|
client = TestClient(app)
|
|
dev_id = str(uuid4())
|
|
qa_id = str(uuid4())
|
|
doc_id = str(uuid4())
|
|
cell_pm_id = str(uuid4())
|
|
main_pm_id = str(uuid4())
|
|
|
|
# ------------------------------------------------------------------
|
|
# 1. Dev: give_me_work -> i_will_work_on -> note -> i_am_done
|
|
# ------------------------------------------------------------------
|
|
r = client.post(
|
|
"/api/v1/flow/developer/give_me_work",
|
|
headers={"X-Agent-ID": dev_id, "X-Agent-Role": "developer"},
|
|
json={},
|
|
)
|
|
assert r.status_code == _HTTP_200
|
|
|
|
r = client.post(
|
|
"/api/v1/flow/developer/i_will_work_on",
|
|
headers={"X-Agent-ID": dev_id, "X-Agent-Role": "developer"},
|
|
json={"task_id": str(uuid4()), "plan": "edit x then y"},
|
|
)
|
|
assert r.json()["status"] == "in_progress"
|
|
|
|
r = client.post(
|
|
"/api/v1/do/note",
|
|
headers={"X-Agent-ID": dev_id, "X-Agent-Role": "developer"},
|
|
json={"text": "Reflected: did the work as planned.", "scope": "reflect"},
|
|
)
|
|
assert r.json()["error"] is None
|
|
|
|
r = client.post(
|
|
"/api/v1/flow/developer/i_am_done",
|
|
headers={"X-Agent-ID": dev_id, "X-Agent-Role": "developer"},
|
|
json={"task_id": str(uuid4()), "notes": "all done"},
|
|
)
|
|
assert r.json()["status"] == "awaiting_qa"
|
|
# Inline evidence in the envelope kills issue #15 (QA false-failing "no PR")
|
|
assert r.json()["evidence"]["pr_number"] == _PR_NUMBER
|
|
|
|
# ------------------------------------------------------------------
|
|
# 2. QA: claim_review -> note -> pass
|
|
# ------------------------------------------------------------------
|
|
r = client.post(
|
|
"/api/v1/flow/qa/claim_review",
|
|
headers={"X-Agent-ID": qa_id, "X-Agent-Role": "qa"},
|
|
json={"task_id": str(uuid4())},
|
|
)
|
|
assert r.json()["evidence"]["pr_number"] == _PR_NUMBER
|
|
|
|
r = client.post(
|
|
"/api/v1/do/note",
|
|
headers={"X-Agent-ID": qa_id, "X-Agent-Role": "developer"},
|
|
json={
|
|
"text": "Reviewed; all acceptance criteria addressed.",
|
|
"scope": "learning",
|
|
},
|
|
)
|
|
assert r.json()["error"] is None
|
|
|
|
long_notes = (
|
|
"Reviewed PR #8 carefully. Branch convention correct. Commit message "
|
|
"includes task ID. README diff matches spec. No security concerns."
|
|
)
|
|
r = client.post(
|
|
"/api/v1/flow/qa/pass",
|
|
headers={"X-Agent-ID": qa_id, "X-Agent-Role": "qa"},
|
|
json={"task_id": str(uuid4()), "notes": long_notes},
|
|
)
|
|
assert r.json()["status"] == "awaiting_documentation"
|
|
|
|
# ------------------------------------------------------------------
|
|
# 3. Doc: claim_doc_task -> i_documented
|
|
# ------------------------------------------------------------------
|
|
r = client.post(
|
|
"/api/v1/flow/documenter/claim_doc_task",
|
|
headers={"X-Agent-ID": doc_id, "X-Agent-Role": "documenter"},
|
|
json={"task_id": str(uuid4())},
|
|
)
|
|
assert r.status_code == _HTTP_200
|
|
|
|
r = client.post(
|
|
"/api/v1/flow/documenter/i_documented",
|
|
headers={"X-Agent-ID": doc_id, "X-Agent-Role": "documenter"},
|
|
json={
|
|
"task_id": str(uuid4()),
|
|
"notes": "Wrote backend/guides/feature-x.md covering usage and config.",
|
|
"files": ["backend/guides/feature-x.md"],
|
|
},
|
|
)
|
|
assert r.json()["status"] == "awaiting_pm_review"
|
|
|
|
# ------------------------------------------------------------------
|
|
# 4. Cell PM: note -> complete (auto-merges leaf PR)
|
|
# ------------------------------------------------------------------
|
|
r = client.post(
|
|
"/api/v1/do/note",
|
|
headers={"X-Agent-ID": cell_pm_id, "X-Agent-Role": "developer"},
|
|
json={"text": "Decision: approve and merge.", "scope": "decision"},
|
|
)
|
|
assert r.json()["error"] is None
|
|
|
|
r = client.post(
|
|
"/api/v1/flow/cell_pm/complete",
|
|
headers={"X-Agent-ID": cell_pm_id, "X-Agent-Role": "cell_pm"},
|
|
json={"task_id": str(uuid4()), "notes": "Approved and merged"},
|
|
)
|
|
assert r.json()["status"] == "completed"
|
|
|
|
# ------------------------------------------------------------------
|
|
# 5. Main PM: note -> complete root (opens master PR + escalates to CEO)
|
|
# ------------------------------------------------------------------
|
|
r = client.post(
|
|
"/api/v1/do/note",
|
|
headers={"X-Agent-ID": main_pm_id, "X-Agent-Role": "developer"},
|
|
json={"text": "Root task ready for prod.", "scope": "decision"},
|
|
)
|
|
assert r.json()["error"] is None
|
|
|
|
r = client.post(
|
|
"/api/v1/flow/main_pm/complete",
|
|
headers={"X-Agent-ID": main_pm_id, "X-Agent-Role": "main_pm"},
|
|
json={"task_id": str(uuid4()), "notes": "Ready for prod"},
|
|
)
|
|
assert r.json()["status"] == "awaiting_ceo_approval"
|