gilles
e902452781
Ajoute search_notes, get_note, create_note, update_note, delete_note au serveur MCP. Tests: 6 nouveaux tests notes (13 tests MCP au total, tous passent). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
124 lines
4.2 KiB
Python
124 lines
4.2 KiB
Python
import json
|
|
import pytest
|
|
from sqlalchemy import delete
|
|
import app.api.mcp_server as mcp_server_module
|
|
from app.api.mcp_server import (
|
|
get_todos, create_todo, update_todo, postpone_todo, delete_todo,
|
|
)
|
|
from app.api.mcp_server import (
|
|
search_notes, get_note, create_note, update_note, delete_note,
|
|
)
|
|
from app.models.todos import TodoItem
|
|
from app.models.notes import NoteItem
|
|
|
|
pytestmark = pytest.mark.usefixtures("mcp_nullpool_session")
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def cleanup_mcp_todos():
|
|
yield
|
|
# Utilise AsyncSessionLocal depuis le module mcp_server (patché par mcp_nullpool_session)
|
|
async with mcp_server_module.AsyncSessionLocal() as session:
|
|
await session.execute(delete(TodoItem).where(TodoItem.title.like("TEST_MCP_%")))
|
|
await session.execute(delete(NoteItem).where(NoteItem.title.like("TEST_MCP_%")))
|
|
await session.commit()
|
|
|
|
|
|
async def test_get_todos_retourne_liste_json():
|
|
result = await get_todos(status="pending")
|
|
data = json.loads(result)
|
|
assert isinstance(data, list)
|
|
|
|
|
|
async def test_create_todo_outil_cree_une_tache():
|
|
result = await create_todo(title="TEST_MCP_todo_create")
|
|
data = json.loads(result)
|
|
assert data["title"] == "TEST_MCP_todo_create"
|
|
assert data["status"] == "pending"
|
|
assert data["priority"] == "medium"
|
|
# Cleanup
|
|
await delete_todo(id=str(data["id"]))
|
|
|
|
|
|
async def test_update_todo_outil():
|
|
created = json.loads(await create_todo(title="TEST_MCP_todo_update"))
|
|
result = await update_todo(id=str(created["id"]), status="done")
|
|
data = json.loads(result)
|
|
assert data["status"] == "done"
|
|
await delete_todo(id=str(created["id"]))
|
|
|
|
|
|
async def test_postpone_todo_outil():
|
|
created = json.loads(await create_todo(title="TEST_MCP_todo_postpone"))
|
|
result = await postpone_todo(id=str(created["id"]), days=3)
|
|
data = json.loads(result)
|
|
assert data["postponed_count"] == 1
|
|
await delete_todo(id=str(created["id"]))
|
|
|
|
|
|
async def test_delete_todo_outil():
|
|
created = json.loads(await create_todo(title="TEST_MCP_todo_delete"))
|
|
result = await delete_todo(id=str(created["id"]))
|
|
data = json.loads(result)
|
|
assert "deleted" in data
|
|
|
|
|
|
async def test_update_todo_id_invalide():
|
|
result = await update_todo(id="pas-un-uuid", title="x")
|
|
data = json.loads(result)
|
|
assert "error" in data
|
|
|
|
|
|
async def test_delete_todo_introuvable():
|
|
result = await delete_todo(id="00000000-0000-0000-0000-000000000000")
|
|
data = json.loads(result)
|
|
assert "error" in data
|
|
|
|
|
|
# ── NOTES ──────────────────────────────────────────────────────────────────────
|
|
|
|
async def test_search_notes_retourne_liste():
|
|
result = await search_notes(query="inexistant_xyz_abc_999")
|
|
data = json.loads(result)
|
|
assert isinstance(data, list)
|
|
assert data == []
|
|
|
|
|
|
async def test_create_note_outil():
|
|
result = await create_note(title="TEST_MCP_note_create", content="Contenu de test MCP")
|
|
data = json.loads(result)
|
|
assert data["title"] == "TEST_MCP_note_create"
|
|
assert "id" in data
|
|
await delete_note(id=str(data["id"]))
|
|
|
|
|
|
async def test_get_note_outil():
|
|
created = json.loads(await create_note(title="TEST_MCP_note_get", content="Contenu get"))
|
|
result = await get_note(id=str(created["id"]))
|
|
data = json.loads(result)
|
|
assert data["title"] == "TEST_MCP_note_get"
|
|
assert data["content"] == "Contenu get"
|
|
assert "attachments" in data
|
|
await delete_note(id=str(created["id"]))
|
|
|
|
|
|
async def test_update_note_outil():
|
|
created = json.loads(await create_note(title="TEST_MCP_note_update", content="avant"))
|
|
result = await update_note(id=str(created["id"]), content="après")
|
|
data = json.loads(result)
|
|
assert "id" in data
|
|
await delete_note(id=str(created["id"]))
|
|
|
|
|
|
async def test_delete_note_outil():
|
|
created = json.loads(await create_note(title="TEST_MCP_note_delete", content="x"))
|
|
result = await delete_note(id=str(created["id"]))
|
|
data = json.loads(result)
|
|
assert "deleted" in data
|
|
|
|
|
|
async def test_get_note_introuvable():
|
|
result = await get_note(id="00000000-0000-0000-0000-000000000000")
|
|
data = json.loads(result)
|
|
assert "error" in data
|