import importlib
import logging
from collections.abc import Iterator

import pytest

import litellm_proxy_extras._logging as extras_logging


@pytest.fixture
def fresh_extras_logger() -> Iterator[logging.Logger]:
    logger = logging.getLogger("litellm_proxy_extras")
    saved_handlers = logger.handlers[:]
    saved_level = logger.level
    logger.handlers[:] = []
    try:
        yield logger
    finally:
        logger.handlers[:] = saved_handlers
        logger.setLevel(saved_level)


def test_litellm_log_error_silences_extras_info_lines(monkeypatch, fresh_extras_logger):
    monkeypatch.setenv("LITELLM_LOG", "ERROR")
    reloaded = importlib.reload(extras_logging).logger
    assert reloaded is fresh_extras_logger
    assert reloaded.isEnabledFor(logging.INFO) is False
    assert reloaded.isEnabledFor(logging.ERROR) is True


@pytest.mark.parametrize("litellm_log", [None, "info", "DEBUG"])
def test_unset_or_verbose_litellm_log_keeps_extras_info_lines(monkeypatch, fresh_extras_logger, litellm_log):
    if litellm_log is None:
        monkeypatch.delenv("LITELLM_LOG", raising=False)
    else:
        monkeypatch.setenv("LITELLM_LOG", litellm_log)
    assert importlib.reload(extras_logging).logger.isEnabledFor(logging.INFO) is True