325 lines
13 KiB
Python
325 lines
13 KiB
Python
import pytest
|
||
import asyncio
|
||
from typing import Dict
|
||
|
||
from src.classes.elixir import Elixir, ElixirType, ConsumedElixir, _load_elixirs
|
||
from src.classes.avatar import Avatar
|
||
from src.classes.cultivation import Realm
|
||
from src.classes.effect import EffectsMixin
|
||
from src.classes.calendar import create_month_stamp, Year, Month
|
||
|
||
# ==========================================
|
||
# Fixtures
|
||
# ==========================================
|
||
|
||
@pytest.fixture
|
||
def test_elixirs() -> Dict[str, Elixir]:
|
||
"""
|
||
手动构造一组测试用丹药，避免依赖外部CSV配置
|
||
"""
|
||
# 1. 练气期破境丹（普通）
|
||
breakthrough = Elixir(
|
||
id=1001,
|
||
name="测试破境丹",
|
||
realm=Realm.Qi_Refinement,
|
||
type=ElixirType.Breakthrough,
|
||
desc="测试用",
|
||
price=100,
|
||
effects=[{"duration_month": 10, "extra_breakthrough_success_rate": 0.1}],
|
||
effect_desc="突破率+10%"
|
||
)
|
||
|
||
# 2. 练气期延寿丹（无限时长）
|
||
lifespan_qi = Elixir(
|
||
id=1002,
|
||
name="测试长生丹",
|
||
realm=Realm.Qi_Refinement,
|
||
type=ElixirType.Lifespan,
|
||
desc="测试用",
|
||
price=200,
|
||
effects=[{"extra_max_lifespan": 10}], # 无 duration_month 视为永久
|
||
effect_desc="寿元+10"
|
||
)
|
||
|
||
# 3. 筑基期延寿丹（境界更高）
|
||
lifespan_found = Elixir(
|
||
id=1003,
|
||
name="测试筑基长生丹",
|
||
realm=Realm.Foundation_Establishment,
|
||
type=ElixirType.Lifespan,
|
||
desc="测试用",
|
||
price=500,
|
||
effects=[{"extra_max_lifespan": 20}],
|
||
effect_desc="寿元+20"
|
||
)
|
||
|
||
# 4. 燃血丹（混合时效：前3个月加攻，前10个月减防）
|
||
# 注意：模拟csv中的 [{"duration_month": 3, "val": 1}, {"duration_month": 10, "val": -1}]
|
||
burn_blood = Elixir(
|
||
id=1004,
|
||
name="测试燃血丹",
|
||
realm=Realm.Qi_Refinement,
|
||
type=ElixirType.BurnBlood,
|
||
desc="测试用",
|
||
price=50,
|
||
effects=[
|
||
{"duration_month": 3, "extra_battle_strength_points": 10},
|
||
{"duration_month": 10, "extra_defense_points": -5}
|
||
],
|
||
effect_desc="爆发"
|
||
)
|
||
|
||
return {
|
||
"breakthrough": breakthrough,
|
||
"lifespan_qi": lifespan_qi,
|
||
"lifespan_found": lifespan_found,
|
||
"burn_blood": burn_blood
|
||
}
|
||
|
||
# ==========================================
|
||
# Tests
|
||
# ==========================================
|
||
|
||
class TestElixirBasic:
|
||
"""测试基本加载和数据结构"""
|
||
|
||
def test_load_config_structure(self):
|
||
"""测试从实际CSV加载的机制是否正常工作（不校验具体数值，只校验结构）"""
|
||
ids, names = _load_elixirs()
|
||
assert len(ids) > 0, "应该能加载到丹药数据"
|
||
first_elixir = list(ids.values())[0]
|
||
assert isinstance(first_elixir, Elixir)
|
||
assert isinstance(first_elixir.effects, (dict, list))
|
||
|
||
def test_elixir_info(self, test_elixirs):
|
||
"""测试信息显示"""
|
||
elixir = test_elixirs["breakthrough"]
|
||
info = elixir.get_detailed_info()
|
||
assert "测试破境丹" in info
|
||
assert "练气" in info
|
||
assert "破境" in info
|
||
|
||
class TestConsumption:
|
||
"""测试服用逻辑"""
|
||
|
||
def test_consume_success(self, dummy_avatar, test_elixirs):
|
||
"""测试正常服用"""
|
||
elixir = test_elixirs["breakthrough"]
|
||
dummy_avatar.consume_elixir(elixir)
|
||
|
||
assert len(dummy_avatar.elixirs) == 1
|
||
record = dummy_avatar.elixirs[0]
|
||
assert record.elixir.id == elixir.id
|
||
assert record.consume_time == int(dummy_avatar.world.month_stamp)
|
||
|
||
def test_consume_realm_restriction(self, dummy_avatar, test_elixirs):
|
||
"""测试境界限制：练气期不能吃筑基期丹药"""
|
||
# 确认当前是练气期
|
||
assert dummy_avatar.cultivation_progress.realm == Realm.Qi_Refinement
|
||
|
||
elixir = test_elixirs["lifespan_found"] # 筑基期丹药
|
||
success = dummy_avatar.consume_elixir(elixir)
|
||
|
||
assert not success
|
||
assert len(dummy_avatar.elixirs) == 0
|
||
|
||
def test_consume_duplicate_lifespan(self, dummy_avatar, test_elixirs):
|
||
"""
|
||
测试：同境界延寿丹再次服用，不会额外再次增加寿元。
|
||
"""
|
||
elixir = test_elixirs["lifespan_qi"]
|
||
|
||
# 1. 第一次服用
|
||
base_lifespan = dummy_avatar.age.max_lifespan
|
||
success1 = dummy_avatar.consume_elixir(elixir)
|
||
assert success1
|
||
|
||
# 检查属性变化
|
||
new_lifespan = dummy_avatar.age.max_lifespan
|
||
# 注意：这里假设 EffectsMixin 已经生效。如果有 extra_max_lifespan 效果，
|
||
# recalc_effects 应该会把 10 加到 max_lifespan 上
|
||
# 如果 EffectsMixin 的实现是将 effect 存在 self.effects_dict 中，
|
||
# 而 Age.max_lifespan 是通过 property 计算 base + extra，那么这里应该能读到。
|
||
# 如果是直接修改数值，则需要看具体实现。
|
||
# 查看 avatar/core.py，Avatar 继承了 EffectsMixin。
|
||
# 延寿效果通常对应 extra_max_lifespan，这应该在 Age 类或者 Avatar 属性计算中体现。
|
||
# 假设 EffectsMixin 提供了 get_effect_value("extra_max_lifespan")
|
||
|
||
# 我们通过 checking effect 值来验证
|
||
assert dummy_avatar.effects.get("extra_max_lifespan") == 10
|
||
|
||
# 2. 立即再次服用（应该失败）
|
||
success2 = dummy_avatar.consume_elixir(elixir)
|
||
assert not success2
|
||
assert len(dummy_avatar.elixirs) == 1
|
||
assert dummy_avatar.effects.get("extra_max_lifespan") == 10
|
||
|
||
# 3. 推进时间很久（延寿丹是永久的，依然不能服用）
|
||
dummy_avatar.world.month_stamp += 1000
|
||
success3 = dummy_avatar.consume_elixir(elixir)
|
||
assert not success3
|
||
assert len(dummy_avatar.elixirs) == 1
|
||
|
||
def test_consume_duplicate_expired(self, dummy_avatar, test_elixirs):
|
||
"""测试：过期后的丹药可以再次服用"""
|
||
elixir = test_elixirs["breakthrough"] # 持续10个月
|
||
|
||
# 1. 服用
|
||
dummy_avatar.consume_elixir(elixir)
|
||
assert len(dummy_avatar.elixirs) == 1
|
||
|
||
# 2. 推进时间到过期 (当前 + 11)
|
||
# consume_time 是服用时的 month_stamp
|
||
# expire_time = consume_time + 10
|
||
# current >= expire_time 时 is_completely_expired 为 True
|
||
|
||
# 先推进9个月（还剩1个月）
|
||
dummy_avatar.world.month_stamp += 9
|
||
# 清理逻辑通常是被动触发的，这里我们手动触发或依赖服用时的检查
|
||
# consume_elixir 内部会检查 is_completely_expired
|
||
success_fail = dummy_avatar.consume_elixir(elixir)
|
||
assert not success_fail # 还没过期，不能吃
|
||
|
||
# 再推进2个月（共+11），过期了
|
||
dummy_avatar.world.month_stamp += 2
|
||
|
||
# 3. 再次服用（应该成功）
|
||
# 注意：consume_elixir 内部逻辑是 "若已服用过同种且【未失效】的丹药，则无效"
|
||
# 即使旧的记录还在 list 里（还没被清理），只要 is_completely_expired 返回 True，就可以再吃
|
||
success_ok = dummy_avatar.consume_elixir(elixir)
|
||
assert success_ok
|
||
|
||
# 此时列表里可能有2个记录（旧的没清理的话），或者旧的被清理了（取决于实现，这里不强制要求清理，只要求能吃）
|
||
# 查看代码：consume_elixir 没有清理逻辑，只是 append。清理逻辑在 process_elixir_expiration
|
||
assert len(dummy_avatar.elixirs) == 2
|
||
|
||
# 验证一个是过期的，一个是新的
|
||
expired_one = dummy_avatar.elixirs[0]
|
||
new_one = dummy_avatar.elixirs[1]
|
||
assert expired_one.is_completely_expired(int(dummy_avatar.world.month_stamp))
|
||
assert not new_one.is_completely_expired(int(dummy_avatar.world.month_stamp))
|
||
|
||
class TestExpirationAndEffects:
|
||
"""测试时效和效果计算"""
|
||
|
||
def test_mixed_duration_effects(self, dummy_avatar, test_elixirs):
|
||
"""测试燃血丹的多段效果"""
|
||
# 先计算基础值（排除丹药影响）
|
||
dummy_avatar.recalc_effects()
|
||
base_atk = dummy_avatar.effects.get("extra_battle_strength_points", 0)
|
||
base_def = dummy_avatar.effects.get("extra_defense_points", 0)
|
||
|
||
elixir = test_elixirs["burn_blood"]
|
||
# effects: duration 3 (atk+10), duration 10 (def-5)
|
||
|
||
dummy_avatar.consume_elixir(elixir)
|
||
start_time = int(dummy_avatar.world.month_stamp)
|
||
|
||
# 1. 刚服用 (Month 0)
|
||
# 效果全在
|
||
active = dummy_avatar.elixirs[0].get_active_effects(start_time)
|
||
assert len(active) == 2
|
||
|
||
dummy_avatar.recalc_effects()
|
||
# 断言应当是 基础值 + 丹药增量
|
||
assert dummy_avatar.effects.get("extra_battle_strength_points") == base_atk + 10
|
||
assert dummy_avatar.effects.get("extra_defense_points") == base_def - 5
|
||
|
||
# 2. 第 2 个月 (Month 0 + 2 < 3)
|
||
current = start_time + 2
|
||
active = dummy_avatar.elixirs[0].get_active_effects(current)
|
||
assert len(active) == 2
|
||
|
||
# 3. 第 5 个月 (Month 0 + 5)
|
||
# 攻击buff(3个月)消失，防御debuff(10个月)还在
|
||
current = start_time + 5
|
||
dummy_avatar.world.month_stamp = create_month_stamp(Year(2000), Month.JANUARY) + 5 # Reset timestamp just to be safe
|
||
# update world time manually
|
||
dummy_avatar.world.month_stamp = start_time + 5
|
||
|
||
# 主动触发清理或重算
|
||
dummy_avatar.process_elixir_expiration(current) # 还没完全过期，不会移除Elixir对象
|
||
dummy_avatar.recalc_effects() # 但 active effects 变了，重算属性
|
||
|
||
# 验证 active effects
|
||
active = dummy_avatar.elixirs[0].get_active_effects(current)
|
||
assert len(active) == 1
|
||
assert "extra_battle_strength_points" not in active[0]
|
||
assert active[0]["extra_defense_points"] == -5
|
||
|
||
# 验证属性面板
|
||
# 注意：Avatar.recalc_effects() 会调用 self.get_all_effects()
|
||
# get_all_effects 会遍历 elixirs 调用 get_active_effects
|
||
assert dummy_avatar.effects.get("extra_battle_strength_points", 0) == base_atk
|
||
assert dummy_avatar.effects.get("extra_defense_points") == base_def - 5
|
||
|
||
# 4. 第 11 个月 (Month 0 + 11)
|
||
# 全部过期
|
||
current = start_time + 11
|
||
dummy_avatar.world.month_stamp = current
|
||
|
||
# 此时 is_completely_expired 应该是 True
|
||
assert dummy_avatar.elixirs[0].is_completely_expired(current)
|
||
|
||
# 触发清理
|
||
dummy_avatar.process_elixir_expiration(current)
|
||
assert len(dummy_avatar.elixirs) == 0
|
||
|
||
# 属性恢复
|
||
dummy_avatar.recalc_effects()
|
||
assert dummy_avatar.effects.get("extra_defense_points", 0) == base_def
|
||
|
||
def test_infinite_duration(self, dummy_avatar, test_elixirs):
|
||
"""测试无限时长（延寿丹）"""
|
||
elixir = test_elixirs["lifespan_qi"]
|
||
dummy_avatar.consume_elixir(elixir)
|
||
|
||
# 推进 100 年
|
||
dummy_avatar.world.month_stamp += 1200
|
||
current = int(dummy_avatar.world.month_stamp)
|
||
|
||
assert not dummy_avatar.elixirs[0].is_completely_expired(current)
|
||
dummy_avatar.process_elixir_expiration(current)
|
||
assert len(dummy_avatar.elixirs) == 1
|
||
|
||
# 效果依然在
|
||
dummy_avatar.recalc_effects()
|
||
assert dummy_avatar.effects.get("extra_max_lifespan") == 10
|
||
|
||
class TestSimulatorIntegration:
|
||
"""测试模拟器集成"""
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_passive_update_loop(self, base_world, dummy_avatar, test_elixirs, mock_llm_managers):
|
||
"""测试在 Simulator step 中自动清理过期丹药"""
|
||
from src.sim.simulator import Simulator
|
||
|
||
# 准备环境
|
||
sim = Simulator(base_world)
|
||
# 确保 dummy_avatar 在世界管理器中（虽然 base_world 可能没有自动加进去）
|
||
base_world.avatar_manager.avatars[dummy_avatar.id] = dummy_avatar
|
||
|
||
# 构造一个超短效丹药 (持续1个月)
|
||
short_elixir = Elixir(
|
||
id=9999,
|
||
name="瞬时丹",
|
||
realm=Realm.Qi_Refinement,
|
||
type=ElixirType.Breakthrough,
|
||
desc="测试用",
|
||
price=1,
|
||
effects=[{"duration_month": 1, "test_val": 1}],
|
||
effect_desc="瞬时"
|
||
)
|
||
dummy_avatar.consume_elixir(short_elixir)
|
||
|
||
# 只需运行 2 个 step (month 0->1 还在, month 1->2 过期清理)
|
||
# month 0: consume. expires at 0+1=1.
|
||
# step 1: check expiration (time=0). not expired. time becomes 1.
|
||
# step 2: check expiration (time=1). expired! cleaned. time becomes 2.
|
||
for _ in range(2):
|
||
await sim.step()
|
||
|
||
# 检查是否自动清理
|
||
assert len(dummy_avatar.elixirs) == 0
|
||
|