Skip to content

Commit ad04a3b

Browse files
committed
fix linter
1 parent 02d1bf7 commit ad04a3b

1 file changed

Lines changed: 15 additions & 9 deletions

File tree

tests/integration/test_client_server_integration.py

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)