scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test(proxy): isolate live route scheduler

Browse source
This commit is contained in:
jibanez-staticduo 2026-09-18 15:58:54 +02:00
parent 3f36fe6420
commit e0726859ef
No known key found for this signature in database
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/proxy/test_live_route_registration.py
View file

@ -50,6 +50,8 @@ def test_public_live_websockets_reach_live_auth_before_legacy_sideband(monkeypat
authenticate = AsyncMock(side_effect=HTTPException(403, "Live authentication rejected"))
monkeypatch.setattr(live, "_auth", authenticate)
monkeypatch.setattr(proxy_server, "general_settings", {})
# A previous proxy test may leave the module scheduler bound to a closed loop.
monkeypatch.setattr(proxy_server, "scheduler", None)
with TestClient(proxy_server.app) as client:
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect(prefix + "/sessions" + suffix, headers={"authorization": "Bearer test"}):