#!/usr/bin/env python3 """ 新战斗引擎系统测试 专注于战斗系统，不包含卡牌功能 """ import sys import os # 添加项目根目录到Python路径 sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..')) from kards_battle.core.battle_engine import BattleEngine from kards_battle.core.enums import LineType from kards_battle.examples.sample_units import create_german_infantry, create_american_gi, create_german_88mm_gun def test_kredits_slot_system(): """测试Kredits Slot系统""" print("\n💰 Kredits Slot系统测试") print("-" * 40) engine = BattleEngine("Germany", "USA", debug_mode=True) # 初始状态检查 print(f"初始状态:") print(f" 德军 Slot: {engine.get_kredits_slot(0)}, Kredits: {engine.get_kredits(0)}") print(f" 美军 Slot: {engine.get_kredits_slot(1)}, Kredits: {engine.get_kredits(1)}") # 模拟多回合，检查槽数增长 results = [] for turn in range(1, 25): # 测试到第25回合，确保能达到12 result = engine.end_turn() player = result["new_active_player"] slot = result["kredits_slot"] kredits = result["kredits"] # 只打印关键回合 if turn <= 5 or slot >= 10 or turn == 24: print(f"回合{turn+1}: {player} - Slot: {slot}, Kredits: {kredits}") results.append((turn+1, player, slot, kredits)) # 验证规则 print("\n验证规则:") slot_12_reached = any(slot >= 12 for _, _, slot, _ in results) kredits_equal_slot = all(slot == kredits for _, _, slot, kredits in results) print(f"✅ Slot达到12: {'是' if slot_12_reached else '否'}") print(f"✅ Kredits等于Slot: {'是' if kredits_equal_slot else '否'}") assert slot_12_reached, "Kredits Slot should reach 12" assert kredits_equal_slot, "Kredits should always equal Kredits Slot" def test_direct_unit_deployment(): """测试直接单位部署功能""" print("\n🚀 直接单位部署测试") print("-" * 40) engine = BattleEngine("Germany", "USA", debug_mode=True) # 创建单位 german_inf = create_german_infantry() american_gi = create_american_gi() german_inf.stats.operation_cost = 1 american_gi.stats.operation_cost = 1 # 部署德军单位 deploy_result1 = engine.deploy_unit_to_support(german_inf, 0) print(f"德军步兵部署: {deploy_result1['success']}") # 部署美军单位 deploy_result2 = engine.deploy_unit_to_support(american_gi, 1) print(f"美军GI部署: {deploy_result2['success']}") print(f"\n战场状态:") print(engine.battlefield) success = deploy_result1["success"] and deploy_result2["success"] print(f"✅ 部署测试: {'通过' if success else '失败'}") assert success, "Unit deployment should succeed" def test_movement_and_attack(): """测试移动和攻击功能""" print("\n⚔️ 移动和攻击测试") print("-" * 40) # Setup units for testing engine = BattleEngine("Germany", "USA", debug_mode=True) german_inf = create_german_infantry() american_gi = create_american_gi() german_inf.stats.operation_cost = 1 american_gi.stats.operation_cost = 1 # Deploy units deploy_result1 = engine.deploy_unit_to_support(german_inf, 0) deploy_result2 = engine.deploy_unit_to_support(american_gi, 1) success = deploy_result1["success"] and deploy_result2["success"] if not success: print("❌ 部署失败，无法测试移动攻击") assert False, "Unit deployment failed" # 等待一回合让单位可以行动 engine.end_turn() engine.end_turn() # 给足够的Kredits engine.debug_set_kredits(0, kredits=10) engine.debug_set_kredits(1, kredits=10) # 德军移动到前线 print(f"\n德军Kredits: {engine.get_kredits(0)}") move_result = engine.move_unit(german_inf.id, (LineType.FRONT, 0), 0) print(f"德军步兵移动到前线: {move_result['success']}") print(f"德军剩余Kredits: {engine.get_kredits(0)}") if move_result["success"]: print(f"前线控制权: {move_result['front_line_controller']}") print(f"战场状态:") print(engine.battlefield) # 切换到美军回合 engine.end_turn() # 美军攻击德军 print(f"\n美军Kredits: {engine.get_kredits(1)}") attack_result = engine.attack_target(american_gi.id, german_inf.id, 1) print(f"美军攻击德军: {attack_result['success']}") if attack_result["success"]: print(f"造成伤害: {attack_result.get('damage_dealt', 0)}") print(f"反击伤害: {attack_result.get('counter_damage', 0)}") print(f"摧毁单位: {len(attack_result.get('units_destroyed', []))}") print(f"美军剩余Kredits: {engine.get_kredits(1)}") movement_success = move_result.get("success", False) attack_success = attack_result.get("success", False) print(f"✅ 移动测试: {'通过' if movement_success else '失败'}") print(f"✅ 攻击测试: {'通过' if attack_success else '失败'}") assert movement_success, "Unit movement should succeed" assert attack_success, "Unit attack should succeed" def test_debug_functions(): """测试DEBUG功能""" print("\n🔧 DEBUG功能测试") print("-" * 40) engine = BattleEngine("Germany", "USA", debug_mode=True) # 测试设置Kredits debug_result1 = engine.debug_set_kredits(0, kredits=15, kredits_slot=8) print(f"设置德军Kredits/Slot: {debug_result1['success']}") print(f" 德军 Kredits: {engine.get_kredits(0)}, Slot: {engine.get_kredits_slot(0)}") # 测试设置HQ防御 debug_result2 = engine.debug_set_hq_defense(1, 10) print(f"设置美军HQ防御: {debug_result2['success']}") print(f" 美军HQ防御: {engine.battlefield.player2_hq.current_defense}") # 创建单位测试属性设置 german_inf = create_german_infantry() engine.deploy_unit_to_support(german_inf, 0) debug_result3 = engine.debug_set_unit_stats(german_inf.id, attack=10, defense=15) print(f"设置单位属性: {debug_result3['success']}") print(f" 单位攻击力: {german_inf.stats.attack}, 防御力: {german_inf.stats.current_defense}") # 测试非DEBUG模式 engine_no_debug = BattleEngine("A", "B", debug_mode=False) debug_result4 = engine_no_debug.debug_set_kredits(0, kredits=20) print(f"非DEBUG模式限制: {not debug_result4['success']}") all_success = all([ debug_result1["success"], debug_result2["success"], debug_result3["success"], not debug_result4["success"] # 这个应该失败 ]) print(f"✅ DEBUG功能测试: {'通过' if all_success else '失败'}") assert all_success, "All debug functions should work correctly" def test_artillery_no_counter(): """测试火炮不受反击""" print("\n🎯 火炮不受反击测试") print("-" * 40) engine = BattleEngine("Germany", "USA", debug_mode=True) # 部署单位 artillery = create_german_88mm_gun() american_gi = create_american_gi() artillery.stats.operation_cost = 1 american_gi.stats.operation_cost = 1 engine.deploy_unit_to_support(artillery, 0) engine.deploy_unit_to_support(american_gi, 1) # 给足够Kredits engine.debug_set_kredits(0, kredits=10) # 火炮攻击GI attack_result = engine.attack_target(artillery.id, american_gi.id, 0) success = attack_result.get("success", False) no_counter = attack_result.get("counter_damage", -1) == 0 print(f"火炮攻击成功: {'是' if success else '否'}") print(f"火炮不受反击: {'是' if no_counter else '否'}") print(f"造成伤害: {attack_result.get('damage_dealt', 0)}") result = success and no_counter print(f"✅ 火炮测试: {'通过' if result else '失败'}") assert result, "Artillery should not receive counter damage" def run_all_tests(): """运行所有测试""" print("🧪 新战斗引擎系统全面测试") print("=" * 50) tests = [ ("Kredits Slot系统", test_kredits_slot_system), ("直接部署功能", lambda: test_direct_unit_deployment()[0]), ("移动和攻击", test_movement_and_attack), ("DEBUG功能", test_debug_functions), ("火炮不受反击", test_artillery_no_counter), ] results = {} for name, test_func in tests: try: results[name] = test_func() except Exception as e: print(f"❌ {name} 测试异常: {e}") results[name] = False print("\n📊 测试结果汇总") print("=" * 50) all_passed = True for name, passed in results.items(): status = "✅ 通过" if passed else "❌ 失败" print(f"{name}: {status}") if not passed: all_passed = False print(f"\n总体结果: {'🎉 全部通过!' if all_passed else '❌ 有测试失败'}") return all_passed if __name__ == "__main__": run_all_tests()