Skip to content

Commit 47e240c

Browse files
committed
Fix docs_src transport test branch coverage for CI.
Avoid a partial-covered POST-only branch and match the usual session_manager pragma so fail_under=100 passes.
1 parent 45951d4 commit 47e240c

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

tests/docs_src/test_client_transports.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -70,16 +70,15 @@ def search_books(query: str) -> str:
7070
"""Search the catalog by title or author."""
7171
return f"Found 3 books matching {query!r}."
7272

73-
seen: list[tuple[str | None, str | None]] = []
73+
seen: list[tuple[str, str | None, str | None]] = []
7474

7575
async def record_headers(request: httpx2.Request) -> None:
7676
await tutorial005.inject_request_headers(request)
77-
if request.method == "POST":
78-
seen.append((request.headers.get("Authorization"), request.headers.get("X-Trace-ID")))
77+
seen.append((request.method, request.headers.get("Authorization"), request.headers.get("X-Trace-ID")))
7978

8079
url = "http://127.0.0.1:8000/mcp"
8180
transport = httpx2.ASGITransport(app=mcp.streamable_http_app())
82-
async with mcp.session_manager.run():
81+
async with mcp.session_manager.run(): # pragma: no branch
8382
async with (
8483
httpx2.AsyncClient(
8584
transport=transport,
@@ -98,5 +97,6 @@ async def record_headers(request: httpx2.Request) -> None:
9897

9998
assert first.structured_content == {"result": "Found 3 books matching 'dune'."}
10099
assert second.structured_content == {"result": "Found 3 books matching 'neuromancer'."}
101-
assert ("Bearer user-123-token", "trace-abc") in seen
102-
assert ("Bearer user-456-token", "trace-def") in seen
100+
post_headers = [(auth, trace) for method, auth, trace in seen if method == "POST"]
101+
assert ("Bearer user-123-token", "trace-abc") in post_headers
102+
assert ("Bearer user-456-token", "trace-def") in post_headers

0 commit comments

Comments
 (0)