@@ -93,40 +93,6 @@ def get_mcp_command_payload(method: str, params, request_id: str):
9393 )
9494
9595
96- async def stdio (server , method : str , params , request_id : str ):
97- read_stream_writer , read_stream = anyio .create_memory_object_stream (0 )
98- write_stream , write_stream_reader = anyio .create_memory_object_stream (0 )
99-
100- result = {}
101-
102- async def run_server ():
103- await server .run (
104- read_stream , write_stream , server .create_initialization_options ()
105- )
106-
107- async def simulate_client (tg , result ):
108- init_request = get_initialization_payload ("1" )
109- await read_stream_writer .send (init_request )
110-
111- await write_stream_reader .receive ()
112-
113- initialized_notification = get_initialized_notification_payload ()
114- await read_stream_writer .send (initialized_notification )
115-
116- request = get_mcp_command_payload (method , params = params , request_id = request_id )
117- await read_stream_writer .send (request )
118-
119- result ["response" ] = await write_stream_reader .receive ()
120-
121- tg .cancel_scope .cancel ()
122-
123- async with anyio .create_task_group () as tg :
124- tg .start_soon (run_server )
125- tg .start_soon (simulate_client , tg , result )
126-
127- return result ["response" ]
128-
129-
13096@pytest .fixture (autouse = True )
13197def reset_request_ctx ():
13298 """Reset request context before and after each test"""
0 commit comments