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

test: fix test params and improve get recovery test

Browse source 
Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-02-26 17:58:26 +00:00
parent a06404451d
commit e4b2f62758
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/test_litellm/llms/custom_httpx/test_httpx_client_closed_error.py
View file

@ -166,16 +166,12 @@ class TestGetSyncHttpxClientClosedCheck:
try:
litellm.in_memory_llm_clients_cache = cache
client_1 = _get_httpx_client(
params={"client_alias": "test_closed_sync"},
)
client_1 = _get_httpx_client()
client_1.client.close()
assert client_1.client.is_closed is True
client_2 = _get_httpx_client(
params={"client_alias": "test_closed_sync"},
)
client_2 = _get_httpx_client()
assert client_2 is not client_1
assert client_2.client.is_closed is False
@ -217,11 +213,18 @@ class TestAsyncHTTPHandlerClosedClientRecovery:
assert handler.client.is_closed is True
mock_response = httpx.Response(200, text="ok")
with patch.object(
httpx.AsyncClient,
"get",
return_value=mock_response,
):
call_count = 0
original_get = httpx.AsyncClient.get
async def side_effect(self_client, *args, **kwargs):
nonlocal call_count
call_count += 1
if self_client.is_closed:
raise RuntimeError("Cannot send a request, as the client has been closed.")
return mock_response
with patch.object(httpx.AsyncClient, "get", side_effect):
response = await handler.get(url="https://example.com/api")
assert response.status_code == 200