Felix Weinberger
113 lines
4.3 KiB
Python
113 lines
4.3 KiB
Python
"""Tests for example servers"""
|
|
# TODO(Marcelo): The `examples` directory needs to be importable as a package.
|
|
# pyright: reportMissingImports=false
|
|
# pyright: reportUnknownVariableType=false
|
|
# pyright: reportUnknownArgumentType=false
|
|
# pyright: reportUnknownMemberType=false
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from pydantic import AnyUrl
|
|
from pytest_examples import CodeExample, EvalExample, find_examples
|
|
|
|
from mcp import Client
|
|
from mcp.types import TextContent, TextResourceContents
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_simple_echo():
|
|
"""Test the simple echo server"""
|
|
from examples.fastmcp.simple_echo import mcp
|
|
|
|
async with Client(mcp) as client:
|
|
result = await client.call_tool("echo", {"text": "hello"})
|
|
assert len(result.content) == 1
|
|
content = result.content[0]
|
|
assert isinstance(content, TextContent)
|
|
assert content.text == "hello"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_complex_inputs():
|
|
"""Test the complex inputs server"""
|
|
from examples.fastmcp.complex_inputs import mcp
|
|
|
|
async with Client(mcp) as client:
|
|
tank = {"shrimp": [{"name": "bob"}, {"name": "alice"}]}
|
|
result = await client.call_tool("name_shrimp", {"tank": tank, "extra_names": ["charlie"]})
|
|
assert len(result.content) == 3
|
|
assert isinstance(result.content[0], TextContent)
|
|
assert isinstance(result.content[1], TextContent)
|
|
assert isinstance(result.content[2], TextContent)
|
|
assert result.content[0].text == "bob"
|
|
assert result.content[1].text == "alice"
|
|
assert result.content[2].text == "charlie"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_direct_call_tool_result_return():
|
|
"""Test the CallToolResult echo server"""
|
|
from examples.fastmcp.direct_call_tool_result_return import mcp
|
|
|
|
async with Client(mcp) as client:
|
|
result = await client.call_tool("echo", {"text": "hello"})
|
|
assert len(result.content) == 1
|
|
content = result.content[0]
|
|
assert isinstance(content, TextContent)
|
|
assert content.text == "hello"
|
|
assert result.structured_content
|
|
assert result.structured_content["text"] == "hello"
|
|
assert isinstance(result.meta, dict)
|
|
assert result.meta["some"] == "metadata"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_desktop(monkeypatch: pytest.MonkeyPatch):
|
|
"""Test the desktop server"""
|
|
# Mock desktop directory listing
|
|
mock_files = [Path("/fake/path/file1.txt"), Path("/fake/path/file2.txt")]
|
|
monkeypatch.setattr(Path, "iterdir", lambda self: mock_files) # type: ignore[reportUnknownArgumentType]
|
|
monkeypatch.setattr(Path, "home", lambda: Path("/fake/home"))
|
|
|
|
from examples.fastmcp.desktop import mcp
|
|
|
|
async with Client(mcp) as client:
|
|
# Test the sum function
|
|
result = await client.call_tool("sum", {"a": 1, "b": 2})
|
|
assert len(result.content) == 1
|
|
content = result.content[0]
|
|
assert isinstance(content, TextContent)
|
|
assert content.text == "3"
|
|
|
|
# Test the desktop resource
|
|
result = await client.read_resource(AnyUrl("dir://desktop"))
|
|
assert len(result.contents) == 1
|
|
content = result.contents[0]
|
|
assert isinstance(content, TextResourceContents)
|
|
assert isinstance(content.text, str)
|
|
if sys.platform == "win32": # pragma: no cover
|
|
file_1 = "/fake/path/file1.txt".replace("/", "\\\\") # might be a bug
|
|
file_2 = "/fake/path/file2.txt".replace("/", "\\\\") # might be a bug
|
|
assert file_1 in content.text
|
|
assert file_2 in content.text
|
|
# might be a bug, but the test is passing
|
|
else: # pragma: no cover
|
|
assert "/fake/path/file1.txt" in content.text
|
|
assert "/fake/path/file2.txt" in content.text
|
|
|
|
|
|
@pytest.mark.parametrize("example", find_examples("README.md"), ids=str)
|
|
def test_docs_examples(example: CodeExample, eval_example: EvalExample):
|
|
ruff_ignore: list[str] = ["F841", "I001", "F821"] # F821: undefined names (snippets lack imports)
|
|
|
|
# Use project's actual line length of 120
|
|
eval_example.set_config(ruff_ignore=ruff_ignore, target_version="py310", line_length=120)
|
|
|
|
# Use Ruff for both formatting and linting (skip Black)
|
|
if eval_example.update_examples: # pragma: no cover
|
|
eval_example.format_ruff(example)
|
|
else:
|
|
eval_example.lint_ruff(example)
|