import pytest
from src.classes.weapon import Weapon, get_random_weapon_by_realm, weapons_by_id
from src.classes.auxiliary import Auxiliary, get_random_auxiliary_by_realm, auxiliaries_by_id
from src.classes.cultivation import Realm
from src.classes.weapon_type import WeaponType

class TestEquipment:
    def test_weapon_structure(self):
        """测试兵器数据结构和加载"""
        assert len(weapons_by_id) > 0
        
        # 检查任意一个兵器
        weapon = next(iter(weapons_by_id.values()))
        assert isinstance(weapon, Weapon)
        assert isinstance(weapon.realm, Realm)
        assert isinstance(weapon.weapon_type, WeaponType)
        assert weapon.name
        
        # 测试 info 方法
        info = weapon.get_info()
        assert info == weapon.name
        
        detailed = weapon.get_detailed_info()
        assert weapon.name in detailed
        assert weapon.realm.value in detailed

    def test_weapon_random_generation(self):
        """测试按境界随机生成兵器"""
        # 测试练气期
        w_qi = get_random_weapon_by_realm(Realm.Qi_Refinement)
        if w_qi:
            assert w_qi.realm == Realm.Qi_Refinement
            
        # 测试筑基期
        w_found = get_random_weapon_by_realm(Realm.Foundation_Establishment)
        if w_found:
            assert w_found.realm == Realm.Foundation_Establishment
            
        # 测试金丹期
        w_core = get_random_weapon_by_realm(Realm.Core_Formation)
        if w_core:
            assert w_core.realm == Realm.Core_Formation

    def test_weapon_deepcopy(self):
        """测试兵器深拷贝独立性"""
        w1 = get_random_weapon_by_realm(Realm.Qi_Refinement)
        if not w1:
            pytest.skip("No Qi Refinement weapons available")
            
        w2 = get_random_weapon_by_realm(Realm.Qi_Refinement)
        # 即使随机到同一个原型，它们也应该是不同的对象
        # 注意：get_random_weapon_by_realm 内部已经做了 deepcopy
        
        # 为了确保测试有效，我们手动获取同一个原型并 deepcopy
        prototype = weapons_by_id[w1.id]
        import copy
        w_copy1 = copy.deepcopy(prototype)
        w_copy2 = copy.deepcopy(prototype)
        
        assert w_copy1 is not w_copy2
        assert w_copy1.id == w_copy2.id
        
        # 修改一个不影响另一个
        w_copy1.special_data["test"] = 123
        assert "test" not in w_copy2.special_data

    def test_auxiliary_structure(self):
        """测试辅助装备数据结构和加载"""
        assert len(auxiliaries_by_id) > 0
        
        # 检查任意一个
        aux = next(iter(auxiliaries_by_id.values()))
        assert isinstance(aux, Auxiliary)
        assert isinstance(aux.realm, Realm)
        assert aux.name
        
        # 测试 info 方法
        info = aux.get_info()
        assert info == aux.name
        
        detailed = aux.get_detailed_info()
        assert aux.name in detailed
        assert aux.realm.value in detailed

    def test_soul_banner(self):
        """测试万魂幡特殊逻辑"""
        # 查找万魂幡
        soul_banner = None
        for a in auxiliaries_by_id.values():
            if a.name == "万魂幡":
                soul_banner = a
                break
        
        if not soul_banner:
            pytest.skip("万魂幡 not found in config")
            
        import copy
        banner = copy.deepcopy(soul_banner)
        
        # 初始状态
        assert "吞噬魂魄" not in banner.get_info()
        
        # 增加魂魄
        banner.special_data["devoured_souls"] = 100
        
        # 检查显示更新
        assert "吞噬魂魄：100" in banner.get_info()
        assert "吞噬魂魄：100" in banner.get_detailed_info()
        
        # 检查结构化信息
        struct = banner.get_structured_info()
        assert "已吞噬魂魄：100" in struct["desc"]

    def test_grade_renaming_compatibility(self):
        """测试 realm 改名后的兼容性（如果有必要）"""
        # 确保 weapon 和 auxiliary 确实有 realm 属性
        weapon = next(iter(weapons_by_id.values()))
        assert hasattr(weapon, 'realm')
        assert not hasattr(weapon, 'grade') # 确保改名彻底
        
        aux = next(iter(auxiliaries_by_id.values()))
        assert hasattr(aux, 'realm')
        assert not hasattr(aux, 'grade') # 确保改名彻底