85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
import pytest
|
||
import os
|
||
from pathlib import Path
|
||
from src.sim.save.save_game import save_game
|
||
from src.sim.load.load_game import load_game
|
||
from src.classes.avatar import Avatar
|
||
from src.classes.death_reason import DeathReason, DeathType
|
||
from src.classes.calendar import MonthStamp, Month, Year, create_month_stamp
|
||
|
||
def test_dead_avatar_stays_dead_after_load(base_world, dummy_avatar):
|
||
"""
|
||
测试死亡的角色在读档后是否仍然被正确归类为死者，
|
||
而不是复活出现在活人列表中。
|
||
"""
|
||
# 1. 准备环境
|
||
dummy_avatar.weapon = None
|
||
base_world.avatar_manager.register_avatar(dummy_avatar)
|
||
|
||
assert dummy_avatar.id in base_world.avatar_manager.avatars
|
||
assert dummy_avatar.id not in base_world.avatar_manager.dead_avatars
|
||
assert not dummy_avatar.is_dead
|
||
|
||
# 2. 杀死角色
|
||
death_time = base_world.month_stamp
|
||
dummy_avatar.set_dead("Test Death", death_time)
|
||
base_world.avatar_manager.handle_death(dummy_avatar.id)
|
||
|
||
assert dummy_avatar.is_dead
|
||
assert dummy_avatar.id not in base_world.avatar_manager.avatars
|
||
assert dummy_avatar.id in base_world.avatar_manager.dead_avatars
|
||
|
||
# 3. 保存游戏
|
||
from src.sim.simulator import Simulator
|
||
simulator = Simulator(base_world)
|
||
|
||
success, save_filename = save_game(base_world, simulator, existed_sects=[])
|
||
assert success
|
||
|
||
from src.utils.config import CONFIG
|
||
save_path = CONFIG.paths.saves / save_filename
|
||
|
||
# 4. 读取游戏
|
||
loaded_world, loaded_sim, _ = load_game(save_path)
|
||
|
||
# 5. 验证读档后的状态
|
||
loaded_avatar = loaded_world.avatar_manager.get_avatar(dummy_avatar.id)
|
||
assert loaded_avatar is not None
|
||
assert loaded_avatar.is_dead
|
||
|
||
assert loaded_avatar.id not in loaded_world.avatar_manager.avatars, "死者不应出现在活人列表 avatars 中"
|
||
assert loaded_avatar.id in loaded_world.avatar_manager.dead_avatars, "死者应该出现在死者列表 dead_avatars 中"
|
||
|
||
living_ids = [a.id for a in loaded_world.avatar_manager.get_living_avatars()]
|
||
assert loaded_avatar.id not in living_ids, "死者不应被 get_living_avatars() 返回"
|
||
|
||
def test_cleanup_long_dead_avatars(base_world, dummy_avatar):
|
||
"""
|
||
测试清理死亡超过20年的角色逻辑
|
||
"""
|
||
# 1. 准备环境
|
||
base_world.avatar_manager.register_avatar(dummy_avatar)
|
||
|
||
# 2. 模拟角色死亡（死亡时间为 Year 1, Month 1）
|
||
death_time = create_month_stamp(Year(1), Month.JANUARY)
|
||
dummy_avatar.set_dead("Old Age", death_time)
|
||
base_world.avatar_manager.handle_death(dummy_avatar.id)
|
||
|
||
assert dummy_avatar.id in base_world.avatar_manager.dead_avatars
|
||
|
||
# 3. 未满20年，不应清理
|
||
# Year 20, Month 1 (经过19年)
|
||
current_time_19y = create_month_stamp(Year(20), Month.JANUARY)
|
||
cleaned_count = base_world.avatar_manager.cleanup_long_dead_avatars(current_time_19y, threshold_years=20)
|
||
assert cleaned_count == 0
|
||
assert dummy_avatar.id in base_world.avatar_manager.dead_avatars
|
||
|
||
# 4. 满20年，应清理
|
||
# Year 21, Month 1 (经过20年)
|
||
current_time_20y = create_month_stamp(Year(21), Month.JANUARY)
|
||
cleaned_count = base_world.avatar_manager.cleanup_long_dead_avatars(current_time_20y, threshold_years=20)
|
||
|
||
assert cleaned_count == 1
|
||
assert dummy_avatar.id not in base_world.avatar_manager.dead_avatars
|
||
assert base_world.avatar_manager.get_avatar(dummy_avatar.id) is None
|