oterm/tests/tools/test_mcp_logging.py

25 lines
635 B
Python

from fastmcp.client.logging import LogMessage
from oterm.tools.mcp.logging import log_handler
async def test_log_handler_routes_by_level_to_log_lines():
import oterm.log
before = len(oterm.log.log_lines)
for level, data in (
("debug", "D"),
("info", "I"),
("warning", "W"),
("error", "E"),
("critical", "C"),
):
await log_handler(LogMessage(level=level, data=data))
messages = [msg for _, msg in oterm.log.log_lines[before:]]
assert "D" in messages
assert "I" in messages
assert "W" in messages
assert "E" in messages
assert "C" in messages