[F109] playbook curation status guards: approve/reject draft-only, archive approved-only

This commit is contained in:
Renn F
2026-06-28 22:21:32 +02:00
parent 27b48dd64e
commit f779fe7453
8 changed files with 284 additions and 21 deletions
+49
View File
@@ -110,3 +110,52 @@ async def test_non_curator_is_forbidden(db_session: AsyncSession) -> None:
assert get_resp.status_code == HTTPStatus.FORBIDDEN
assert approve_resp.status_code == HTTPStatus.FORBIDDEN
app.dependency_overrides.clear()
# --- F109: curation status-precondition guards at the route layer ------------- #
# approve/reject only act on a draft; archive only retires an approved playbook.
# A violation is a 409 Conflict (the curation is already finished), not a 200
# that silently no-ops or a 500 from an uncaught ConflictError.
@pytest.mark.asyncio
async def test_approve_already_approved_is_409(
db_session: AsyncSession, auditor_client: AsyncClient
) -> None:
pid = await _seed_draft(db_session, title="Once only")
first = await auditor_client.post(f"/api/playbooks/{pid}/approve")
assert first.status_code == HTTPStatus.OK
second = await auditor_client.post(f"/api/playbooks/{pid}/approve")
assert second.status_code == HTTPStatus.CONFLICT
@pytest.mark.asyncio
async def test_reject_approved_is_409(
db_session: AsyncSession, auditor_client: AsyncClient
) -> None:
pid = await _seed_draft(db_session, title="Published route")
await auditor_client.post(f"/api/playbooks/{pid}/approve")
resp = await auditor_client.post(
f"/api/playbooks/{pid}/reject", json={"reason": "changed my mind"}
)
assert resp.status_code == HTTPStatus.CONFLICT
@pytest.mark.asyncio
async def test_archive_approved_succeeds(
db_session: AsyncSession, auditor_client: AsyncClient
) -> None:
pid = await _seed_draft(db_session, title="Retire route")
await auditor_client.post(f"/api/playbooks/{pid}/approve")
resp = await auditor_client.post(f"/api/playbooks/{pid}/archive")
assert resp.status_code == HTTPStatus.OK
assert resp.json()["status"] == "archived"
@pytest.mark.asyncio
async def test_archive_draft_is_409(
db_session: AsyncSession, auditor_client: AsyncClient
) -> None:
pid = await _seed_draft(db_session, title="Not yet approved")
resp = await auditor_client.post(f"/api/playbooks/{pid}/archive")
assert resp.status_code == HTTPStatus.CONFLICT
+79 -1
View File
@@ -111,7 +111,11 @@ async def test_approve_indexes_when_org_memory_on(
)
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Index me"), created_by=uuid4())
await svc.approve(pb.id, approver_id=uuid4())
approved = await svc.approve(pb.id, approver_id=uuid4())
# approve() only flushes the status; indexing is the separate post-commit
# step the route/verb runs after committing (so the index never leads the
# status transaction). Mirror that ordering here.
await svc.index_approved(approved)
fake_optimal.index_playbook.assert_awaited_once()
@@ -143,3 +147,77 @@ async def test_approve_survives_index_failure(
pb = await svc.draft(_create(title="Resilient"), created_by=uuid4())
approved = await svc.approve(pb.id, approver_id=uuid4()) # must not raise
assert approved.status == PlaybookStatus.APPROVED
# --- F109: approve/reject/archive status-precondition guards ---------------- #
# The lifecycle is draft -> approved | archived, both terminal. approve/reject
# only act on a DRAFT; archive only retires an APPROVED playbook. An archived
# playbook is terminal — none of the three may touch it again. Without these
# guards an archived playbook could be re-approved and an approved one rejected,
# silently undoing a finished curation.
@pytest.mark.asyncio
async def test_approve_rejects_already_approved(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Once"), created_by=uuid4())
await svc.approve(pb.id, approver_id=uuid4())
with pytest.raises(ConflictError):
await svc.approve(pb.id, approver_id=uuid4())
@pytest.mark.asyncio
async def test_approve_rejects_archived(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Done"), created_by=uuid4())
await svc.reject(pb.id, approver_id=uuid4(), reason="duplicate")
with pytest.raises(ConflictError):
await svc.approve(pb.id, approver_id=uuid4())
@pytest.mark.asyncio
async def test_reject_rejects_already_approved(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Published svc"), created_by=uuid4())
await svc.approve(pb.id, approver_id=uuid4())
with pytest.raises(ConflictError):
await svc.reject(pb.id, approver_id=uuid4(), reason="changed my mind")
@pytest.mark.asyncio
async def test_reject_rejects_archived(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Closed"), created_by=uuid4())
await svc.reject(pb.id, approver_id=uuid4(), reason="duplicate")
with pytest.raises(ConflictError):
await svc.reject(pb.id, approver_id=uuid4(), reason="again")
@pytest.mark.asyncio
async def test_archive_retires_approved(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
auditor = uuid4()
pb = await svc.draft(_create(title="Retire svc"), created_by=uuid4())
await svc.approve(pb.id, approver_id=auditor)
archived = await svc.archive(pb.id, approver_id=auditor)
assert archived.status == PlaybookStatus.ARCHIVED
assert archived.approved_by == auditor
assert archived.approved_at is not None
@pytest.mark.asyncio
async def test_archive_rejects_draft(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Not yet"), created_by=uuid4())
with pytest.raises(ConflictError):
await svc.archive(pb.id, approver_id=uuid4())
@pytest.mark.asyncio
async def test_archive_rejects_already_archived(db_session: AsyncSession) -> None:
svc = PlaybookService(db_session)
pb = await svc.draft(_create(title="Twice"), created_by=uuid4())
await svc.approve(pb.id, approver_id=uuid4())
await svc.archive(pb.id, approver_id=uuid4())
with pytest.raises(ConflictError):
await svc.archive(pb.id, approver_id=uuid4())
+43
View File
@@ -11,6 +11,7 @@ from unittest.mock import AsyncMock, MagicMock
from uuid import uuid4
import pytest
from roboco.services.base import ConflictError
from roboco.services.gateway.content_actions import ContentActions, ContentActionsDeps
from roboco.services.gateway.role_config import get_role_config
@@ -140,3 +141,45 @@ async def test_reject_playbook_archives_for_auditor(
# F057: de-index is the post-commit step (commit gates it).
actions.task.session.commit.assert_awaited_once()
svc.unindex_playbook.assert_awaited_once_with(archived)
@pytest.mark.asyncio
async def test_archive_playbook_retires_approved_for_auditor(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""archive_playbook is the distinct APPROVED->archived retire path (F109):
it calls ``svc.archive`` (NOT ``svc.reject``), commits, then de-indexes."""
archived = MagicMock()
archived.id = uuid4()
archived.status = "archived"
svc = MagicMock()
svc.archive = AsyncMock(return_value=archived)
svc.reject = AsyncMock()
svc.unindex_playbook = AsyncMock()
monkeypatch.setattr("roboco.services.playbook.get_playbook_service", lambda _s: svc)
actions = _actions("auditor")
env = await actions.archive_playbook(agent_id=uuid4(), playbook_id=uuid4())
assert env.status == "playbook_archived"
svc.archive.assert_awaited_once()
svc.reject.assert_not_awaited()
actions.task.session.commit.assert_awaited_once()
svc.unindex_playbook.assert_awaited_once_with(archived)
@pytest.mark.asyncio
async def test_approve_playbook_invalid_state_envelope(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""A status-precondition ConflictError from the service becomes a clean
invalid_state envelope (not a 500) — the agent gets a remediate hint to
re-fetch the playbook's current status before re-trying (F109)."""
svc = MagicMock()
svc.approve = AsyncMock(
side_effect=ConflictError("not draft", resource_type="playbook")
)
monkeypatch.setattr("roboco.services.playbook.get_playbook_service", lambda _s: svc)
env = await _actions("auditor").approve_playbook(
agent_id=uuid4(), playbook_id=uuid4()
)
assert env.error == "invalid_state"
assert env.remediate # the agent is told how to recover
@@ -37,7 +37,10 @@ _APPROVER = uuid4()
def _playbook_mock() -> Any:
pb = MagicMock(name="playbook")
pb.id = _PID
pb.status = "approved"
# approve/reject act on a DRAFT (their legitimate starting state — F109
# guards both to DRAFT-only). The index/unindex ordering assertions below
# are independent of the starting status.
pb.status = "draft"
pb.title = "T"
pb.problem = "P"
pb.procedure = "Pr"
+17 -13
View File
@@ -1,14 +1,16 @@
"""PlaybookService.reject must flush the status change WITHOUT de-indexing
inline, and ``unindex_playbook`` is the separate post-commit de-index step.
"""PlaybookService.reject/archive must flush the status change WITHOUT
de-indexing inline, and ``unindex_playbook`` is the separate post-commit step.
A rejected/archived playbook that was previously approved must stop surfacing in
agent briefings, so a reject drops its chunks + tracking row. Originally (F011)
``reject`` called an ``_unindex_playbook`` helper inline. F057 split that out:
the RAG index write runs through its own auto-committing pool connection, so
de-indexing inline (before the caller commits the status) would drop a playbook
from the corpus even if the status transaction rolled back — a divergence. So
``reject`` now flushes the status ONLY; ``unindex_playbook`` is a separate
public step the caller runs AFTER committing. Both helpers stay gated on
agent briefings, so the curation drop its chunks + tracking row. Originally
(F011) ``reject`` called an ``_unindex_playbook`` helper inline. F057 split that
out: the RAG index write runs through its own auto-committing pool connection,
so de-indexing inline (before the caller commits the status) would drop a
playbook from the corpus even if the status transaction rolled back — a
divergence. So ``reject``/``archive`` now flush the status ONLY;
``unindex_playbook`` is a separate public step the caller runs AFTER
committing. F109 split the APPROVED->archived retire path into ``archive``
(distinct from ``reject``, which declines a DRAFT). Both helpers stay gated on
``org_memory_enabled`` (inert when the loop is off) and best-effort.
"""
@@ -49,11 +51,13 @@ def _session_with(pb: Any) -> MagicMock:
@pytest.mark.asyncio
async def test_reject_archives_but_does_not_deindex_inline(
async def test_archive_archives_but_does_not_deindex_inline(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""``reject`` flushes the ARCHIVED status but must NOT touch the RAG index —
the de-index is a separate post-commit step (F057 ordering)."""
"""``archive`` flushes the ARCHIVED status but must NOT touch the RAG index —
the de-index is a separate post-commit step (F057 ordering). Archive retires
an APPROVED playbook (F109); the previously-approved one was indexed on
approval, so the post-commit ``unindex_playbook`` is what removes it."""
monkeypatch.setattr(settings, "org_memory_enabled", True)
playbook_id = uuid4()
pb = _mock_playbook(playbook_id, status=PlaybookStatus.APPROVED.value)
@@ -67,7 +71,7 @@ async def test_reject_archives_but_does_not_deindex_inline(
AsyncMock(return_value=optimal),
),
):
out = await svc.reject(playbook_id, approver_id=uuid4(), reason="stale")
out = await svc.archive(playbook_id, approver_id=uuid4())
assert out.status == PlaybookStatus.ARCHIVED.value
optimal.unindex_playbook.assert_not_awaited()