feat(plan): services/notes — task_kind create param + list filter
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
"""task_kind create + list-filter behavior in services/notes.py."""
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
def _make_mock_session():
|
||||
s = AsyncMock()
|
||||
s.__aenter__ = AsyncMock(return_value=s)
|
||||
s.__aexit__ = AsyncMock(return_value=False)
|
||||
s.add = MagicMock()
|
||||
s.commit = AsyncMock()
|
||||
s.refresh = AsyncMock()
|
||||
return s
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_note_passes_task_kind_to_model():
|
||||
mock_session = _make_mock_session()
|
||||
captured = {}
|
||||
|
||||
def _capture_add(obj):
|
||||
captured["task_kind"] = getattr(obj, "task_kind", "MISSING")
|
||||
|
||||
mock_session.add = MagicMock(side_effect=_capture_add)
|
||||
with patch("fabledassistant.services.notes.async_session") as mock_cls, \
|
||||
patch("fabledassistant.services.notes._maybe_reactivate_project", AsyncMock()):
|
||||
mock_cls.return_value = mock_session
|
||||
from fabledassistant.services.notes import create_note
|
||||
await create_note(user_id=1, title="P", status="todo", task_kind="plan")
|
||||
assert captured["task_kind"] == "plan"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_notes_filters_by_task_kind():
|
||||
mock_session = _make_mock_session()
|
||||
mock_session.scalar = AsyncMock(return_value=0)
|
||||
exec_result = MagicMock()
|
||||
exec_result.scalars.return_value.all.return_value = []
|
||||
mock_session.execute = AsyncMock(return_value=exec_result)
|
||||
with patch("fabledassistant.services.notes.async_session") as mock_cls:
|
||||
mock_cls.return_value = mock_session
|
||||
from fabledassistant.services.notes import list_notes
|
||||
# Should not raise; task_kind is a recognized kwarg.
|
||||
rows, total = await list_notes(user_id=1, is_task=True, task_kind="plan")
|
||||
assert rows == []
|
||||
assert total == 0
|
||||
Reference in New Issue
Block a user