35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
from fastmcp import Client
|
|
from mcp_types import TextContent
|
|
from mcp_types import TextResourceContents
|
|
|
|
|
|
async def assert_server_contract(client: Client) -> None:
|
|
resources = {str(resource.uri) for resource in await client.list_resources()}
|
|
templates = {str(template.uri_template) for template in await client.list_resource_templates()}
|
|
prompts = {prompt.name for prompt in await client.list_prompts()}
|
|
tools = {tool.name for tool in await client.list_tools()}
|
|
|
|
assert tools == {"get_prompt", "list_prompts", "list_resources", "read_resource"}
|
|
assert "skill://pytesting/SKILL.md" in resources
|
|
assert "resource://docs/{path*}" in templates
|
|
assert "skill://pytesting/{path*}" in templates
|
|
assert "pytest-fill-scaffold" in prompts
|
|
|
|
skill_content = await client.read_resource("skill://pytesting/SKILL.md")
|
|
docs_content = await client.read_resource("resource://docs/index.md")
|
|
|
|
assert isinstance(skill_content[0], TextResourceContents)
|
|
assert "# Pytesting" in skill_content[0].text
|
|
assert isinstance(docs_content[0], TextResourceContents)
|
|
assert '"format": "markdown"' in docs_content[0].text
|
|
|
|
listed = await client.call_tool("list_resources")
|
|
listed_content = listed.content[0]
|
|
assert isinstance(listed_content, TextContent)
|
|
assert "skill://pytesting/SKILL.md" in listed_content.text
|
|
|
|
read = await client.call_tool("read_resource", {"uri": "skill://pytesting/SKILL.md"})
|
|
read_content = read.content[0]
|
|
assert isinstance(read_content, TextContent)
|
|
assert "# Pytesting" in read_content.text
|