Skip to content

Commit 107b584

Browse files
committed
Adding unit tests
1 parent c654f20 commit 107b584

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

tests/mcp/helpers.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,10 @@ def __init__(
7575
tool_filter: ToolFilter = None,
7676
server_name: str = "fake_mcp_server",
7777
):
78-
if not resources:
79-
resources = ListResourcesResult(resources=[])
80-
if not resources_templates:
81-
resources_templates = ListResourceTemplatesResult(resourceTemplates=[])
82-
self.resources = resources
83-
self.resources_templates = resources_templates
78+
self.resources = (resources or
79+
ListResourcesResult(resources=[]))
80+
self.resources_templates = (resources_templates or
81+
ListResourceTemplatesResult(resourceTemplates=[]))
8482
self.tools: list[MCPTool] = tools or []
8583
self.tool_calls: list[str] = []
8684
self.tool_results: list[str] = []
@@ -139,11 +137,13 @@ async def read_resource(self, uri: AnyUrl) -> ReadResourceResult:
139137
"""Return a fake resource read for fake server"""
140138
return ReadResourceResult(contents=[])
141139

142-
def add_resource(self, uri: AnyUrl, description: str | None = None):
143-
self.resources.append(Resource(uri=uri, description=description))
140+
def add_resource(self, uri: AnyUrl, name: str, description: str | None = None):
141+
self.resources.resources.append(Resource(uri=uri, description=description, name=name))
144142

145-
def add_resource_template(self, uri: AnyUrl, description: str | None = None):
146-
self.resources_templates.append(ResourceTemplate(uriTemplate=uri, description=description))
143+
def add_resource_template(self, uri: AnyUrl, name: str, description: str | None = None):
144+
self.resources_templates.resourceTemplates.append(
145+
ResourceTemplate(uriTemplate=uri, description=description, name=name)
146+
)
147147

148148
@property
149149
def name(self) -> str:

0 commit comments

Comments
 (0)