af80e3a971
Per https://modelcontextprotocol.io/specification/draft/basic/lifecycle#timeouts "Implementations SHOULD establish timeouts for all sent requests, to prevent hung connections and resource exhaustion. When the request has not received a success or error response within the timeout period, the sender SHOULD issue a cancellation notification for that request and stop waiting for a response. SDKs and other middleware SHOULD allow these timeouts to be configured on a per-request basis." I picked 5 seconds since that's the default for SSE
42 lines
1 KiB
Python
42 lines
1 KiB
Python
import pytest
|
|
|
|
from agents.exceptions import UserError
|
|
from agents.mcp.server import _MCPServerWithClientSession
|
|
|
|
|
|
class CrashingClientSessionServer(_MCPServerWithClientSession):
|
|
def __init__(self):
|
|
super().__init__(cache_tools_list=False, client_session_timeout_seconds=5)
|
|
self.cleanup_called = False
|
|
|
|
def create_streams(self):
|
|
raise ValueError("Crash!")
|
|
|
|
async def cleanup(self):
|
|
self.cleanup_called = True
|
|
await super().cleanup()
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "crashing_client_session_server"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_server_errors_cause_error_and_cleanup_called():
|
|
server = CrashingClientSessionServer()
|
|
|
|
with pytest.raises(ValueError):
|
|
await server.connect()
|
|
|
|
assert server.cleanup_called
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_not_calling_connect_causes_error():
|
|
server = CrashingClientSessionServer()
|
|
|
|
with pytest.raises(UserError):
|
|
await server.list_tools()
|
|
|
|
with pytest.raises(UserError):
|
|
await server.call_tool("foo", {})
|