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

test: add cache reset fixture and LITELLM_LOCAL_BLOG_POSTS test

Browse source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-02-21 16:48:54 -08:00
parent be1a543b55
commit 7e5c63c134
2 changed files with 24 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/get_blog_posts.py
View file

@ -63,7 +63,7 @@ class GetBlogPosts:
return content.get("posts", [])
@staticmethod
def fetch_remote_blog_posts(url: str, timeout: int = 5) -> Any:
def fetch_remote_blog_posts(url: str, timeout: int = 5) -> dict:
"""
Fetch blog posts JSON from a remote URL.

38
tests/test_litellm/test_get_blog_posts.py
View file

@ -25,6 +25,15 @@ SAMPLE_RESPONSE = {
}
@pytest.fixture(autouse=True)
def reset_blog_posts_cache():
GetBlogPosts._cached_posts = None
GetBlogPosts._last_fetch_time = 0.0
yield
GetBlogPosts._cached_posts = None
GetBlogPosts._last_fetch_time = 0.0
def test_load_local_blog_posts_returns_list():
posts = GetBlogPosts.load_local_blog_posts()
assert isinstance(posts, list)
@ -52,12 +61,8 @@ def test_validate_blog_posts_not_dict():
assert GetBlogPosts.validate_blog_posts("not a dict") is False
def test_get_blog_posts_success(monkeypatch):
def test_get_blog_posts_success():
"""Fetches from remote on first call."""
# Reset class cache state
GetBlogPosts._cached_posts = None
GetBlogPosts._last_fetch_time = 0.0
mock_response = MagicMock()
mock_response.json.return_value = SAMPLE_RESPONSE
mock_response.raise_for_status = MagicMock()
@ -69,11 +74,8 @@ def test_get_blog_posts_success(monkeypatch):
assert posts[0]["title"] == "Test Post"
def test_get_blog_posts_network_error_falls_back_to_local(monkeypatch):
def test_get_blog_posts_network_error_falls_back_to_local():
"""Falls back to local backup on network error."""
GetBlogPosts._cached_posts = None
GetBlogPosts._last_fetch_time = 0.0
with patch(
"litellm.litellm_core_utils.get_blog_posts.httpx.get",
side_effect=Exception("Network error"),
@ -84,11 +86,8 @@ def test_get_blog_posts_network_error_falls_back_to_local(monkeypatch):
assert len(posts) > 0
def test_get_blog_posts_invalid_json_falls_back_to_local(monkeypatch):
def test_get_blog_posts_invalid_json_falls_back_to_local():
"""Falls back when remote returns non-dict."""
GetBlogPosts._cached_posts = None
GetBlogPosts._last_fetch_time = 0.0
mock_response = MagicMock()
mock_response.json.return_value = "not a dict"
mock_response.raise_for_status = MagicMock()
@ -100,7 +99,7 @@ def test_get_blog_posts_invalid_json_falls_back_to_local(monkeypatch):
assert len(posts) > 0
def test_get_blog_posts_ttl_cache_not_refetched(monkeypatch):
def test_get_blog_posts_ttl_cache_not_refetched():
"""Within TTL window, does not re-fetch."""
GetBlogPosts._cached_posts = SAMPLE_RESPONSE["posts"]
GetBlogPosts._last_fetch_time = time.time() # just now
@ -122,7 +121,7 @@ def test_get_blog_posts_ttl_cache_not_refetched(monkeypatch):
assert len(posts) == 1
def test_get_blog_posts_ttl_expired_refetches(monkeypatch):
def test_get_blog_posts_ttl_expired_refetches():
"""After TTL window, re-fetches from remote."""
GetBlogPosts._cached_posts = SAMPLE_RESPONSE["posts"]
GetBlogPosts._last_fetch_time = time.time() - 7200 # 2 hours ago
@ -140,6 +139,15 @@ def test_get_blog_posts_ttl_expired_refetches(monkeypatch):
assert len(posts) == 1
def test_get_blog_posts_local_env_var_skips_remote(monkeypatch):
monkeypatch.setenv("LITELLM_LOCAL_BLOG_POSTS", "true")
with patch("litellm.litellm_core_utils.get_blog_posts.httpx.get") as mock_get:
posts = get_blog_posts()
mock_get.assert_not_called()
assert isinstance(posts, list)
assert len(posts) > 0
def test_blog_post_pydantic_model():
post = BlogPost(
title="T",