scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

Refactor fake_kill to use Mock for testing

Browse source 
test(cli): assert both kills still fire when every kill raises
This commit is contained in:
Yoggy 2026-08-24 10:45:07 +09:00 • committed by GitHub
parent fe24bcb83e
commit db7bd4e072
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/client/cli/autoroute/test_process.py
View file

@ -331,8 +331,7 @@ class TestTerminateHardKillSignal:
this same "already dead or inaccessible" case.
"""
def fake_kill(pid: int, sig: int) -> None:
raise raised
def fake_kill: Final = Mock(side_effect=raised)
monkeypatch.setattr(process_module.os, "kill", fake_kill)
monkeypatch.setattr(process_module, "is_running", lambda pid: True)
@ -340,6 +339,7 @@ class TestTerminateHardKillSignal:
terminate(4242, grace_period=0.0)
assert _signals_sent(fake_kill) == (signal.SIGTERM, getattr(signal, "SIGKILL", signal.SIGTERM))
class TestPollLiveliness:
def test_succeeds_when_health_check_returns_200_quickly(self, monkeypatch, tmp_path):