2121from mcp .client .sse import sse_client
2222from mcp .server import Server
2323from mcp .server .sse import SseServerTransport
24- from mcp .server .transport_security import TransportSecuritySettings
2524from mcp .server .streaming_asgi_transport import StreamingASGITransport
25+ from mcp .server .transport_security import TransportSecuritySettings
2626from mcp .shared .exceptions import McpError
2727from mcp .types import (
2828 EmptyResult ,
@@ -418,7 +418,10 @@ def create_test_client(
418418 follow_redirects = True ,
419419 )
420420
421- async with sse_client ("http://testserver/sse" , headers = custom_headers , httpx_client_factory = create_test_client ) as (
421+ async with sse_client ("http://testserver/sse" ,
422+ headers = custom_headers ,
423+ httpx_client_factory = create_test_client ,
424+ sse_read_timeout = 0.5 ) as (
422425 read_stream ,
423426 write_stream ,
424427 ):
@@ -432,7 +435,8 @@ def create_test_client(
432435
433436 # Parse the JSON response
434437 assert len (tool_result .content ) == 1
435- headers_data = json .loads (tool_result .content [0 ].text if tool_result .content [0 ].type == "text" else "{}" )
438+ content_item = tool_result .content [0 ]
439+ headers_data = json .loads (content_item .text if content_item .type == "text" else "{}" )
436440
437441 # Verify headers were propagated
438442 assert headers_data .get ("authorization" ) == "Bearer test-token"
0 commit comments