mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-12 04:00:42 +00:00
62626d4d94
Use pytest's caplog fixture to suppress ERROR logs when deliberately triggering errors in test_sse_generator_error_before_response_starts. This keeps test output clean while still validating error handling behavior.
29 lines
846 B
Python
29 lines
846 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
import logging # allow-direct-logging
|
|
import os
|
|
import warnings
|
|
|
|
import pytest
|
|
|
|
|
|
def pytest_sessionstart(session) -> None:
|
|
if "LLAMA_STACK_LOGGING" not in os.environ:
|
|
os.environ["LLAMA_STACK_LOGGING"] = "all=WARNING"
|
|
|
|
# Silence common deprecation spam during unit tests.
|
|
warnings.filterwarnings("ignore", category=DeprecationWarning)
|
|
warnings.filterwarnings("ignore", category=PendingDeprecationWarning)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def suppress_httpx_logs(caplog):
|
|
"""Suppress httpx INFO logs for all unit tests"""
|
|
caplog.set_level(logging.WARNING, logger="httpx")
|
|
|
|
|
|
pytest_plugins = ["tests.unit.fixtures"]
|