#!/usr/bin/env python
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

import logging
import os
import pytest

from util.logging import get_level, get_logger

@pytest.fixture(scope="function", autouse=True)
def reset_logging_defaults():
    """Remove any logging configuration defaults that might have existed before starting any test"""
    try:
        os.environ.pop("LOG_LEVEL")
    except KeyError:
        pass


@pytest.mark.parametrize("level", ["DEBUG", "INFO", "WARNING", "ERROR"])
def test_valid_levels(level):
    os.environ["LOG_LEVEL"] = level
    assert get_level() == level


def test_invalid_level():
    os.environ["LOG_LEVEL"] = "TRACE"
    assert get_level() == "INFO"
    os.environ["LOG_LEVEL"] = "INFO"


def test_get_logger():
    logger = get_logger(__name__)
    assert logger.level == logging.INFO


def test_logger_log(caplog):
    logger = get_logger(__name__)
    logger.error("This is an error")
    logger.warning("This is a warning")
    logger.info("This is an informational message")
    logger.debug("This is a debug message")
    assert "This is an error" in caplog.text
    assert "This is a warning" in caplog.text
    assert "This is an informational message" in caplog.text
    assert "This is a debug message" not in caplog.text