From d25d31a2225660505852af06072bce45763c506f Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Sat, 8 Aug 2026 23:18:59 +0000
Subject: [PATCH] test(router): cover _resolve_cache_type directly

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 tests/test_litellm/test_router.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/test_litellm/test_router.py b/tests/test_litellm/test_router.py
index 85eab35344b..7f0bbb6bfe3 100644
--- a/tests/test_litellm/test_router.py
+++ b/tests/test_litellm/test_router.py
@@ -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(