@@ -368,12 +368,13 @@ def context_server(server_port: int) -> Generator[None, None, None]:
368368 if proc .is_alive ():
369369 print ("context server process failed to terminate" )
370370
371+
371372@pytest .fixture ()
372373async def context_app () -> Starlette :
373374 """Fixture that provides the context server app"""
374375 security_settings = TransportSecuritySettings (
375- allowed_hosts = ["127.0.0.1:*" , "localhost:*" , "testserver" ],
376- allowed_origins = ["http://127.0.0.1:*" , "http://localhost:*" , "http://testserver" ]
376+ allowed_hosts = ["127.0.0.1:*" , "localhost:*" , "testserver" ],
377+ allowed_origins = ["http://127.0.0.1:*" , "http://localhost:*" , "http://testserver" ],
377378 )
378379 sse = SseServerTransport ("/messages/" , security_settings = security_settings )
379380 context_server = RequestContextServer ()
@@ -403,6 +404,7 @@ async def test_request_context_propagation(context_app: Starlette) -> None:
403404 }
404405
405406 async with anyio .create_task_group () as tg :
407+
406408 def create_test_client (
407409 headers : dict [str , str ] | None = None ,
408410 timeout : httpx .Timeout | None = None ,
@@ -418,10 +420,12 @@ def create_test_client(
418420 follow_redirects = True ,
419421 )
420422
421- async with sse_client ("http://testserver/sse" ,
422- headers = custom_headers ,
423- httpx_client_factory = create_test_client ,
424- sse_read_timeout = 0.5 ) as (
423+ async with sse_client (
424+ "http://testserver/sse" ,
425+ headers = custom_headers ,
426+ httpx_client_factory = create_test_client ,
427+ sse_read_timeout = 0.5 ,
428+ ) as (
425429 read_stream ,
426430 write_stream ,
427431 ):
0 commit comments