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

Get baseline 100% clean coverage (#1553)

Browse source
This commit is contained in:
Max Isbey 2025-11-11 13:09:32 +00:00 • committed by GitHub
commit 89e9c43acf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1032 additions and 1084 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/client/test_stdio.py
View file

@ -54,7 +54,7 @@ async def test_stdio_client():
read_messages: list[JSONRPCMessage] = []
async with read_stream:
async for message in read_stream:
if isinstance(message, Exception):
if isinstance(message, Exception): # pragma: no cover
raise message
read_messages.append(message.message)
@ -93,7 +93,7 @@ async def test_stdio_client_nonexistent_command():
# Should raise an error when trying to start the process
with pytest.raises(OSError) as exc_info:
async with stdio_client(server_params) as (_, _):
pass
pass # pragma: no cover
# The error should indicate the command was not found (ENOENT: No such file or directory)
assert exc_info.value.errno == errno.ENOENT
@ -144,7 +144,7 @@ async def test_stdio_client_universal_cleanup():
)
# Check if we timed out
if cancel_scope.cancelled_caught:
if cancel_scope.cancelled_caught: # pragma: no cover
pytest.fail(
"stdio_client cleanup timed out after 8.0 seconds. "
"This indicates the cleanup mechanism is hanging and needs fixing."
@ -153,7 +153,7 @@ async def test_stdio_client_universal_cleanup():
@pytest.mark.anyio
@pytest.mark.skipif(sys.platform == "win32", reason="Windows signal handling is different")
async def test_stdio_client_sigint_only_process():
async def test_stdio_client_sigint_only_process(): # pragma: no cover
"""
Test cleanup with a process that ignores SIGTERM but responds to SIGINT.
"""
@ -195,7 +195,7 @@ async def test_stdio_client_sigint_only_process():
# Exit context triggers cleanup - this should not hang
pass
if cancel_scope.cancelled_caught:
if cancel_scope.cancelled_caught: # pragma: no cover
raise TimeoutError("Test timed out")
end_time = time.time()
@ -208,7 +208,7 @@ async def test_stdio_client_sigint_only_process():
f"Expected < {SIGTERM_IGNORING_PROCESS_TIMEOUT} seconds. "
"This suggests the cleanup needs SIGINT/SIGKILL fallback."
)
except (TimeoutError, Exception) as e:
except (TimeoutError, Exception) as e: # pragma: no cover
if isinstance(e, TimeoutError) or "timed out" in str(e):
pytest.fail(
f"stdio_client cleanup timed out after {SIGTERM_IGNORING_PROCESS_TIMEOUT} seconds "
@ -303,7 +303,7 @@ class TestChildProcessCleanup:
assert os.path.exists(parent_marker), "Parent process didn't start"
# Verify child is writing
if os.path.exists(marker_file):
if os.path.exists(marker_file): # pragma: no branch
initial_size = os.path.getsize(marker_file)
await anyio.sleep(0.3)
size_after_wait = os.path.getsize(marker_file)
@ -318,7 +318,7 @@ class TestChildProcessCleanup:
# Verify processes stopped
await anyio.sleep(0.5)
if os.path.exists(marker_file):
if os.path.exists(marker_file): # pragma: no branch
size_after_cleanup = os.path.getsize(marker_file)
await anyio.sleep(0.5)
final_size = os.path.getsize(marker_file)
@ -335,7 +335,7 @@ class TestChildProcessCleanup:
for f in [marker_file, parent_marker]:
try:
os.unlink(f)
except OSError:
except OSError: # pragma: no cover
pass
@pytest.mark.anyio
@ -406,7 +406,7 @@ class TestChildProcessCleanup:
# Verify all are writing
for file_path, name in [(parent_file, "parent"), (child_file, "child"), (grandchild_file, "grandchild")]:
if os.path.exists(file_path):
if os.path.exists(file_path): # pragma: no branch
initial_size = os.path.getsize(file_path)
await anyio.sleep(0.3)
new_size = os.path.getsize(file_path)
@ -420,7 +420,7 @@ class TestChildProcessCleanup:
# Verify all stopped
await anyio.sleep(0.5)
for file_path, name in [(parent_file, "parent"), (child_file, "child"), (grandchild_file, "grandchild")]:
if os.path.exists(file_path):
if os.path.exists(file_path): # pragma: no branch
size1 = os.path.getsize(file_path)
await anyio.sleep(0.3)
size2 = os.path.getsize(file_path)
@ -433,7 +433,7 @@ class TestChildProcessCleanup:
for f in [parent_file, child_file, grandchild_file]:
try:
os.unlink(f)
except OSError:
except OSError: # pragma: no cover
pass
@pytest.mark.anyio
@ -487,7 +487,7 @@ class TestChildProcessCleanup:
await anyio.sleep(0.5)
# Verify child is writing
if os.path.exists(marker_file):
if os.path.exists(marker_file): # pragma: no cover
size1 = os.path.getsize(marker_file)
await anyio.sleep(0.3)
size2 = os.path.getsize(marker_file)
@ -500,7 +500,7 @@ class TestChildProcessCleanup:
# Verify child stopped
await anyio.sleep(0.5)
if os.path.exists(marker_file):
if os.path.exists(marker_file): # pragma: no branch
size3 = os.path.getsize(marker_file)
await anyio.sleep(0.3)
size4 = os.path.getsize(marker_file)
@ -512,7 +512,7 @@ class TestChildProcessCleanup:
# Clean up marker file
try:
os.unlink(marker_file)
except OSError:
except OSError: # pragma: no cover
pass
@ -560,7 +560,7 @@ async def test_stdio_client_graceful_stdin_exit():
pytest.fail(
"stdio_client cleanup timed out after 5.0 seconds. "
"Process should have exited gracefully when stdin was closed."
)
) # pragma: no cover
end_time = time.time()
elapsed = end_time - start_time
@ -619,7 +619,7 @@ async def test_stdio_client_stdin_close_ignored():
pytest.fail(
"stdio_client cleanup timed out after 7.0 seconds. "
"Process should have been terminated via SIGTERM escalation."
)
) # pragma: no cover
end_time = time.time()
elapsed = end_time - start_time