scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-28 01:32:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 43

test(router_strategy): add explicit routing args test for 100% patch coverage

Browse source
This commit is contained in:
Rohit Kanithi 2026-09-13 02:34:31 -05:00
parent c598dff3db
commit f9738390be
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/test_litellm/router_strategy/test_base_routing_strategy.py
View file

@ -216,3 +216,22 @@ class TestRouterStrategyInstanceIsolation:
finally:
await h1.cleanup()
await h2.cleanup()
@pytest.mark.asyncio
async def test_handlers_accept_explicit_routing_args(self) -> None:
"""Verify handlers accept and respect explicitly provided routing_args dict."""
mock_cache = MagicMock(spec=DualCache)
h_cost = LowestCostLoggingHandler(router_cache=mock_cache, routing_args={"custom": True})
assert h_cost.routing_args == {"custom": True}
h_lat = LowestLatencyLoggingHandler(router_cache=mock_cache, routing_args={"ttl": 120})
assert h_lat.routing_args.ttl == 120
h_tpm = LowestTPMLoggingHandler(router_cache=mock_cache, routing_args={"ttl": 180})
assert h_tpm.routing_args.ttl == 180
h_tpm_v2 = LowestTPMLoggingHandler_v2(router_cache=mock_cache, routing_args={"ttl": 240})
try:
assert h_tpm_v2.routing_args.ttl == 240
finally:
await h_tpm_v2.cleanup()