scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

fix(test): use patch.object for aiohttp transport test to work in parallel execution

Browse source 
The @patch decorator was not intercepting the static method call in parallel
xdist workers. Using patch.object on the directly-imported class is more reliable.

Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-02-28 06:06:47 +00:00
parent c1a516d994
commit 8ff4529cf2
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/test_litellm/llms/custom_httpx/test_aiohttp_handler.py
View file

@ -337,19 +337,24 @@ class TestBaseLLMAIOHTTPHandler:
mock_client_session.assert_called_once_with()
assert result is mock_session_instance
@patch(
"litellm.llms.custom_httpx.http_handler.AsyncHTTPHandler._create_aiohttp_transport"
)
def test_get_or_create_transport(self, mock_create_transport):
def test_get_or_create_transport(self):
"""Test transport creation when none provided"""
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler
mock_transport_instance = Mock(spec=LiteLLMAiohttpTransport)
mock_create_transport.return_value = mock_transport_instance
handler = BaseLLMAIOHTTPHandler()
result = handler._get_or_create_transport()
# Patch the static method directly on the class object that aiohttp_handler.py uses
with patch.object(
AsyncHTTPHandler,
'_create_aiohttp_transport',
return_value=mock_transport_instance,
) as mock_create_transport:
result = handler._get_or_create_transport()
mock_create_transport.assert_called_once()
mock_create_transport.assert_called_once()
assert result is mock_transport_instance
assert handler.transport is mock_transport_instance
assert handler._owns_transport is True