mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
test fix
This commit is contained in:
parent
5377b17bdd
commit
8cb66168bc
1 changed files with 2 additions and 6 deletions
|
|
@ -1182,10 +1182,8 @@ def test_async_http_handler(mock_async_client):
|
|||
assert call_args["transport"] == mock_transport
|
||||
assert call_args["event_hooks"] == event_hooks
|
||||
assert call_args["headers"] == headers
|
||||
assert isinstance(call_args["limits"], httpx.Limits)
|
||||
assert call_args["limits"].max_connections == concurrent_limit
|
||||
assert call_args["limits"].max_keepalive_connections == concurrent_limit
|
||||
assert call_args["timeout"] == timeout
|
||||
assert call_args["follow_redirects"] is True
|
||||
|
||||
|
||||
@mock.patch("httpx.AsyncClient")
|
||||
|
|
@ -1224,12 +1222,10 @@ def test_async_http_handler_force_ipv4(mock_async_client):
|
|||
# Assert other parameters match
|
||||
assert call_args["event_hooks"] == event_hooks
|
||||
assert call_args["headers"] == headers
|
||||
assert isinstance(call_args["limits"], httpx.Limits)
|
||||
assert call_args["limits"].max_connections == concurrent_limit
|
||||
assert call_args["limits"].max_keepalive_connections == concurrent_limit
|
||||
assert call_args["timeout"] == timeout
|
||||
assert isinstance(call_args["verify"], ssl.SSLContext)
|
||||
assert call_args["cert"] is None
|
||||
assert call_args["follow_redirects"] is True
|
||||
|
||||
finally:
|
||||
# Reset force_ipv4 to default
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue