95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
#
|
|
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
import json
|
|
import logging
|
|
from typing import Dict
|
|
|
|
import pytest
|
|
from airbyte_cdk.logger import AirbyteLogFormatter
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def logger():
|
|
logger = logging.getLogger("airbyte.Testlogger")
|
|
return logger
|
|
|
|
|
|
def test_formatter(logger, caplog):
|
|
formatter = AirbyteLogFormatter()
|
|
logger.info("Test formatter")
|
|
record = caplog.records[0]
|
|
formatted_record = formatter.format(record)
|
|
formatted_record_data = json.loads(formatted_record)
|
|
assert formatted_record_data.get("type") == "LOG"
|
|
log = formatted_record_data.get("log")
|
|
assert isinstance(log, Dict)
|
|
level = log.get("level")
|
|
message = log.get("message")
|
|
assert level == "INFO"
|
|
assert message == "Test formatter"
|
|
|
|
|
|
def test_level_transform(logger, caplog):
|
|
formatter = AirbyteLogFormatter()
|
|
logger.warning("Test level transform warn")
|
|
logger.critical("Test level transform critical")
|
|
record_warn = caplog.records[0]
|
|
record_critical = caplog.records[1]
|
|
formatted_record_warn = formatter.format(record_warn)
|
|
formatted_record_warn_data = json.loads(formatted_record_warn)
|
|
log_warn = formatted_record_warn_data.get("log")
|
|
level_warn = log_warn.get("level")
|
|
formatted_record_critical = formatter.format(record_critical)
|
|
formatted_record_critical_data = json.loads(formatted_record_critical)
|
|
log_critical = formatted_record_critical_data.get("log")
|
|
level_critical = log_critical.get("level")
|
|
assert level_warn == "WARN"
|
|
assert level_critical == "FATAL"
|
|
|
|
|
|
def test_debug(logger, caplog):
|
|
# Test debug logger in isolation since the default logger is initialized to TRACE (15) instead of DEBUG (10).
|
|
debug_logger = logging.getLogger("airbyte.Debuglogger")
|
|
debug_logger.setLevel(logging.DEBUG)
|
|
debug_logger.debug("Test debug 1")
|
|
record = caplog.records[0]
|
|
assert record.levelname == "DEBUG"
|
|
assert record.message == "Test debug 1"
|
|
|
|
|
|
def test_default_debug_is_ignored(logger, caplog):
|
|
logger.debug("Test debug that is ignored since log level is TRACE")
|
|
assert len(caplog.records) == 0
|
|
|
|
|
|
def test_info(logger, caplog):
|
|
logger.info("Test info 1")
|
|
logger.info("Test info 2")
|
|
assert len(caplog.records) == 2
|
|
first_record = caplog.records[0]
|
|
assert first_record.levelname == "INFO"
|
|
assert first_record.message == "Test info 1"
|
|
|
|
|
|
def test_warn(logger, caplog):
|
|
logger.warn("Test warn 1")
|
|
record = caplog.records[0]
|
|
assert record.levelname == "WARNING"
|
|
assert record.message == "Test warn 1"
|
|
|
|
|
|
def test_error(logger, caplog):
|
|
logger.error("Test error 1")
|
|
record = caplog.records[0]
|
|
assert record.levelname == "ERROR"
|
|
assert record.message == "Test error 1"
|
|
|
|
|
|
def test_fatal(logger, caplog):
|
|
logger.fatal("Test fatal 1")
|
|
record = caplog.records[0]
|
|
assert record.levelname == "CRITICAL"
|
|
assert record.message == "Test fatal 1"
|