@@ -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