Skip to content

Commit a2db43a

Browse files
committed
Adding unit tests
1 parent 754e364 commit a2db43a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/mcp/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ async def read_resource(self, uri: AnyUrl) -> ReadResourceResult:
137137
if resource.uri == uri:
138138
return ReadResourceResult(**resource.model_dump(), contents=[])
139139

140-
raise KeyError
140+
raise KeyError(f"Resource {uri} not found")
141141

142142
def add_resource(self, uri: AnyUrl, name: str, description: str | None = None):
143143
"""Add a resource to the fake server"""

tests/mcp/test_resources_server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,6 @@ async def test_read_resource_not_found():
4040
"""Test getting a resource that doesn't exist"""
4141
server = FakeMCPServer()
4242

43-
uri = AnyUrl("docs://api/reference")
43+
uri = "docs://api/reference"
4444
with pytest.raises(KeyError, match=f"Resource {uri} not found"):
45-
await server.read_resource(uri)
45+
await server.read_resource(AnyUrl(uri))

0 commit comments

Comments
 (0)