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(proxy): cover late SSE strings, mappings, and unknown payloads

Browse source 
The string branch, the empty-object fallback, and a mappingproxy were untested.
This commit is contained in:
MengMa 2026-09-27 19:37:20 +08:00
parent 0cc4d66624
commit f75a2b8651
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
tests/test_litellm/proxy/common_utils/test_sse_keepalive.py
View file

@ -445,3 +445,35 @@ async def test_relay_late_response_serializes_a_dict_body_attribute():
assert frames[0] == b'data: {"ok":true}\n\n'
assert frames[1] == b"data: [DONE]\n\n"
@pytest.mark.asyncio
async def test_relay_late_response_encodes_a_string_payload():
from litellm.proxy.common_request_processing import _relay_late_response
frames: Final = [chunk async for chunk in _relay_late_response("late")]
assert frames == [b"data: late\n\n", b"data: [DONE]\n\n"]
@pytest.mark.asyncio
async def test_relay_late_response_uses_an_empty_object_for_an_unknown_payload():
from litellm.proxy.common_request_processing import _relay_late_response
class _NoBody:
pass
frames: Final = [chunk async for chunk in _relay_late_response(_NoBody())]
assert frames == [b"data: {}\n\n", b"data: [DONE]\n\n"]
@pytest.mark.asyncio
async def test_relay_late_response_serializes_a_mapping_proxy():
from types import MappingProxyType
from litellm.proxy.common_request_processing import _relay_late_response
frames: Final = [chunk async for chunk in _relay_late_response(MappingProxyType({"ok": True}))]
assert frames == [b'data: {"ok":true}\n\n', b"data: [DONE]\n\n"]