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 87

test(router): directly cover _deregister_strategy_router

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Krrish Dholakia 2026-07-08 15:07:55 +00:00
parent 68ba8a7b99
commit b91c845998
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/router_unit_tests/test_router_helper_utils.py
View file

@ -1867,6 +1867,27 @@ def _auto_router_deployment(config="/path/to/config", model_id="ar-id"):
)
def test_deregister_strategy_router_removes_all_registries(model_list):
"""_deregister_strategy_router must drop the model_name from every per-strategy registry,
and tolerate a missing name or None."""
router = Router(model_list=model_list)
sentinel = MagicMock()
router.auto_routers["x"] = sentinel
router.complexity_routers["x"] = sentinel
router.quality_routers["x"] = sentinel
router.adaptive_routers["x"] = sentinel
router._deregister_strategy_router(model_name="x")
assert "x" not in router.auto_routers
assert "x" not in router.complexity_routers
assert "x" not in router.quality_routers
assert "x" not in router.adaptive_routers
router._deregister_strategy_router(model_name="not-present")
router._deregister_strategy_router(model_name=None)
@patch("litellm.router_strategy.auto_router.auto_router.AutoRouter")
def test_delete_deployment_deregisters_auto_router(mock_auto_router, model_list):
"""Regression: deleting an auto-router deployment must drop it from ``auto_routers`` so the