alainivars commited on
Commit
1b6e518
·
1 Parent(s): 67afc4b

Fix pre-commit ruff-format errors.

Browse files
Files changed (1) hide show
  1. 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(result, ReadResourceResult)
 
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(result, ReadResourceResult)
 
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(result, ReadResourceResult)
 
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"