ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-19 03:54:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix pre-commit ruff-format errors.

Browse source
This commit is contained in:
Alain IVARS 2025-05-03 15:48:10 +01:00
commit 09bbdb6178
1 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_examples.py
View file

@ -6,10 +6,10 @@ from mcp.shared.memory import (
)
from mcp.types import (
PromptMessage,
ReadResourceResult,
TextContent,
TextResourceContents,
)
from pydantic import AnyUrl
@pytest.mark.anyio
@ -63,12 +63,14 @@ async def test_desktop(monkeypatch):
assert content.text == "3"
async with client_session(mcp._mcp_server) as client:
result = await client.read_resource("greeting://rooter12")
result = await client.read_resource(AnyUrl("greeting://rooter12"))
assert len(result.contents) == 1
content = result.contents[0]
assert isinstance(result, ReadResourceResult)
assert isinstance(content, TextResourceContents)
assert isinstance(content.text, str)
assert content.text == "Hello, rooter12!"
async with client_session(mcp._mcp_server) as client:
# Test the desktop resource
result = await client.read_resource(AnyUrl("dir://desktop"))
assert len(result.contents) == 1
@ -92,17 +94,19 @@ async def test_echo():
assert content.text == "hello"
async with client_session(mcp._mcp_server) as client:
result = await client.read_resource("echo://static")
result = await client.read_resource(AnyUrl("echo://static"))
assert len(result.contents) == 1
content = result.contents[0]
assert isinstance(result, ReadResourceResult)
assert isinstance(content, TextResourceContents)
assert isinstance(content.text, str)
assert content.text == "Echo!"
async with client_session(mcp._mcp_server) as client:
result = await client.read_resource("echo://server42")
result = await client.read_resource(AnyUrl("echo://server42"))
assert len(result.contents) == 1
content = result.contents[0]
assert isinstance(result, ReadResourceResult)
assert isinstance(content, TextResourceContents)
assert isinstance(content.text, str)
assert content.text == "Echo: server42"
async with client_session(mcp._mcp_server) as client:
@ -110,4 +114,5 @@ async def test_echo():
assert len(result.messages) == 1
assert isinstance(content, TextResourceContents)
assert isinstance(result.messages[0], PromptMessage)
assert isinstance(result.messages[0].content, TextContent)
assert result.messages[0].content.text == "hello"