scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(security): expect the async release check on offline mode (#51213)

Browse source 
Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
devin-ai-integration[bot] 2026-11-11 18:15:42 +00:00 • committed by GitHub
parent 20d53f2951
commit 3cd48a6ec7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/integration/security/test_offline_mode.py
View file

@ -6,7 +6,7 @@ from pathlib import Path
from typing import Final
import pytest
from integration._support.client import Gateway, object_value
from integration._support.client import Gateway, eventually, object_value
from integration._support.forward_proxy import refusing_forward_proxy
from integration._support.process import owned_proxy
from pydantic import JsonValue
@ -120,7 +120,19 @@ def test_offline_mode_stops_every_self_initiated_request(
source: Final = object_value(_json(isolated, "GET", "/model/cost_map/source"))
reloaded: Final = isolated.request("POST", "/reload/model_cost_map")
beta_headers: Final = object_value(_json(isolated, "POST", "/reload/anthropic_beta_headers"))
release: Final = _json(isolated, "GET", "/get/latest_release_info")
release: Final = isolated.request("GET", "/get/latest_release_info")
assert (release.status_code, release.json()) == (202 if offline_mode is None else 200, None), release.text
if offline_mode is None:
settled_release: Final = eventually(
lambda: isolated.request("GET", "/get/latest_release_info"),
lambda response: response.status_code != 202,
)
assert (settled_release.status_code, settled_release.json()) == (
502,
{"detail": "Latest release info unavailable"},
), settled_release.text
else:
assert _json(isolated, "GET", "/get/latest_release_info") is None
presets: Final = object_value(_json(isolated, "GET", "/public/autorouter_presets"))
templates: Final = _json(isolated, "GET", "/policy/templates")
blog: Final = object_value(_json(isolated, "GET", "/public/litellm_blog_posts"))
@ -164,7 +176,6 @@ def test_offline_mode_stops_every_self_initiated_request(
assert source["is_env_forced"] is (offline_mode is not None), source
assert reloaded.status_code == reload_status, reloaded.text
assert beta_headers["status"] == "success", beta_headers
assert release is None
assert sorted(presets) == sorted(object_value(_bundled("proxy/public_endpoints/autorouter_presets.json")))
assert templates == _bundled("policy_templates_backup.json")
assert [object_value(post)["url"] for post in _list(blog["posts"])] == [