67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
"""Tests for the trash MCP tools — patch the trash service."""
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from fabledassistant.mcp._context import _user_id_ctx
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _bind_user():
|
|
token = _user_id_ctx.set(7)
|
|
yield
|
|
_user_id_ctx.reset(token)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_trash_wraps_batches():
|
|
with patch("fabledassistant.mcp.tools.trash.trash_svc.list_trash",
|
|
AsyncMock(return_value=[{"batch_id": "b1", "count": 2}])):
|
|
from fabledassistant.mcp.tools.trash import list_trash
|
|
out = await list_trash()
|
|
assert out["batches"][0]["batch_id"] == "b1"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_restore_delegates():
|
|
with patch("fabledassistant.mcp.tools.trash.trash_svc.restore",
|
|
AsyncMock(return_value=3)) as mock:
|
|
from fabledassistant.mcp.tools.trash import restore
|
|
out = await restore("b1")
|
|
assert out == {"restored": 3, "batch_id": "b1"}
|
|
assert mock.call_args.args == (7, "b1")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_purge_trash_requires_confirmed():
|
|
with patch("fabledassistant.mcp.tools.trash.trash_svc.purge", AsyncMock()) as mock:
|
|
from fabledassistant.mcp.tools.trash import purge_trash
|
|
out = await purge_trash("b1", confirmed=False)
|
|
assert out.get("confirmed_required") is True
|
|
assert not mock.called
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_purge_trash_when_confirmed():
|
|
with patch("fabledassistant.mcp.tools.trash.trash_svc.purge",
|
|
AsyncMock(return_value=5)) as mock:
|
|
from fabledassistant.mcp.tools.trash import purge_trash
|
|
out = await purge_trash("b1", confirmed=True)
|
|
assert out == {"purged": 5, "batch_id": "b1"}
|
|
assert mock.called
|
|
|
|
|
|
def test_register_attaches_three_tools():
|
|
from fabledassistant.mcp.tools.trash import register
|
|
names: list[str] = []
|
|
|
|
class FakeMCP:
|
|
def tool(self, name=None):
|
|
def deco(fn):
|
|
names.append(name)
|
|
return fn
|
|
return deco
|
|
|
|
register(FakeMCP())
|
|
assert set(names) == {"list_trash", "restore", "purge_trash"}
|