Skip to content

Commit 806e8bc

Browse files
committed
[BugFix]fix console log metrics waitting queue count unittest
1 parent 7eefa24 commit 806e8bc

1 file changed

Lines changed: 24 additions & 0 deletions

File tree

tests/engine/test_common_engine.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -848,3 +848,27 @@ def __init__(self, *a, **k):
848848
eng._finalizer.detach()
849849
except Exception:
850850
pass
851+
852+
def test_get_scheduler_unhandled_request_num(self):
853+
"""Cover _get_scheduler_unhandled_request_num normal/fallback paths."""
854+
eng = EngineService.__new__(EngineService)
855+
eng.llm_logger = Mock()
856+
857+
# Scheduler does not provide API -> fallback 0
858+
eng.scheduler = object()
859+
self.assertEqual(eng._get_scheduler_unhandled_request_num(), 0)
860+
861+
# Positive value -> return int value
862+
eng.scheduler = type("SchedOK", (), {"get_unhandled_request_num": lambda self: "3"})()
863+
self.assertEqual(eng._get_scheduler_unhandled_request_num(), 3)
864+
865+
# Negative value -> clamp to 0
866+
eng.scheduler = type("SchedNeg", (), {"get_unhandled_request_num": lambda self: -5})()
867+
self.assertEqual(eng._get_scheduler_unhandled_request_num(), 0)
868+
869+
# Exception -> debug log + fallback 0
870+
eng.scheduler = type(
871+
"SchedErr", (), {"get_unhandled_request_num": lambda self: (_ for _ in ()).throw(RuntimeError("boom"))}
872+
)()
873+
self.assertEqual(eng._get_scheduler_unhandled_request_num(), 0)
874+
eng.llm_logger.debug.assert_called()

0 commit comments

Comments
 (0)