@@ -1046,34 +1046,40 @@ async def mock_generator(*args, **kwargs):
10461046 getattr (handler , handler_attr ).side_effect = mock_generator
10471047
10481048 transport = client ._transport
1049-
1049+
10501050 if isinstance (transport , (RestTransport , JsonRpcTransport )):
10511051 # Spy on httpx client to check response headers
10521052 original_send = transport .httpx_client .send
10531053 response_headers = {}
1054-
1054+
10551055 async def mock_send (* args , ** kwargs ):
10561056 resp = await original_send (* args , ** kwargs )
10571057 response_headers ['Content-Type' ] = resp .headers .get ('Content-Type' )
10581058 return resp
1059-
1059+
10601060 transport .httpx_client .send = mock_send
1061-
1061+
10621062 try :
10631063 with pytest .raises (error_cls ):
1064- async for _ in getattr (client , client_method )(request = request_params ):
1064+ async for _ in getattr (client , client_method )(
1065+ request = request_params
1066+ ):
10651067 pass
10661068 finally :
10671069 transport .httpx_client .send = original_send
1068-
1070+
10691071 # Verify that the response content type was NOT text/event-stream
1070- assert not response_headers .get ('Content-Type' , '' ).startswith ('text/event-stream' )
1072+ assert not response_headers .get ('Content-Type' , '' ).startswith (
1073+ 'text/event-stream'
1074+ )
10711075 else :
10721076 # For gRPC, we just verify it raises the error
10731077 with pytest .raises (error_cls ):
1074- async for _ in getattr (client , client_method )(request = request_params ):
1078+ async for _ in getattr (client , client_method )(
1079+ request = request_params
1080+ ):
10751081 pass
1076-
1082+
10771083 getattr (handler , handler_attr ).side_effect = None
10781084 await client .close ()
10791085
0 commit comments