import pytest
from unittest.mock import patch, MagicMock
from src.classes.action.attack import Attack
from src.classes.event import Event
from src.classes.cultivation import Realm
from src.classes.avatar import Avatar

# 定义一个简单的 Result Mock
class MockResolutionResult:
    def __init__(self, obj):
        self.obj = obj

def test_attack_can_start_success(dummy_avatar):
    """测试攻击条件检查通过"""
    target = MagicMock(spec=Avatar)
    target.name = "TargetAvatar"
    target.is_dead = False
    
    with patch("src.classes.action.attack.resolve_query") as mock_resolve:
        mock_resolve.return_value = MockResolutionResult(target)
        
        action = Attack(dummy_avatar, dummy_avatar.world)
        can_start, reason = action.can_start("TargetAvatar")
        
        assert can_start is True
        assert reason == ""

def test_attack_can_start_fail_no_target(dummy_avatar):
    """测试目标不存在"""
    with patch("src.classes.action.attack.resolve_query") as mock_resolve:
        mock_resolve.return_value = MockResolutionResult(None)
        
        action = Attack(dummy_avatar, dummy_avatar.world)
        can_start, reason = action.can_start("Ghost")
        
        assert can_start is False
        assert "目标不存在" in reason

def test_attack_can_start_fail_dead_target(dummy_avatar):
    """测试目标已死亡"""
    target = MagicMock(spec=Avatar)
    target.is_dead = True
    
    with patch("src.classes.action.attack.resolve_query") as mock_resolve:
        mock_resolve.return_value = MockResolutionResult(target)
        
        action = Attack(dummy_avatar, dummy_avatar.world)
        can_start, reason = action.can_start("Zombie")
        
        assert can_start is False
        assert "目标已死亡" in reason

def test_attack_start_event(dummy_avatar):
    """测试开始攻击生成的事件"""
    target = MagicMock(spec=Avatar)
    target.name = "Enemy"
    target.id = "enemy-id"
    
    # Mock combat strength calculation
    with patch("src.classes.action.attack.resolve_query") as mock_resolve, \
         patch("src.classes.action.attack.get_effective_strength_pair") as mock_strength:
        
        mock_resolve.return_value = MockResolutionResult(target)
        mock_strength.return_value = (100, 80)
        
        action = Attack(dummy_avatar, dummy_avatar.world)
        event = action.start("Enemy")
        
        assert isinstance(event, Event)
        assert "TestDummy" in event.content
        assert "Enemy" in event.content
        assert "100" in event.content # 战斗力显示
        assert event.is_major is True

def test_attack_execute_logic(dummy_avatar):
    """测试执行战斗逻辑"""
    target = MagicMock(spec=Avatar)
    target.name = "Enemy"
    
    # Setup HP mocks
    dummy_avatar.hp = MagicMock()
    target.hp = MagicMock()
    
    # Setup proficiency mocks (methods on MagicMock)
    dummy_avatar.increase_weapon_proficiency = MagicMock()
    target.increase_weapon_proficiency = MagicMock()

    with patch("src.classes.action.attack.resolve_query") as mock_resolve, \
         patch("src.classes.action.attack.decide_battle") as mock_decide:
        
        mock_resolve.return_value = MockResolutionResult(target)
        
        # winner, loser, loser_damage, winner_damage
        # 假设 dummy_avatar 赢了
        mock_decide.return_value = (dummy_avatar, target, 50, 10)
        
        action = Attack(dummy_avatar, dummy_avatar.world)
        action._execute("Enemy")
        
        # 验证伤害应用
        # loser (target) takes 50 dmg
        target.hp.reduce.assert_called_with(50)
        # winner (dummy) takes 10 dmg
        dummy_avatar.hp.reduce.assert_called_with(10)
        
        # 验证熟练度增加
        assert dummy_avatar.increase_weapon_proficiency.called
        assert target.increase_weapon_proficiency.called
        
        # 验证结果保存
        assert action._last_result == (dummy_avatar, target, 50, 10)