Fix unit tests

Author: amri369

tests/mcp/helpers.py

@@ -3,18 +3,18 @@
 import shutil
 from typing import Any
 
-from pydantic import AnyUrl
-
 from mcp import Tool as MCPTool
 from mcp.types import (
     CallToolResult,
     GetPromptResult,
     ListPromptsResult,
     ListResourcesResult,
     ListResourceTemplatesResult,
+    PromptMessage,
     ReadResourceResult,
-    TextContent
+    TextContent,
 )
+from pydantic import AnyUrl
 
 from agents.mcp import MCPServer
 from agents.mcp.server import _MCPServerWithClientSession
@@ -120,7 +120,8 @@ async def list_resources(self, run_context=None, agent=None) -> ListResourcesRes
         """Return empty list of resources for fake server"""
         return ListResourcesResult(resources=[])
 
-    async def list_resource_templates(self, run_context=None, agent=None) -> ListResourceTemplatesResult:
+    async def list_resource_templates(self, run_context=None, agent=None) \
+            -> ListResourceTemplatesResult:
         """Return empty list of resources templates for fake server"""
         return ListResourceTemplatesResult(resourceTemplates=[])
 

tests/mcp/test_prompt_server.py

@@ -1,6 +1,8 @@
 from typing import Any
 
 import pytest
+from mcp.types import ListResourcesResult, ListResourceTemplatesResult, ReadResourceResult
+from pydantic import AnyUrl
 
 from agents import Agent, Runner
 from agents.mcp import MCPServer
@@ -66,6 +68,19 @@ async def list_tools(self, run_context=None, agent=None):
     async def call_tool(self, tool_name: str, arguments: dict[str, Any] | None = None):
         raise NotImplementedError("This fake server doesn't support tools")
 
+    async def list_resources(self, run_context=None, agent=None) -> ListResourcesResult:
+        """Return empty list of resources for fake server"""
+        return ListResourcesResult(resources=[])
+
+    async def list_resource_templates(self, run_context=None, agent=None) \
+            -> ListResourceTemplatesResult:
+        """Return empty list of resources templates for fake server"""
+        return ListResourceTemplatesResult(resourceTemplates=[])
+
+    async def read_resource(self, uri: AnyUrl) -> ReadResourceResult:
+        """Return a fake resource read for fake server"""
+        return ReadResourceResult(contents=[])
+
     @property
     def name(self) -> str:
         return self._server_name