Skip to content

Commit 900548c

Browse files
committed
Fix tests
1 parent 44bb310 commit 900548c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/issues/test_malformed_input.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import anyio
55
import pytest
66

7+
from mcp.server.models import InitializationOptions
78
from mcp.server.session import ServerSession
89
from mcp.shared.message import SessionMessage
910
from mcp.types import (
@@ -39,7 +40,7 @@ async def test_malformed_initialize_request_does_not_crash_server():
3940
async with ServerSession(
4041
read_stream=read_receive_stream,
4142
write_stream=write_send_stream,
42-
initialization_options={},
43+
init_options=InitializationOptions(),
4344
):
4445
# Send the malformed request
4546
await read_send_stream.send(request_message)
@@ -99,7 +100,7 @@ async def test_multiple_concurrent_malformed_requests():
99100
async with ServerSession(
100101
read_stream=read_receive_stream,
101102
write_stream=write_send_stream,
102-
initialization_options={},
103+
init_options=InitializationOptions(),
103104
):
104105
# Send multiple malformed requests concurrently
105106
malformed_requests = []

0 commit comments

Comments
 (0)