scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 181

test(rust): capture deferred stream callback gap

Browse source
This commit is contained in:
Yujong Lee 2026-09-08 11:05:40 -07:00
parent bd2ce1db16
commit 07f89628d5
1 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/test_litellm_rust/test_messages_callbacks.py
View file

@ -143,6 +143,28 @@ async def test_messages_callbacks_run_once(messages_server: RecordingServer) ->
assert "async_log_failure_event" not in recorder.names
@pytest.mark.asyncio
@pytest.mark.xfail(
reason="Buffered Rust Messages defers success callbacks until the client consumes or closes the fake stream",
strict=True,
)
async def test_messages_buffered_stream_callbacks_settle_before_client_consumption(
messages_server: RecordingServer,
) -> None:
recorder: Final = RecordingLogger()
stream: Final = await call_messages(messages_server, [recorder], stream=True)
await drain_logging()
callbacks_before_close: Final = recorder.names.count("async_log_success_event")
await stream.aclose()
await drain_logging()
callbacks_after_close: Final = recorder.names.count("async_log_success_event")
assert callbacks_before_close == 1
assert callbacks_after_close == 1
@pytest.mark.asyncio
async def test_messages_logging_drain_waits_for_suspended_callback(messages_server: RecordingServer) -> None:
started: Final = asyncio.Event()