Spaces:
Running
Running
Commit ·
1b6e518
1
Parent(s): 67afc4b
Fix pre-commit ruff-format errors.
Browse files- tests/test_examples.py +12 -7
tests/test_examples.py
CHANGED
|
@@ -6,10 +6,10 @@ from mcp.shared.memory import (
|
|
| 6 |
)
|
| 7 |
from mcp.types import (
|
| 8 |
PromptMessage,
|
| 9 |
-
ReadResourceResult,
|
| 10 |
TextContent,
|
| 11 |
TextResourceContents,
|
| 12 |
)
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
@pytest.mark.anyio
|
|
@@ -63,12 +63,14 @@ async def test_desktop(monkeypatch):
|
|
| 63 |
assert content.text == "3"
|
| 64 |
|
| 65 |
async with client_session(mcp._mcp_server) as client:
|
| 66 |
-
result = await client.read_resource("greeting://rooter12")
|
| 67 |
assert len(result.contents) == 1
|
| 68 |
content = result.contents[0]
|
| 69 |
-
assert isinstance(
|
|
|
|
| 70 |
assert content.text == "Hello, rooter12!"
|
| 71 |
|
|
|
|
| 72 |
# Test the desktop resource
|
| 73 |
result = await client.read_resource(AnyUrl("dir://desktop"))
|
| 74 |
assert len(result.contents) == 1
|
|
@@ -92,17 +94,19 @@ async def test_echo():
|
|
| 92 |
assert content.text == "hello"
|
| 93 |
|
| 94 |
async with client_session(mcp._mcp_server) as client:
|
| 95 |
-
result = await client.read_resource("echo://static")
|
| 96 |
assert len(result.contents) == 1
|
| 97 |
content = result.contents[0]
|
| 98 |
-
assert isinstance(
|
|
|
|
| 99 |
assert content.text == "Echo!"
|
| 100 |
|
| 101 |
async with client_session(mcp._mcp_server) as client:
|
| 102 |
-
result = await client.read_resource("echo://server42")
|
| 103 |
assert len(result.contents) == 1
|
| 104 |
content = result.contents[0]
|
| 105 |
-
assert isinstance(
|
|
|
|
| 106 |
assert content.text == "Echo: server42"
|
| 107 |
|
| 108 |
async with client_session(mcp._mcp_server) as client:
|
|
@@ -110,4 +114,5 @@ async def test_echo():
|
|
| 110 |
assert len(result.messages) == 1
|
| 111 |
assert isinstance(content, TextResourceContents)
|
| 112 |
assert isinstance(result.messages[0], PromptMessage)
|
|
|
|
| 113 |
assert result.messages[0].content.text == "hello"
|
|
|
|
| 6 |
)
|
| 7 |
from mcp.types import (
|
| 8 |
PromptMessage,
|
|
|
|
| 9 |
TextContent,
|
| 10 |
TextResourceContents,
|
| 11 |
)
|
| 12 |
+
from pydantic import AnyUrl
|
| 13 |
|
| 14 |
|
| 15 |
@pytest.mark.anyio
|
|
|
|
| 63 |
assert content.text == "3"
|
| 64 |
|
| 65 |
async with client_session(mcp._mcp_server) as client:
|
| 66 |
+
result = await client.read_resource(AnyUrl("greeting://rooter12"))
|
| 67 |
assert len(result.contents) == 1
|
| 68 |
content = result.contents[0]
|
| 69 |
+
assert isinstance(content, TextResourceContents)
|
| 70 |
+
assert isinstance(content.text, str)
|
| 71 |
assert content.text == "Hello, rooter12!"
|
| 72 |
|
| 73 |
+
async with client_session(mcp._mcp_server) as client:
|
| 74 |
# Test the desktop resource
|
| 75 |
result = await client.read_resource(AnyUrl("dir://desktop"))
|
| 76 |
assert len(result.contents) == 1
|
|
|
|
| 94 |
assert content.text == "hello"
|
| 95 |
|
| 96 |
async with client_session(mcp._mcp_server) as client:
|
| 97 |
+
result = await client.read_resource(AnyUrl("echo://static"))
|
| 98 |
assert len(result.contents) == 1
|
| 99 |
content = result.contents[0]
|
| 100 |
+
assert isinstance(content, TextResourceContents)
|
| 101 |
+
assert isinstance(content.text, str)
|
| 102 |
assert content.text == "Echo!"
|
| 103 |
|
| 104 |
async with client_session(mcp._mcp_server) as client:
|
| 105 |
+
result = await client.read_resource(AnyUrl("echo://server42"))
|
| 106 |
assert len(result.contents) == 1
|
| 107 |
content = result.contents[0]
|
| 108 |
+
assert isinstance(content, TextResourceContents)
|
| 109 |
+
assert isinstance(content.text, str)
|
| 110 |
assert content.text == "Echo: server42"
|
| 111 |
|
| 112 |
async with client_session(mcp._mcp_server) as client:
|
|
|
|
| 114 |
assert len(result.messages) == 1
|
| 115 |
assert isinstance(content, TextResourceContents)
|
| 116 |
assert isinstance(result.messages[0], PromptMessage)
|
| 117 |
+
assert isinstance(result.messages[0].content, TextContent)
|
| 118 |
assert result.messages[0].content.text == "hello"
|