doris-sync/tests/test_logger.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
日志模块测试
"""

import os
import pytest
from unittest.mock import patch, MagicMock

import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))


class TestLogger:
    """日志测试"""
    
    @patch("os.makedirs")
    @patch("logging.FileHandler")
    @patch("logging.getLogger")
    def test_get_logger(self, mock_get_logger, mock_handler, mock_makedirs):
        """测试获取日志记录器"""
        mock_logger = MagicMock()
        mock_handler_instance = MagicMock()
        mock_handler.return_value = mock_handler_instance
        mock_get_logger.return_value = mock_logger
        
        from logger.handler import get_logger
        logger = get_logger("test_logger")
        
        # 验证 logger 被正确创建
        assert logger is not None
        mock_logger.setLevel.assert_called()
    
    def test_clean_old_logs(self):
        """测试清理旧日志"""
        from logger.handler import clean_old_logs
        
        # 测试函数可执行
        result = clean_old_logs(days=30)
        assert result >= 0