scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

fix(test): use _mounts length to detect custom transport in proxy tests

Browse source 
httpx always wraps the pool in HTTPTransport regardless of whether a
custom transport was passed, so isinstance checks cannot distinguish
them. Use _mounts instead: custom transport produces an empty dict,
while httpx's default mount system is non-empty.
This commit is contained in:
z 2026-05-09 01:03:09 +08:00
parent 81e79f62c6
commit 0015564fc1
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/test_litellm/llms/custom_httpx/test_http_handler.py
View file

@ -684,27 +684,29 @@ def test_sync_http_handler_skips_custom_transport_when_proxy_set(monkeypatch):
httpx ignores proxy env vars when an explicit transport is provided,
which breaks sync-path providers (e.g. ChatGPT authenticator) behind
a proxy. See GitHub issue #24562.
We check _mounts: a custom transport produces an empty dict,
while httpx's default mount system produces a non-empty dict.
"""
original_force_ipv4 = litellm.force_ipv4
litellm.force_ipv4 = True
try:
# With proxy: custom transport should be dropped
# With proxy: custom transport should be dropped, so httpx
# builds its default mount system (non-empty _mounts).
monkeypatch.setenv("HTTPS_PROXY", "http://proxy:8080")
handler = HTTPHandler()
try:
transport = handler.client._transport
assert not isinstance(transport, httpx.HTTPTransport)
assert len(handler.client._mounts) > 0
finally:
handler.close()
# Without proxy: custom transport should be applied (force_ipv4)
# Without proxy: custom transport is used (empty _mounts).
monkeypatch.delenv("HTTPS_PROXY", raising=False)
monkeypatch.delenv("HTTP_PROXY", raising=False)
handler2 = HTTPHandler()
try:
transport2 = handler2.client._transport
assert isinstance(transport2, httpx.HTTPTransport)
assert len(handler2.client._mounts) == 0
finally:
handler2.close()
finally:
@ -725,8 +727,7 @@ def test_sync_http_handler_prefers_https_proxy_over_http_proxy(monkeypatch):
handler = HTTPHandler()
try:
# Should not have custom transport (proxy takes priority)
transport = handler.client._transport
assert not isinstance(transport, httpx.HTTPTransport)
assert len(handler.client._mounts) > 0
finally:
handler.close()
finally: