From 5c15ef50142536f1b5336982b3f0d9f5486645ff Mon Sep 17 00:00:00 2001
From: Joshua Valluru <326636767+joshua-berri@users.noreply.github.com>
Date: Wed, 28 Oct 2026 18:34:08 -0700
Subject: [PATCH] test: retain callbacks after bounded worker cancellation

---
 .../litellm_core_utils/test_logging_worker.py | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/tests/unit/litellm_core_utils/test_logging_worker.py b/tests/unit/litellm_core_utils/test_logging_worker.py
index d7824d612ca..b67842805e2 100644
--- a/tests/unit/litellm_core_utils/test_logging_worker.py
+++ b/tests/unit/litellm_core_utils/test_logging_worker.py
@@ -132,6 +132,35 @@ async def test_stop_does_not_replay_a_callback_dequeued_before_its_processor_sta
     await asyncio.wait_for(worker._queue.join(), timeout=1)
 
 
+@pytest.mark.asyncio
+async def test_stop_timeout_keeps_callbacks_observable_until_queue_cleanup_finishes() -> None:
+    worker: Final = LoggingWorker(concurrency=1)
+    started: Final = asyncio.Event()
+    release: Final = asyncio.Event()
+    completed: Final = asyncio.Event()
+
+    async def running() -> None:
+        started.set()
+        await asyncio.Event().wait()
+
+    async def queued() -> None:
+        await release.wait()
+        completed.set()
+
+    worker.ensure_initialized_and_enqueue(running())
+    await started.wait()
+    worker.enqueue(queued())
+    try:
+        await worker.stop(timeout=0)
+        assert not completed.is_set()
+        release.set()
+        await worker.flush()
+        assert completed.is_set(), "Queue cleanup must survive a bounded cancellation wait"
+    finally:
+        release.set()
+        await worker.stop()
+
+
 class TestLoggingWorker:
     """Test cases for LoggingWorker functionality."""