9fa474b3c4
CI & Build / Python lint (push) Successful in 3s
CI & Build / integration (push) Successful in 31s
CI & Build / TypeScript typecheck (push) Successful in 42s
CI & Build / Python tests (push) Successful in 56s
CI & Build / Build & push image (push) Successful in 1m0s
The injected menu tells the agent to open any hit with get_note(id), and that menu can list a collaborator's record reached through a shared project. The fetch was still owner-only, so those lines answered "not found" — for a record the same user opens fine in the browser. The agent path was strictly narrower than the web path for the same id. Same boundary miss as #2093: the list side was widened for sharing, the fetch side wasn't. Both tools now resolve through get_note_for_user, apply the trash filter themselves (permission resolution says nothing about liveness), and attach describe_provenance so a shared record arrives marked as someone else's rather than passing as the caller's. routes/tasks.py's parent-title lookup had the same narrowness: a shared subtask rendered as an orphan when its parent was equally shared. Four fetch tests across three files were patching notes_svc.get_note and had to be retargeted — note 2109's third sub-case, caught by grepping tests/ for the old name before pushing rather than by CI (#2159). Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01RLwAaV4DQEmVyn496HnEvt
299 lines
11 KiB
Python
299 lines
11 KiB
Python
"""Tests for fable_*_task tools."""
|
|
from datetime import datetime, timezone
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from scribe.mcp._context import _user_id_ctx
|
|
from scribe.mcp.tools.tasks import (
|
|
list_tasks, get_task, create_task,
|
|
update_task, add_task_log,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _bind_user():
|
|
token = _user_id_ctx.set(7)
|
|
yield
|
|
_user_id_ctx.reset(token)
|
|
|
|
|
|
def _fake_task(*, parent_id: int | None = None, user_id: int = 7,
|
|
**overrides) -> MagicMock:
|
|
n = MagicMock()
|
|
n.parent_id = parent_id
|
|
base = {
|
|
"id": 1, "title": "t", "body": "", "status": "todo",
|
|
"priority": "none", "tags": [], "parent_id": parent_id,
|
|
"is_task": True,
|
|
}
|
|
base.update(overrides)
|
|
n.to_dict.return_value = base
|
|
n.title = base["title"]
|
|
n.id = base["id"]
|
|
# Real values, not auto-attributes: get_task reads deleted_at and compares
|
|
# user_id against the bound caller for the shared/owner marker — a MagicMock
|
|
# is truthy on both (note 2109).
|
|
n.user_id = user_id
|
|
n.deleted_at = None
|
|
return n
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_tasks_passes_is_task_true_and_repackages():
|
|
rows = [_fake_task(id=1), _fake_task(id=2)]
|
|
mock = AsyncMock(return_value=(rows, 2))
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.list_notes", mock):
|
|
out = await list_tasks()
|
|
assert out["total"] == 2
|
|
assert len(out["tasks"]) == 2
|
|
assert mock.call_args.kwargs["is_task"] is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_tasks_status_filter():
|
|
mock = AsyncMock(return_value=([], 0))
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.list_notes", mock):
|
|
await list_tasks(status="in_progress")
|
|
assert mock.call_args.kwargs["status"] == "in_progress"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_tasks_empty_status_means_no_filter():
|
|
mock = AsyncMock(return_value=([], 0))
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.list_notes", mock):
|
|
await list_tasks(status="")
|
|
assert mock.call_args.kwargs["status"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_task_with_no_parent_returns_null_parent_title():
|
|
fake = _fake_task(id=5, title="solo", parent_id=None)
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.notes_svc.get_note_for_user",
|
|
AsyncMock(return_value=(fake, "owner")),
|
|
):
|
|
out = await get_task(task_id=5)
|
|
assert out["parent_title"] is None
|
|
assert out["title"] == "solo"
|
|
assert "shared" not in out
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_task_enriches_with_parent_title():
|
|
"""When parent_id is set, get_task fetches the parent and adds parent_title."""
|
|
child = _fake_task(id=10, title="child", parent_id=5)
|
|
parent = _fake_task(id=5, title="parent of 10", parent_id=None)
|
|
# fetched twice: once for the child, once for the parent
|
|
mock_get = AsyncMock(side_effect=[(child, "owner"), (parent, "owner")])
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.get_note_for_user", mock_get):
|
|
out = await get_task(task_id=10)
|
|
assert out["parent_title"] == "parent of 10"
|
|
assert mock_get.await_count == 2
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_task_parent_missing_returns_null():
|
|
"""If parent_id is set but the parent is gone (orphaned), parent_title is None."""
|
|
child = _fake_task(id=10, parent_id=5)
|
|
mock_get = AsyncMock(side_effect=[(child, "owner"), None])
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.get_note_for_user", mock_get):
|
|
out = await get_task(task_id=10)
|
|
assert out["parent_title"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_task_opens_a_shared_task_and_says_whose_it_is():
|
|
"""A task in a shared project opens in the web UI, so the agent path must not
|
|
answer "not found" for the same id — and must say it isn't the caller's."""
|
|
theirs = _fake_task(id=5, title="Their task", user_id=9)
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.notes_svc.get_note_for_user",
|
|
AsyncMock(return_value=(theirs, "viewer")),
|
|
), patch(
|
|
"scribe.services.access.describe_provenance",
|
|
AsyncMock(return_value={"shared": True, "owner": "alex",
|
|
"permission": "viewer"}),
|
|
):
|
|
out = await get_task(task_id=5)
|
|
assert out["shared"] is True
|
|
assert out["owner"] == "alex"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_task_raises_when_not_found():
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.notes_svc.get_note_for_user",
|
|
AsyncMock(return_value=None),
|
|
):
|
|
with pytest.raises(ValueError, match="task 999 not found"):
|
|
await get_task(task_id=999)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_passes_status():
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.create_note", mock):
|
|
await create_task(title="do x", status="todo")
|
|
assert mock.call_args.kwargs["status"] == "todo"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_blocked_by_duplicate_gate():
|
|
"""A near-duplicate blocks creation and returns the existing id (no insert)."""
|
|
from scribe.services.dedup import DuplicateMatch
|
|
dup = DuplicateMatch(id=42, title="Set up CI", similarity=1.0, reason="title")
|
|
create_mock = AsyncMock()
|
|
with patch("scribe.mcp.tools.tasks.dedup_svc.find_duplicate_note",
|
|
AsyncMock(return_value=dup)), \
|
|
patch("scribe.mcp.tools.tasks.notes_svc.create_note", create_mock):
|
|
out = await create_task(title="set up ci")
|
|
assert out["duplicate"] is True
|
|
assert out["existing_id"] == 42
|
|
create_mock.assert_not_called() # nothing was created
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_force_bypasses_duplicate_gate():
|
|
"""force=true skips the gate entirely and creates."""
|
|
find_mock = AsyncMock()
|
|
with patch("scribe.mcp.tools.tasks.dedup_svc.find_duplicate_note", find_mock), \
|
|
patch("scribe.mcp.tools.tasks.notes_svc.create_note",
|
|
AsyncMock(return_value=_fake_task(id=9))):
|
|
out = await create_task(title="dup", force=True)
|
|
assert out["id"] == 9
|
|
find_mock.assert_not_called() # gate not even consulted
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_rejects_retired_plan_kind():
|
|
"""kind=plan is hard-retired — plans are milestones (start_planning)."""
|
|
mock = AsyncMock()
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.create_note", mock):
|
|
with pytest.raises(ValueError, match="kind=plan is retired"):
|
|
await create_task(title="x", kind="plan")
|
|
assert not mock.called # never reached the create
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_priority_empty_becomes_none():
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.create_note", mock):
|
|
await create_task(title="x", priority="")
|
|
assert mock.call_args.kwargs["priority"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_task_zero_id_sentinels_become_none():
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.create_note", mock):
|
|
await create_task(title="x", project_id=0, milestone_id=0, parent_id=0)
|
|
assert mock.call_args.kwargs["project_id"] is None
|
|
assert mock.call_args.kwargs["milestone_id"] is None
|
|
assert mock.call_args.kwargs["parent_id"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_only_sends_non_default_fields():
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, status="done")
|
|
args, kwargs = mock.call_args
|
|
assert args == (7, 1)
|
|
assert kwargs == {"status": "done"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_empty_priority_is_omitted():
|
|
"""Priority="" is "leave unchanged" — must not reach service as empty string."""
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, status="done", priority="")
|
|
assert "priority" not in mock.call_args.kwargs
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_raises_when_not_found():
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.notes_svc.update_note",
|
|
AsyncMock(return_value=None),
|
|
):
|
|
with pytest.raises(ValueError, match="task 999 not found"):
|
|
await update_task(task_id=999, status="done")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_milestone_zero_is_omitted():
|
|
"""milestone_id=0 is 'leave unchanged' — must not reach the service."""
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, milestone_id=0)
|
|
assert "milestone_id" not in mock.call_args.kwargs
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_milestone_positive_is_set():
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, milestone_id=42)
|
|
assert mock.call_args.kwargs["milestone_id"] == 42
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_milestone_negative_one_clears():
|
|
"""milestone_id=-1 clears the milestone (sets the column NULL)."""
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, milestone_id=-1)
|
|
assert mock.call_args.kwargs["milestone_id"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_update_task_clearing_project_also_clears_milestone():
|
|
"""project_id=-1 clears the project and, with it, the milestone."""
|
|
fake = _fake_task()
|
|
mock = AsyncMock(return_value=fake)
|
|
with patch("scribe.mcp.tools.tasks.notes_svc.update_note", mock):
|
|
await update_task(task_id=1, project_id=-1)
|
|
assert mock.call_args.kwargs["project_id"] is None
|
|
assert mock.call_args.kwargs["milestone_id"] is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_task_log_returns_log_dict():
|
|
log = MagicMock()
|
|
log.id = 5
|
|
log.task_id = 1
|
|
log.content = "checkpoint"
|
|
log.created_at = datetime(2026, 5, 26, 12, tzinfo=timezone.utc)
|
|
# No to_dict on TaskLog model (per task_logs.py inspection) — fall through to manual dict
|
|
del log.to_dict
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.task_logs_svc.create_log",
|
|
AsyncMock(return_value=log),
|
|
):
|
|
out = await add_task_log(task_id=1, content="checkpoint")
|
|
assert out["id"] == 5
|
|
assert out["task_id"] == 1
|
|
assert out["content"] == "checkpoint"
|
|
assert out["created_at"].startswith("2026-05-26")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_task_log_propagates_value_error_when_task_missing():
|
|
"""create_log raises ValueError if the task doesn't exist; we let it through."""
|
|
with patch(
|
|
"scribe.mcp.tools.tasks.task_logs_svc.create_log",
|
|
AsyncMock(side_effect=ValueError("Task 999 not found")),
|
|
):
|
|
with pytest.raises(ValueError):
|
|
await add_task_log(task_id=999, content="x")
|