mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-06 22:58:30 +00:00
35c45e3386
- Remove unused sys and redirect_stderr imports from test_simple_logging_fix.py - Remove unused sys and pytest imports from tests/test_custom_logger_fix.py - All lint checks now pass locally Co-Authored-By: Jo\u00E3o <joao@crewai.com>
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""
|
|
Simple test to verify the logging fix works without external API calls
|
|
"""
|
|
import logging
|
|
import io
|
|
from contextlib import redirect_stdout
|
|
from crewai.utilities.events.utils.console_formatter import ConsoleFormatter
|
|
from rich.tree import Tree
|
|
|
|
|
|
def test_console_formatter_logging_fix():
|
|
"""Test that ConsoleFormatter allows custom logging when Live session is active"""
|
|
print("Testing ConsoleFormatter logging fix...")
|
|
|
|
logger = logging.getLogger("test_logger")
|
|
logger.setLevel(logging.INFO)
|
|
|
|
log_buffer = io.StringIO()
|
|
handler = logging.StreamHandler(log_buffer)
|
|
formatter = logging.Formatter('CUSTOM_LOG: %(message)s')
|
|
handler.setFormatter(formatter)
|
|
logger.addHandler(handler)
|
|
|
|
console_formatter = ConsoleFormatter(verbose=True)
|
|
|
|
tree = Tree("Test Tree")
|
|
console_formatter.print(tree)
|
|
|
|
assert console_formatter._live is not None, "Live session should be active"
|
|
|
|
logger.info("This should appear in the log buffer")
|
|
|
|
log_output = log_buffer.getvalue()
|
|
assert "CUSTOM_LOG: This should appear in the log buffer" in log_output, f"Custom log not found in output: {log_output}"
|
|
|
|
assert console_formatter._live is not None, "Live session should still be active after custom logging"
|
|
|
|
print("✅ ConsoleFormatter logging fix test passed!")
|
|
|
|
logger.removeHandler(handler)
|
|
handler.close()
|
|
|
|
|
|
def test_console_formatter_print_behavior():
|
|
"""Test that non-Tree content properly pauses/resumes Live sessions"""
|
|
print("Testing ConsoleFormatter print behavior...")
|
|
|
|
console_formatter = ConsoleFormatter(verbose=True)
|
|
|
|
tree = Tree("Test Tree")
|
|
console_formatter.print(tree)
|
|
|
|
assert console_formatter._live is not None, "Live session should be active"
|
|
|
|
stdout_buffer = io.StringIO()
|
|
with redirect_stdout(stdout_buffer):
|
|
console_formatter.print("Non-tree content")
|
|
|
|
output = stdout_buffer.getvalue()
|
|
assert "Non-tree content" in output, f"Non-tree content not found in output: {output}"
|
|
|
|
assert console_formatter._live is not None, "Live session should be restored after printing non-Tree content"
|
|
|
|
print("✅ ConsoleFormatter print behavior test passed!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_console_formatter_logging_fix()
|
|
test_console_formatter_print_behavior()
|
|
print("🎉 All simple logging fix tests passed!")
|