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(proxy): type the new spend counter test parameters
Some checks failed
LiteLLM Rust / rust-lint (push) Has been cancelled
Details
LiteLLM Rust / rust-test (push) Has been cancelled
Details
LiteLLM Rust / rust-wheel (push) Has been cancelled
Details

Browse source
This commit is contained in:
yucheng 2026-10-26 18:54:28 -07:00
parent efefd3949c
commit b8e12c9144
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/proxy/proxy_server/test_spend_counters.py
View file

@ -1355,7 +1355,9 @@ class _SharedRedisCounters:
@pytest.mark.asyncio
@pytest.mark.parametrize("with_redis", (True, False))
@pytest.mark.parametrize("counter_key", ("spend:end_user:customer-1", "spend:tag:tag-1"))
async def test_overlapping_cold_starts_count_the_cached_spend_once(monkeypatch, with_redis, counter_key):
async def test_overlapping_cold_starts_count_the_cached_spend_once(
monkeypatch: pytest.MonkeyPatch, with_redis: bool, counter_key: str
) -> None:
cached_spend: Final = {"first-start": 2.5, "second-start": 6.0, "third-start": 6.0, "fourth-start": 2.5}
async def _cached_object(key: str, **_: object) -> dict[str, float]:
@ -1395,7 +1397,9 @@ class _RedisThatFailsTheLift:
@pytest.mark.asyncio
async def test_a_redis_failure_while_lifting_a_cold_counter_drops_the_local_copy_and_raises(monkeypatch):
async def test_a_redis_failure_while_lifting_a_cold_counter_drops_the_local_copy_and_raises(
monkeypatch: pytest.MonkeyPatch,
) -> None:
counter_key: Final = "spend:end_user:customer-1"
counters: Final = DualCache(in_memory_cache=InMemoryCache(), redis_cache=cast(RedisCache, _RedisThatFailsTheLift()))
counters.in_memory_cache.set_cache(key=counter_key, value=2.0)