157 lines
3.7 KiB
Python
157 lines
3.7 KiB
Python
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from src.utils.log_manager import (
|
|
parse_size,
|
|
FileHandlerFactory,
|
|
LoggerPrint,
|
|
LogManager,
|
|
register,
|
|
register_global_params,
|
|
get_logger,
|
|
setup,
|
|
print as log_print,
|
|
get_log_manager,
|
|
LOG_CATEGORY,
|
|
LOG_CONSOLE,
|
|
LOG_STDERR,
|
|
LOG_FILE,
|
|
LOG_PATH,
|
|
LOG_LEVEL,
|
|
LOG_ROTATION,
|
|
LOG_ROTATION_SIZE,
|
|
LOG_ROTATION_COUNT,
|
|
LOG_TIMESTAMP,
|
|
DEFAULT_TIMESTAMP,
|
|
)
|
|
|
|
|
|
def test_parse_size():
|
|
assert parse_size("100B") == 100
|
|
assert parse_size("1KB") == 1024
|
|
assert parse_size("1MB") == 1024 * 1024
|
|
assert parse_size("1GB") == 1024 * 1024 * 1024
|
|
assert parse_size("10MB") == 10 * 1024 * 1024
|
|
|
|
|
|
def test_constants():
|
|
assert LOG_CATEGORY == "logger"
|
|
assert LOG_CONSOLE == "log_console"
|
|
assert LOG_STDERR == "log_stderr"
|
|
assert LOG_FILE == "log_file"
|
|
assert DEFAULT_TIMESTAMP == "%Y-%m-%d %H:%M:%S"
|
|
|
|
|
|
def test_file_handler_factory():
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
tmpdir_path = Path(tmpdir)
|
|
tmpdir_path.mkdir(parents=True, exist_ok=True)
|
|
handler = FileHandlerFactory.create(
|
|
tmpdir, "test.log", rotation="size", max_bytes="1KB", backup_count=3
|
|
)
|
|
assert handler is not None
|
|
assert isinstance(handler.backupCount, int)
|
|
assert handler.backupCount == 3
|
|
handler.close()
|
|
|
|
|
|
def test_logger_print():
|
|
logger = LoggerPrint(
|
|
name="komAI.test", module_name="test", console_enabled=True, stderr_enabled=True
|
|
)
|
|
logger.print("Test message", level="info")
|
|
logger.print("Warning message", level="warning")
|
|
logger.print("Error message", level="error")
|
|
|
|
|
|
def test_log_manager_registration():
|
|
import src.utils.config_manager as config
|
|
|
|
config.config.reset()
|
|
register_global_params()
|
|
|
|
register(module="test_mod", log_console=True, log_file="test_mod.log")
|
|
|
|
logger = get_logger("test_mod")
|
|
assert logger is not None
|
|
assert logger.module_name == "test_mod"
|
|
|
|
|
|
def test_log_manager_no_console():
|
|
import src.utils.config_manager as config
|
|
|
|
config.config.reset()
|
|
register_global_params()
|
|
|
|
register(module="test_mod", log_console=False)
|
|
|
|
logger = get_logger("test_mod")
|
|
assert logger.console_enabled is False
|
|
|
|
|
|
def test_log_manager_with_file():
|
|
import src.utils.config_manager as config
|
|
|
|
config.config.reset()
|
|
register_global_params()
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
register(module="test_mod", log_file="test_mod.log")
|
|
config.config.set(LOG_PATH, tmpdir, cat=LOG_CATEGORY)
|
|
|
|
logger = get_logger("test_mod")
|
|
assert logger.file_enabled is True
|
|
|
|
|
|
def test_log_manager_print():
|
|
import src.utils.config_manager as config
|
|
|
|
config.config.reset()
|
|
register_global_params()
|
|
|
|
register(module="test_mod", log_console=True)
|
|
|
|
setup()
|
|
log_print("Global print test", level="info")
|
|
|
|
|
|
def test_get_log_manager_singleton():
|
|
mgr1 = get_log_manager()
|
|
mgr2 = get_log_manager()
|
|
assert mgr1 is mgr2
|
|
|
|
|
|
def run_tests():
|
|
tests = [
|
|
test_parse_size,
|
|
test_constants,
|
|
test_file_handler_factory,
|
|
test_logger_print,
|
|
test_log_manager_registration,
|
|
test_log_manager_no_console,
|
|
test_log_manager_with_file,
|
|
test_log_manager_print,
|
|
test_get_log_manager_singleton,
|
|
]
|
|
|
|
passed = 0
|
|
failed = 0
|
|
|
|
for test in tests:
|
|
try:
|
|
test()
|
|
print(f"PASS: {test.__name__}")
|
|
passed += 1
|
|
except Exception as e:
|
|
print(f"FAIL: {test.__name__} - {e}")
|
|
failed += 1
|
|
|
|
print(f"\n{passed}/{passed + failed} tests passed")
|
|
return failed == 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = run_tests()
|
|
exit(0 if success else 1)
|