mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
"""
|
|
Test interaction of logging with configuration system.
|
|
"""
|
|
|
|
import io
|
|
import logging
|
|
import re
|
|
|
|
from invokeai.app.services.config import InvokeAIAppConfig
|
|
from invokeai.backend.util.logging import LOG_FORMATTERS, InvokeAILogger
|
|
|
|
|
|
# test formatting
|
|
# Would prefer to use the capfd/capsys fixture here, but it is broken
|
|
# when used with the logging module: https://github.com/pytest-dev/pytest/issue
|
|
def test_formatting():
|
|
logger = InvokeAILogger.get_logger()
|
|
stream = io.StringIO()
|
|
handler = logging.StreamHandler(stream)
|
|
handler.setFormatter(LOG_FORMATTERS["plain"]())
|
|
logger.addHandler(handler)
|
|
logger.info("test1")
|
|
output = stream.getvalue()
|
|
assert re.search(r"\[InvokeAI\]::INFO --> test1$", output)
|
|
|
|
handler.setFormatter(LOG_FORMATTERS["legacy"]())
|
|
logger.info("test2")
|
|
output = stream.getvalue()
|
|
assert re.search(r">> test2$", output)
|
|
|
|
|
|
# test independence of two loggers with different names
|
|
def test_independence():
|
|
logger1 = InvokeAILogger.get_logger()
|
|
logger2 = InvokeAILogger.get_logger("Test")
|
|
assert logger1.name == "InvokeAI"
|
|
assert logger2.name == "Test"
|
|
assert logger1.level == logging.INFO
|
|
assert logger2.level == logging.INFO
|
|
logger2.setLevel(logging.DEBUG)
|
|
assert logger1.level == logging.INFO
|
|
assert logger2.level == logging.DEBUG
|
|
|
|
|
|
# test that the logger is returned from two similar get_logger() calls
|
|
def test_retrieval():
|
|
logger1 = InvokeAILogger.get_logger()
|
|
logger2 = InvokeAILogger.get_logger()
|
|
logger3 = InvokeAILogger.get_logger("Test")
|
|
assert logger1 == logger2
|
|
assert logger1 != logger3
|
|
|
|
|
|
# test that the configuration is used to set the initial logging level
|
|
def test_config():
|
|
config = InvokeAIAppConfig(log_level="debug")
|
|
logger1 = InvokeAILogger.get_logger("DebugTest", config=config)
|
|
assert logger1.level == logging.DEBUG
|