litellm/tests/e2e/logging/conftest.py

"""Fixtures for the logging e2e suite.

Missing proxy, provider keys, or integration credentials are hard failures.
Never pytest.skip from this suite for environment gaps.
"""

from __future__ import annotations

import os

import pytest

from logging_client import LangfuseCreds, LoggingClient, build_logging_client, load_langfuse_creds
from datadog_reader import DdLogsReader, build_dd_logs_reader
from otel_client import OtelReader, build_otel_reader


def pytest_configure(config: pytest.Config) -> None:
    config.addinivalue_line(
        "markers",
        "covers: registry cell a test covers, e.g. logging.langfuse.success.logs_spend",
    )


@pytest.fixture(scope="session")
def client() -> LoggingClient:
    """The logging suite's client: holds the shared Gateway so `resources` /
    `scoped_key` clean up keys and teams, and adds `/metrics` scraping plus
    Langfuse read-back."""
    return build_logging_client()


@pytest.fixture(scope="session")
def otel_reader() -> OtelReader:
    """Read-back client for the compose stack's Jaeger trace destination."""
    return build_otel_reader()


@pytest.fixture(scope="session")
def dd_logs() -> DdLogsReader:
    """Read-back client for the real DataDog Logs Search API (keys from the
    secret manager on the cluster, tests/e2e/.env locally)."""
    return build_dd_logs_reader()


@pytest.fixture
def datadog_creds() -> None:
    """Require Datadog shipping credentials. Hard-fail when absent; never skip."""
    if not (os.getenv("DD_API_KEY") and os.getenv("DD_SITE")):
        pytest.fail(
            "Datadog e2e requires DD_API_KEY and DD_SITE; missing credentials is a hard failure, not a skip"
        )


@pytest.fixture(scope="session")
def langfuse_creds() -> LangfuseCreds:
    """Require real Langfuse cloud credentials for team callback + trace poll."""
    return load_langfuse_creds()