scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test: capture audio handle through mock call

Browse source
This commit is contained in:
Charan Rathore 2026-09-28 09:42:31 +05:30
parent 8345d35278
commit db16d964c9
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/unit/litellm_core_utils/test_health_check_helpers.py
View file

@ -40,11 +40,8 @@ def _distinct_rgb_colors(png: bytes) -> set[bytes]:
@pytest.mark.asyncio
@pytest.mark.parametrize("raises", [False, True])
async def test_audio_transcription_health_check_closes_file(raises: bool):
file_handles = []
async def transcription(**kwargs):
file = kwargs["file"]
file_handles.append(file)
assert not file.closed
assert file.read(4) == b"RIFF"
if raises:
@ -56,15 +53,15 @@ async def test_audio_transcription_health_check_closes_file(raises: bool):
custom_llm_provider="openai",
model_params={"model": "openai/whisper-1", "api_key": "sk-test"},
)
with patch("litellm.atranscription", side_effect=transcription):
with patch("litellm.atranscription", side_effect=transcription) as mock_transcription:
if raises:
with pytest.raises(RuntimeError, match="transcription failed"):
await handlers["audio_transcription"]()
else:
assert await handlers["audio_transcription"]() == {"text": "healthy"}
assert len(file_handles) == 1
assert file_handles[0].closed
mock_transcription.assert_called_once()
assert mock_transcription.call_args.kwargs["file"].closed
@pytest.mark.asyncio