"""测试 new_avatar 模块的角色创建逻辑."""
import pytest
from src.sim.new_avatar import make_avatars, AvatarFactory, PopulationPlanner
from src.classes.age import Age
from src.classes.cultivation import CultivationProgress


class TestAgeLifespanConstraint:
    """测试角色创建时年龄不超过寿命上限的约束."""

    def test_batch_creation_age_within_lifespan(self, base_world):
        """批量创建角色时，年龄应不超过境界寿命上限.

        注意：只有当随机生成的年龄 >= 寿命上限时才会触发调整，
        调整后的年龄会在 80%-95% 区间内。正常随机生成的年龄
        （如 77 岁 / 80 寿命）不会被调整，所以可能接近但不超过上限。
        """
        # 创建大量角色以增加覆盖率
        avatars = make_avatars(base_world, count=100)

        for avatar in avatars.values():
            max_lifespan = Age.REALM_LIFESPAN.get(
                avatar.cultivation_progress.realm, 100
            )
            assert avatar.age.age < max_lifespan, (
                f"角色 {avatar.name} 年龄 {avatar.age.age} 超过了"
                f"境界 {avatar.cultivation_progress.realm} 的寿命上限 {max_lifespan}"
            )

    def test_batch_creation_no_immediate_death(self, base_world):
        """批量创建的角色不应该一出生就处于老死状态."""
        avatars = make_avatars(base_world, count=100)

        for avatar in avatars.values():
            # 不应该有老死概率
            death_prob = avatar.age.get_death_probability()
            assert death_prob == 0.0, (
                f"角色 {avatar.name} 年龄 {avatar.age.age}/"
                f"{avatar.age.max_lifespan} 有老死概率 {death_prob}"
            )

    def test_multiple_batch_creations_consistent(self, base_world):
        """多次批量创建应该都满足年龄约束."""
        for _ in range(5):
            avatars = make_avatars(base_world, count=50)
            for avatar in avatars.values():
                max_lifespan = Age.REALM_LIFESPAN.get(
                    avatar.cultivation_progress.realm, 100
                )
                assert avatar.age.age < max_lifespan


class TestRealmLifespanMapping:
    """测试各境界的寿命上限映射."""

    def test_qi_refinement_lifespan(self, base_world):
        """练气期角色年龄应不超过80岁."""
        from src.classes.cultivation import Realm

        avatars = make_avatars(base_world, count=100)
        qi_refinement_avatars = [
            av for av in avatars.values()
            if av.cultivation_progress.realm == Realm.Qi_Refinement
        ]

        for avatar in qi_refinement_avatars:
            assert avatar.age.age < 80, (
                f"练气期角色 {avatar.name} 年龄 {avatar.age.age} 超过 80"
            )

    def test_foundation_establishment_lifespan(self, base_world):
        """筑基期角色年龄应不超过120岁."""
        from src.classes.cultivation import Realm

        avatars = make_avatars(base_world, count=100)
        fe_avatars = [
            av for av in avatars.values()
            if av.cultivation_progress.realm == Realm.Foundation_Establishment
        ]

        for avatar in fe_avatars:
            assert avatar.age.age < 120, (
                f"筑基期角色 {avatar.name} 年龄 {avatar.age.age} 超过 120"
            )