mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
7cc29fe192
* package setup with logger * set config once * add unit test workflow * updated naming of workflows * add mypy check * renamed job * smaller name * ignore certain files for ruff * remove assignment * fix ruff config * removed pyright from pre-commit * fixed logging test * fix mypy setup * mypy fix * mypy * mypy
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
|
|
import pytest
|
|
|
|
from agent_framework import get_logger
|
|
from agent_framework.exceptions import AgentFrameworkException
|
|
|
|
|
|
def test_get_logger():
|
|
"""Test that the logger is created with the correct name."""
|
|
logger = get_logger()
|
|
assert logger.name == "agent_framework"
|
|
|
|
|
|
def test_get_logger_custom_name():
|
|
"""Test that the logger can be created with a custom name."""
|
|
custom_name = "agent_framework.custom"
|
|
logger = get_logger(custom_name)
|
|
assert logger.name == custom_name
|
|
|
|
|
|
def test_get_logger_invalid_name():
|
|
"""Test that an exception is raised for an invalid logger name."""
|
|
with pytest.raises(AgentFrameworkException, match="Logger name must start with 'agent_framework'."):
|
|
get_logger("invalid_name")
|
|
|
|
|
|
def test_log(caplog):
|
|
"""Test that the logger can log messages and adheres to the expected format."""
|
|
logger = get_logger()
|
|
with caplog.at_level("DEBUG"):
|
|
logger.debug("This is a debug message")
|
|
assert len(caplog.records) == 1
|
|
record = caplog.records[0]
|
|
assert record.levelname == "DEBUG"
|
|
assert record.message == "This is a debug message"
|
|
assert record.name == "agent_framework"
|
|
assert record.pathname.endswith("test_logging.py")
|