e3bf36bcd4
Verify that: 1. Avatar without sect knows current location's region 2. Avatar with sect knows their sect headquarters 3. Avatar without sect does not automatically know sect regions 4. Avatar only knows their own sect's headquarters, not others
212 lines
6.9 KiB
Python
212 lines
6.9 KiB
Python
"""
|
||
测试 Avatar._init_known_regions 的行为。
|
||
|
||
验证：
|
||
1. 无宗门的 avatar：known_regions 只包含当前位置的区域（如果有）
|
||
2. 有宗门的 avatar：known_regions 包含宗门驻地
|
||
"""
|
||
import pytest
|
||
from unittest.mock import MagicMock
|
||
|
||
from src.classes.map import Map
|
||
from src.classes.tile import TileType
|
||
from src.classes.world import World
|
||
from src.classes.calendar import Month, Year, create_month_stamp
|
||
from src.classes.avatar import Avatar, Gender
|
||
from src.classes.age import Age
|
||
from src.classes.cultivation import Realm
|
||
from src.classes.root import Root
|
||
from src.classes.alignment import Alignment
|
||
from src.classes.sect import sects_by_id
|
||
from src.classes.sect_region import SectRegion
|
||
from src.classes.region import NormalRegion
|
||
from src.utils.id_generator import get_avatar_id
|
||
|
||
|
||
@pytest.fixture
|
||
def world_with_sect_region():
|
||
"""创建一个带有宗门驻地的世界"""
|
||
width, height = 10, 10
|
||
game_map = Map(width=width, height=height)
|
||
|
||
# 创建 tiles
|
||
for x in range(width):
|
||
for y in range(height):
|
||
game_map.create_tile(x, y, TileType.PLAIN)
|
||
|
||
# 获取一个真实的宗门（明心剑宗，id=1）
|
||
sect = sects_by_id.get(1)
|
||
assert sect is not None, "宗门 ID=1 应该存在"
|
||
|
||
# 创建对应的 SectRegion
|
||
sect_region = SectRegion(
|
||
id=401,
|
||
name="连霞山",
|
||
desc="测试宗门驻地",
|
||
cors=[(5, 5)],
|
||
sect_name=sect.name,
|
||
sect_id=sect.id
|
||
)
|
||
|
||
# 注册到 map
|
||
game_map.regions[sect_region.id] = sect_region
|
||
game_map.sect_regions[sect_region.id] = sect_region
|
||
|
||
# 将区域绑定到 tile
|
||
tile = game_map.get_tile(5, 5)
|
||
tile.region = sect_region
|
||
|
||
world = World(map=game_map, month_stamp=create_month_stamp(Year(1), Month.JANUARY))
|
||
return world, sect, sect_region
|
||
|
||
|
||
@pytest.fixture
|
||
def world_with_normal_region():
|
||
"""创建一个带有普通区域的世界"""
|
||
width, height = 10, 10
|
||
game_map = Map(width=width, height=height)
|
||
|
||
for x in range(width):
|
||
for y in range(height):
|
||
game_map.create_tile(x, y, TileType.PLAIN)
|
||
|
||
# 创建普通区域
|
||
normal_region = NormalRegion(
|
||
id=101,
|
||
name="测试区域",
|
||
desc="一片普通的区域",
|
||
cors=[(0, 0)]
|
||
)
|
||
|
||
game_map.regions[normal_region.id] = normal_region
|
||
|
||
# 绑定到 tile
|
||
tile = game_map.get_tile(0, 0)
|
||
tile.region = normal_region
|
||
|
||
world = World(map=game_map, month_stamp=create_month_stamp(Year(1), Month.JANUARY))
|
||
return world, normal_region
|
||
|
||
|
||
class TestInitKnownRegions:
|
||
"""测试 Avatar._init_known_regions"""
|
||
|
||
def test_avatar_without_sect_knows_current_region(self, world_with_normal_region):
|
||
"""无宗门的 avatar 应该知道当前位置的区域"""
|
||
world, normal_region = world_with_normal_region
|
||
|
||
avatar = Avatar(
|
||
world=world,
|
||
name="散修测试",
|
||
id=get_avatar_id(),
|
||
birth_month_stamp=create_month_stamp(Year(2000), Month.JANUARY),
|
||
age=Age(20, Realm.Qi_Refinement),
|
||
gender=Gender.MALE,
|
||
pos_x=0,
|
||
pos_y=0,
|
||
root=Root.GOLD,
|
||
alignment=Alignment.RIGHTEOUS,
|
||
sect=None # 散修
|
||
)
|
||
|
||
# 设置 tile（模拟正常初始化流程）
|
||
avatar.tile = world.map.get_tile(0, 0)
|
||
avatar._init_known_regions()
|
||
|
||
assert normal_region.id in avatar.known_regions, (
|
||
f"散修应该知道当前位置的区域，但 known_regions={avatar.known_regions}"
|
||
)
|
||
|
||
def test_avatar_with_sect_knows_headquarters(self, world_with_sect_region):
|
||
"""有宗门的 avatar 应该知道宗门驻地位置"""
|
||
world, sect, sect_region = world_with_sect_region
|
||
|
||
avatar = Avatar(
|
||
world=world,
|
||
name="宗门弟子测试",
|
||
id=get_avatar_id(),
|
||
birth_month_stamp=create_month_stamp(Year(2000), Month.JANUARY),
|
||
age=Age(20, Realm.Qi_Refinement),
|
||
gender=Gender.MALE,
|
||
pos_x=0,
|
||
pos_y=0, # 不在宗门驻地
|
||
root=Root.GOLD,
|
||
alignment=Alignment.RIGHTEOUS,
|
||
sect=sect # 有宗门
|
||
)
|
||
|
||
# _init_known_regions 在 __post_init__ 中自动调用
|
||
assert sect_region.id in avatar.known_regions, (
|
||
f"宗门弟子应该知道宗门驻地位置，但 known_regions={avatar.known_regions}"
|
||
)
|
||
|
||
def test_avatar_without_sect_does_not_know_sect_region(self, world_with_sect_region):
|
||
"""无宗门的 avatar 不应该自动知道宗门驻地"""
|
||
world, sect, sect_region = world_with_sect_region
|
||
|
||
avatar = Avatar(
|
||
world=world,
|
||
name="散修测试",
|
||
id=get_avatar_id(),
|
||
birth_month_stamp=create_month_stamp(Year(2000), Month.JANUARY),
|
||
age=Age(20, Realm.Qi_Refinement),
|
||
gender=Gender.MALE,
|
||
pos_x=0,
|
||
pos_y=0,
|
||
root=Root.GOLD,
|
||
alignment=Alignment.RIGHTEOUS,
|
||
sect=None # 散修
|
||
)
|
||
|
||
# 散修不应该知道宗门驻地（除非在那个位置）
|
||
assert sect_region.id not in avatar.known_regions, (
|
||
f"散修不应该自动知道宗门驻地，但 known_regions={avatar.known_regions}"
|
||
)
|
||
|
||
def test_avatar_knows_only_their_sect_headquarters(self, world_with_sect_region):
|
||
"""avatar 只应该知道自己宗门的驻地，不知道其他宗门"""
|
||
world, sect, sect_region = world_with_sect_region
|
||
|
||
# 添加另一个宗门的驻地
|
||
other_sect = sects_by_id.get(2) # 百兽宗
|
||
assert other_sect is not None
|
||
|
||
other_sect_region = SectRegion(
|
||
id=402,
|
||
name="玄灵洞",
|
||
desc="另一个宗门驻地",
|
||
cors=[(7, 7)],
|
||
sect_name=other_sect.name,
|
||
sect_id=other_sect.id
|
||
)
|
||
world.map.regions[other_sect_region.id] = other_sect_region
|
||
world.map.sect_regions[other_sect_region.id] = other_sect_region
|
||
|
||
# 创建明心剑宗弟子
|
||
avatar = Avatar(
|
||
world=world,
|
||
name="明心剑宗弟子",
|
||
id=get_avatar_id(),
|
||
birth_month_stamp=create_month_stamp(Year(2000), Month.JANUARY),
|
||
age=Age(20, Realm.Qi_Refinement),
|
||
gender=Gender.MALE,
|
||
pos_x=0,
|
||
pos_y=0,
|
||
root=Root.GOLD,
|
||
alignment=Alignment.RIGHTEOUS,
|
||
sect=sect # 明心剑宗
|
||
)
|
||
|
||
# 应该知道自己宗门的驻地
|
||
assert sect_region.id in avatar.known_regions, (
|
||
"应该知道自己宗门的驻地"
|
||
)
|
||
# 不应该知道其他宗门的驻地
|
||
assert other_sect_region.id not in avatar.known_regions, (
|
||
"不应该知道其他宗门的驻地"
|
||
)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
pytest.main([__file__, "-v"])
|