"""通知模块测试."""

import pytest
from unittest.mock import MagicMock, patch

from sms_forwarder.models import SMSMessage
from sms_forwarder.notifier import NotificationManager


@pytest.fixture
def notification_manager():
    """通知管理器测试夹具."""
    with patch("sms_forwarder.notifier.get_config") as mock_get_config:
        # 模拟配置
        mock_config = MagicMock()
        mock_config.notifications.services = [
            MagicMock(name="test", url="test://test", enabled=True)
        ]
        mock_config.notifications.templates.sms.title = "SMS: {sender}"
        mock_config.notifications.templates.sms.body = "From: {sender}\nContent: {content}\nTime: {timestamp}"
        mock_config.notifications.templates.system.title = "System"
        mock_config.notifications.templates.system.body = "{message}"
        
        mock_get_config.return_value = mock_config
        
        manager = NotificationManager()
        # 模拟 Apprise 对象
        manager.apprise_obj = MagicMock()
        manager.apprise_obj.__len__.return_value = 1
        
        yield manager


def test_send_sms_notification_success(notification_manager):
    """测试成功发送短信通知."""
    notification_manager.apprise_obj.notify.return_value = True
    
    sms = SMSMessage(
        sender="Test Sender",
        content="Test Message"
    )
    
    result = notification_manager.send_sms_notification(sms)
    
    assert result is True
    assert notification_manager.stats["sent"] == 1
    assert notification_manager.stats["failed"] == 0
    notification_manager.apprise_obj.notify.assert_called_once()


def test_send_sms_notification_failure(notification_manager):
    """测试发送短信通知失败."""
    notification_manager.apprise_obj.notify.return_value = False
    
    sms = SMSMessage(
        sender="Test Sender",
        content="Test Message"
    )
    
    result = notification_manager.send_sms_notification(sms)
    
    assert result is False
    assert notification_manager.stats["sent"] == 0
    assert notification_manager.stats["failed"] == 1
    notification_manager.apprise_obj.notify.assert_called_once()


def test_send_system_notification(notification_manager):
    """测试发送系统通知."""
    notification_manager.apprise_obj.notify.return_value = True
    
    result = notification_manager.send_system_notification("Test System Message")
    
    assert result is True
    assert notification_manager.stats["sent"] == 1
    notification_manager.apprise_obj.notify.assert_called_once()


def test_no_services_available(notification_manager):
    """测试没有可用服务时的行为."""
    # 模拟没有可用服务
    notification_manager.apprise_obj.__len__.return_value = 0
    
    sms = SMSMessage(
        sender="Test Sender",
        content="Test Message"
    )
    
    result = notification_manager.send_sms_notification(sms)
    
    assert result is False
    assert notification_manager.stats["failed"] == 1
    # 应该不会调用 notify
    notification_manager.apprise_obj.notify.assert_not_called()


def test_get_stats(notification_manager):
    """测试获取统计信息."""
    notification_manager.stats = {"sent": 5, "failed": 2}
    
    stats = notification_manager.get_stats()
    
    assert stats == {"sent": 5, "failed": 2}
    # 确保返回的是副本
    assert stats is not notification_manager.stats