mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-16 23:41:43 +00:00
442fdc181e
The e2e docs claimed `e2e`-marked tests skip when no proxy answers the liveness probe, but the harness has always hard-failed: conftest.py's pytest_runtest_setup calls pytest.fail, its module docstring states "hard failures only ... never skip", and logging/conftest.py forbids skipping outright. Align the docs to the code so the single most important contract reads the same everywhere; a dead proxy turns a run red instead of being silently skipped and mistaken for a pass. The per-suite conftest docstrings that described the shared hook as a "proxy liveness skip" are corrected to "liveness gate" for the same reason. Also scope the no-unit-tests hard rule to what it means: never substitute a unit test for e2e feature coverage, while explicitly allowing tests that cover the harness itself (e.g. coverage_registry/test_collector.py), which carry no e2e marker and run whether or not a proxy is up. No product code and no harness logic changed. Resolves LIT-4554
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
"""Batches suite's `client` fixture.
|
|
|
|
The shared lifecycle (resources/scoped_key), proxy liveness gate, and e2e marker
|
|
live in the parent tests/e2e/conftest.py. BatchClient holds the shared Gateway, so
|
|
the `resources` fixture cleans up keys through it; tests register file deletes and
|
|
batch cancels via `resources.defer(...)`.
|
|
|
|
Batch deployments (openai-batch, azure-batch, vertex-batch, ...) are registered
|
|
once per session via /model/new and deleted on teardown so they need not live in
|
|
the proxy config.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Iterator
|
|
|
|
import pytest
|
|
|
|
from batch_client import BatchClient, build_client
|
|
from capabilities import PROVIDERS
|
|
from e2e_http import NoBody
|
|
|
|
|
|
def pytest_configure(config: pytest.Config) -> None:
|
|
config.addinivalue_line(
|
|
"markers",
|
|
"covers: registry cell a test covers, e.g. llm.batches.openai.basic.nonstream.works",
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def client() -> BatchClient:
|
|
return build_client()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def batch_deployments(client: BatchClient) -> Iterator[None]:
|
|
probe = client.gateway.probe("/health/liveliness", params=NoBody())
|
|
if not probe.healthy:
|
|
yield
|
|
return
|
|
|
|
registered: list[str] = []
|
|
try:
|
|
for provider in PROVIDERS:
|
|
registered.append(
|
|
client.create_model(provider.model, provider.litellm_params())
|
|
)
|
|
yield
|
|
finally:
|
|
for model_id in registered:
|
|
client.delete_model(model_id)
|