FabledGrimoire/tests/test_services_notes_task_kind.py

"""task_kind create + list-filter behavior in services/notes.py."""
from unittest.mock import AsyncMock, MagicMock, patch

import pytest
from tests.helpers import make_mock_session


@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("grimoire.services.notes.async_session") as mock_cls, \
         patch("grimoire.services.notes._maybe_reactivate_project", AsyncMock()):
        mock_cls.return_value = mock_session
        from grimoire.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("grimoire.services.notes.async_session") as mock_cls:
        mock_cls.return_value = mock_session
        from grimoire.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