wehub-resource-sync
cddb07a176
build container image / cpu (push) Waiting to run
build container image / cuda (push) Waiting to run
build container image / rocm (push) Waiting to run
frontend tests / frontend-tests (push) Waiting to run
openapi checks / openapi-checks (push) Waiting to run
python tests / py3.12: macos-default (push) Waiting to run
python tests / py3.11: windows-cpu (push) Waiting to run
python tests / py3.12: windows-cpu (push) Waiting to run
python tests / py3.11: linux-cpu (push) Waiting to run
python tests / py3.12: linux-cpu (push) Waiting to run
typegen checks / typegen-checks (push) Waiting to run
uv lock checks / uv-lock-checks (push) Waiting to run
frontend checks / frontend-checks (push) Waiting to run
lfs checks / lfs-check (push) Waiting to run
python checks / python-checks (push) Waiting to run
python tests / py3.11: macos-default (push) Waiting to run
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
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
|