mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-22 23:31:12 +02:00
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
"""surfsense_get_document round-trips through the real tool registration.
|
|
|
|
The markdown form must ask the backend for the OKF concept via content
|
|
negotiation (``Accept: text/markdown``) and pass it through untouched; the JSON
|
|
form must leave the default ``application/json`` Accept in place. This is the
|
|
only coverage of the MCP-side glue that forwards the header.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from unittest.mock import MagicMock
|
|
|
|
import httpx
|
|
from mcp.server.fastmcp import FastMCP
|
|
|
|
from mcp_server.core.client import SurfSenseClient
|
|
from mcp_server.features.knowledge_base import search_tools
|
|
|
|
_CONCEPT = "---\ntype: Note\ntitle: T\n---\n\nBody."
|
|
|
|
|
|
def _client_recording(seen: dict) -> SurfSenseClient:
|
|
async def handler(request: httpx.Request) -> httpx.Response:
|
|
seen["path"] = request.url.path
|
|
seen["accept"] = request.headers.get("accept")
|
|
if "text/markdown" in (seen["accept"] or ""):
|
|
return httpx.Response(
|
|
200, text=_CONCEPT, headers={"content-type": "text/markdown"}
|
|
)
|
|
return httpx.Response(200, json={"id": 1, "title": "T"})
|
|
|
|
client = SurfSenseClient(
|
|
api_base="http://test/api/v1", timeout=5, fallback_api_key="ss_pat_x"
|
|
)
|
|
client._http = httpx.AsyncClient(
|
|
base_url="http://test/api/v1",
|
|
headers={"Accept": "application/json"},
|
|
transport=httpx.MockTransport(handler),
|
|
)
|
|
return client
|
|
|
|
|
|
def _call_get_document(client: SurfSenseClient, **arguments) -> str:
|
|
mcp = FastMCP("test")
|
|
search_tools.register(mcp, client, MagicMock())
|
|
blocks = asyncio.run(mcp.call_tool("surfsense_get_document", arguments))
|
|
return "".join(block.text for block in blocks)
|
|
|
|
|
|
def test_markdown_requests_okf_concept_and_passes_it_through():
|
|
seen: dict = {}
|
|
text = _call_get_document(_client_recording(seen), document_id=1)
|
|
|
|
assert seen["path"] == "/api/v1/documents/1"
|
|
assert "text/markdown" in seen["accept"]
|
|
assert text == _CONCEPT
|
|
|
|
|
|
def test_json_keeps_default_accept():
|
|
seen: dict = {}
|
|
text = _call_get_document(
|
|
_client_recording(seen), document_id=1, response_format="json"
|
|
)
|
|
|
|
assert seen["path"] == "/api/v1/documents/1"
|
|
assert seen["accept"] == "application/json"
|
|
assert '"id": 1' in text
|