mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-02 07:13:02 +00:00
780c504e00
* test: migrate tests/proxy_behavior into integration and unit tiers (partial) * test(integration): run settings-dependent management tests on an owned proxy * test(integration): assert stored values in the rewritten management rows * test(integration): pin sibling team allocations against the org guaranteed throughput cap * test(integration): keep Prisma clients out of the shared integration conftest import path --------- Co-authored-by: yuneng <yuneng@berri.ai>
24 lines
1,023 B
Python
24 lines
1,023 B
Python
import pytest
|
|
|
|
from tests.integration._support.actors import Actor
|
|
|
|
pytestmark = pytest.mark.asyncio(loop_scope="session")
|
|
|
|
|
|
# POST /key/health has no role gate — it reflects the caller's OWN key logging
|
|
# metadata. The actors keys carry no "logging" metadata, so every authenticated
|
|
# actor gets 200 with key="healthy". This pins auth-required + route coverage.
|
|
@pytest.mark.parametrize("actor", list(Actor), ids=[a.value for a in Actor])
|
|
async def test_key_health_each_actor_is_healthy(actor: Actor, gateway_client, actors):
|
|
caller = actors.keys[actor]
|
|
resp = await gateway_client.post(
|
|
"/key/health",
|
|
headers={"Authorization": f"Bearer {caller.cleartext}"},
|
|
)
|
|
assert resp.status_code == 200, f"{actor.value}: {resp.status_code} {resp.text}"
|
|
assert resp.json()["key"] == "healthy"
|
|
|
|
|
|
async def test_key_health_requires_auth(gateway_client):
|
|
resp = await gateway_client.post("/key/health", headers={"Authorization": "Bearer invalid"})
|
|
assert resp.status_code == 401, resp.text
|