# requires: mcp, pydantic, pytest, anyio
import json

import anyio
import pytest

from mcp import Client

from kb_server import TICKETS, mcp


@pytest.fixture
def anyio_backend():
    return "asyncio"


async def _session():
    async with Client(mcp) as c:               # 测试时直接传入服务对象，进程内通信
        tools = {t.name: t for t in (await c.list_tools()).tools}
        assert set(tools) == {"search_docs", "create_ticket"}
        assert "request_id" in json.dumps(tools["create_ticket"].input_schema)

        r = await c.call_tool("search_docs", {"query": "VPN 失败"})
        assert not r.is_error and "vpn" in json.dumps(r.structured_content, ensure_ascii=False)

        args = {"ticket": {"title": "VPN 无法连接", "category": "it", "request_id": "req-1"}}
        first = await c.call_tool("create_ticket", args)
        second = await c.call_tool("create_ticket", args)
        assert first.structured_content == second.structured_content and len(TICKETS) == 1

        bad = await c.call_tool("create_ticket", {"ticket": {"title": "x", "category": "other", "request_id": "r"}})
        assert bad.is_error                    # 参数校验失败以错误结果返回给客户端

        res = await c.read_resource("kb://docs/expense")
        assert "600" in res.contents[0].text

        p = await c.get_prompt("answer_with_citation", {"question": "住宿标准"})
        assert "search_docs" in p.messages[0].content.text
        return r


@pytest.mark.anyio
async def test_kb_server():
    await _session()


if __name__ == "__main__":
    r = anyio.run(_session)
    print("search_docs 结构化结果：", r.structured_content)
    print("工单：", TICKETS)