modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Harden the dual-era stream loop's era-lock and rejection semantics (#3040)

Browse source
This commit is contained in:
Max 2026-07-01 17:07:12 +01:00 • committed by GitHub
commit 080f2a869d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 652 additions and 125 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/server/test_stdio.py
View file

@ -16,6 +16,7 @@ from mcp_types import (
JSONRPCResponse,
jsonrpc_message_adapter,
)
from typing_extensions import Buffer
from mcp.server.mcpserver import MCPServer
from mcp.server.stdio import stdio_server
@ -123,10 +124,11 @@ class _GatedStdin(io.RawIOBase):
def readable(self) -> bool:
return True
def readinto(self, b: bytearray | memoryview) -> int: # pyright: ignore[reportIncompatibleMethodOverride]
def readinto(self, b: Buffer) -> int:
view = memoryview(b)
if self._pending:
n = min(len(b), len(self._pending))
b[:n] = self._pending[:n]
n = min(len(view), len(self._pending))
view[:n] = self._pending[:n]
self._pending = self._pending[n:]
return n
# A missed release falls through to EOF after the bound; the caller's
@ -155,7 +157,7 @@ class _NotifyingStdout(io.RawIOBase):
def writable(self) -> bool:
return True
def write(self, b: bytes | bytearray | memoryview) -> int: # pyright: ignore[reportIncompatibleMethodOverride]
def write(self, b: Buffer) -> int:
data = bytes(b)
with self._cond:
self._chunks.append(data)