47 lines
2.2 KiB
Python
47 lines
2.2 KiB
Python
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_planning_creates_plan_task_and_returns_rules():
|
|
fake_note = MagicMock()
|
|
fake_note.to_dict.return_value = {"id": 5, "title": "Plan it", "task_kind": "plan"}
|
|
applicable = {
|
|
"rules": [{"id": 1, "title": "dev is home", "statement": "...",
|
|
"topic_title": "git-workflow", "rulebook_title": "FabledSword family"}],
|
|
"truncated": False,
|
|
"subscribed_rulebooks": [{"id": 2, "title": "FabledSword family"}],
|
|
}
|
|
with patch("fabledassistant.services.planning.notes_svc.create_note",
|
|
AsyncMock(return_value=fake_note)) as mock_create, \
|
|
patch("fabledassistant.services.planning.rulebooks_svc.get_applicable_rules",
|
|
AsyncMock(return_value=applicable)), \
|
|
patch("fabledassistant.services.planning.notes_svc.list_notes",
|
|
AsyncMock(return_value=([], 3))), \
|
|
patch("fabledassistant.services.planning.projects_svc.get_project",
|
|
AsyncMock(return_value=MagicMock(goal="ship it"))):
|
|
from fabledassistant.services.planning import start_planning
|
|
out = await start_planning(user_id=7, project_id=3, title="Plan it")
|
|
|
|
# Created a plan-task (status set => task, kind=plan)
|
|
kwargs = mock_create.call_args.kwargs
|
|
assert kwargs["task_kind"] == "plan"
|
|
assert kwargs["status"] == "todo"
|
|
assert kwargs["project_id"] == 3
|
|
assert "## Goal" in kwargs["body"] # seeded template
|
|
# Returned shape
|
|
assert out["task"]["id"] == 5
|
|
assert out["applicable_rules"][0]["title"] == "dev is home"
|
|
assert out["subscribed_rulebooks"] == [{"id": 2, "title": "FabledSword family"}]
|
|
assert out["open_task_count"] == 3
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_planning_raises_when_project_not_found():
|
|
with patch("fabledassistant.services.planning.projects_svc.get_project",
|
|
AsyncMock(return_value=None)):
|
|
from fabledassistant.services.planning import start_planning
|
|
with pytest.raises(ValueError, match="project 999 not found"):
|
|
await start_planning(user_id=7, project_id=999, title="x")
|