118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
from src.utils.gather import execute_gather, check_can_start_gather
|
|
from src.classes.cultivation import Realm
|
|
from src.classes.region import NormalRegion
|
|
|
|
@pytest.fixture
|
|
def mock_region(dummy_avatar):
|
|
"""设置一个 Mock 的 NormalRegion 到 avatar 所在 tile"""
|
|
real_region = NormalRegion(id=999, name="TestRegion", desc="Testing")
|
|
# Bypass post_init loading from global dicts by manually setting fields
|
|
real_region.lodes = []
|
|
real_region.animals = []
|
|
real_region.plants = []
|
|
|
|
dummy_avatar.tile.region = real_region
|
|
return real_region
|
|
|
|
@pytest.fixture
|
|
def mock_resource_material():
|
|
material = MagicMock()
|
|
material.name = "TestMaterial"
|
|
material.realm = Realm.Qi_Refinement
|
|
return material
|
|
|
|
@pytest.fixture
|
|
def mock_resource(mock_resource_material):
|
|
"""创建一个通用的资源对象 (Lode/Animal/Plant)"""
|
|
res = MagicMock()
|
|
res.realm = Realm.Qi_Refinement
|
|
res.materials = [mock_resource_material]
|
|
return res
|
|
|
|
def test_check_can_start_gather_success(dummy_avatar, mock_region, mock_resource):
|
|
"""测试采集检查通过的情况"""
|
|
mock_region.lodes = [mock_resource]
|
|
|
|
can, msg = check_can_start_gather(dummy_avatar, "lodes", "矿脉")
|
|
assert can is True
|
|
assert msg == ""
|
|
|
|
def test_check_can_start_gather_not_normal_region(dummy_avatar):
|
|
"""测试不在普通区域的情况"""
|
|
dummy_avatar.tile.region = "NotARegion"
|
|
|
|
can, msg = check_can_start_gather(dummy_avatar, "lodes", "矿脉")
|
|
assert can is False
|
|
assert "当前不在普通区域" in msg
|
|
|
|
def test_check_can_start_gather_no_resources(dummy_avatar, mock_region):
|
|
"""测试区域没有资源的情况"""
|
|
mock_region.lodes = []
|
|
|
|
can, msg = check_can_start_gather(dummy_avatar, "lodes", "矿脉")
|
|
assert can is False
|
|
assert "当前区域没有矿脉" in msg
|
|
|
|
def test_check_can_start_gather_realm_too_low(dummy_avatar, mock_region, mock_resource):
|
|
"""测试境界不足的情况"""
|
|
# 提升资源境界到筑基
|
|
mock_resource.realm = Realm.Foundation_Establishment
|
|
mock_region.lodes = [mock_resource]
|
|
# avatar 默认为练气
|
|
|
|
can, msg = check_can_start_gather(dummy_avatar, "lodes", "矿脉")
|
|
assert can is False
|
|
assert "当前区域的矿脉境界过高" in msg
|
|
|
|
def test_execute_gather_success(dummy_avatar, mock_region, mock_resource, mock_resource_material):
|
|
"""测试执行采集逻辑成功"""
|
|
mock_region.lodes = [mock_resource]
|
|
|
|
# 模拟 add_material
|
|
dummy_avatar.add_material = MagicMock()
|
|
|
|
result = execute_gather(dummy_avatar, "lodes", "extra_mine_materials")
|
|
|
|
assert "TestMaterial" in result
|
|
assert result["TestMaterial"] >= 1
|
|
dummy_avatar.add_material.assert_called_once()
|
|
|
|
# 验证获得的材料是正确的
|
|
args, _ = dummy_avatar.add_material.call_args
|
|
assert args[0] == mock_resource_material
|
|
assert args[1] >= 1
|
|
|
|
def test_execute_gather_with_extra_effect(dummy_avatar, mock_region, mock_resource):
|
|
"""测试带有加成效果的采集"""
|
|
mock_region.lodes = [mock_resource]
|
|
|
|
# effects 是只读属性，它通过合并各个组件的 effects 来计算。
|
|
# 为了测试，我们 Mock 掉 effects 属性。
|
|
with patch.object(type(dummy_avatar), 'effects', new_callable=lambda: {"extra_mine_materials": 2}):
|
|
dummy_avatar.add_material = MagicMock()
|
|
|
|
result = execute_gather(dummy_avatar, "lodes", "extra_mine_materials")
|
|
|
|
# 基础1 + 加成2 = 3
|
|
assert result["TestMaterial"] == 3
|
|
|
|
def test_execute_gather_random_selection(dummy_avatar, mock_region):
|
|
"""测试从多个资源中随机选择"""
|
|
res1 = MagicMock()
|
|
res1.realm = Realm.Qi_Refinement
|
|
res1.materials = [MagicMock(name="Material1")]
|
|
res1.materials[0].name = "Material1"
|
|
|
|
res2 = MagicMock()
|
|
res2.realm = Realm.Qi_Refinement
|
|
res2.materials = [MagicMock(name="Material2")]
|
|
res2.materials[0].name = "Material2"
|
|
|
|
mock_region.lodes = [res1, res2]
|
|
dummy_avatar.add_material = MagicMock()
|
|
|
|
execute_gather(dummy_avatar, "lodes", "extra_mine_materials")
|
|
dummy_avatar.add_material.assert_called_once()
|