scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 48

test(router): cover _resolve_cache_type directly

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-08-08 23:18:59 +00:00
parent e895bf20e0
commit d25d31a222
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/test_router.py
View file

@ -16,6 +16,8 @@ sys.path.insert(
import litellm
from litellm.exceptions import MidStreamFallbackError
from litellm.integrations.custom_logger import CustomLogger
from litellm.router import _resolve_cache_type
from litellm.types.caching import LiteLLMCacheType
def test_update_kwargs_does_not_mutate_defaults_and_merges_metadata():
@ -7671,6 +7673,20 @@ def test_router_cache_kwargs_type_ignored_when_redis_params_given(reset_litellm_
assert created_caches == [{"host": "localhost", "port": "6379", "socket_timeout": 5}]
@pytest.mark.parametrize(
"requested_type, expected",
[(None, LiteLLMCacheType.LOCAL), ("disk", LiteLLMCacheType.DISK), (LiteLLMCacheType.S3, LiteLLMCacheType.S3)],
)
def test_resolve_cache_type(requested_type, expected):
assert _resolve_cache_type(requested_type) == expected
@pytest.mark.parametrize("requested_type", ["not-a-cache", 5])
def test_resolve_cache_type_rejects_invalid(requested_type):
with pytest.raises(ValueError):
_resolve_cache_type(requested_type)
def test_router_invalid_cache_kwargs_type_raises(reset_litellm_cache):
with pytest.raises(ValueError):
litellm.Router(