94 lines
3.7 KiB
Python
94 lines
3.7 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
import asyncio
|
|
|
|
from src.classes.action.play import Reading, TeaTasting, Traveling, ZitherPlaying
|
|
from src.classes.mutual_action.play import TeaParty, Chess
|
|
from src.classes.event import Event
|
|
from src.utils.config import CONFIG
|
|
|
|
class TestActionPlay:
|
|
|
|
@pytest.fixture
|
|
def play_avatar(self, dummy_avatar):
|
|
"""配置一个适合消遣的角色"""
|
|
# 确保没有初始临时效果
|
|
dummy_avatar.temporary_effects = []
|
|
return dummy_avatar
|
|
|
|
def test_single_play_action_instantiation(self, play_avatar):
|
|
"""测试单人消遣动作实例化"""
|
|
world = play_avatar.world
|
|
actions = [Reading, TeaTasting, Traveling, ZitherPlaying]
|
|
|
|
for action_cls in actions:
|
|
action = action_cls(play_avatar, world)
|
|
assert action.duration_months == 1
|
|
assert action.can_start()[0] is True
|
|
|
|
@patch('src.classes.action.play.random.random')
|
|
def test_single_play_benefit_trigger(self, mock_random, play_avatar):
|
|
"""测试单人消遣触发收益"""
|
|
# mock random < 0.05 to trigger benefit
|
|
# CONFIG.play.base_benefit_probability is 0.05
|
|
mock_random.return_value = 0.01
|
|
|
|
action = Reading(play_avatar, play_avatar.world)
|
|
|
|
# Execute finish (async)
|
|
events = asyncio.run(action.finish())
|
|
|
|
# Check event content
|
|
assert len(events) == 1
|
|
# 检查是否包含突破概率提升的描述 (英文或中文，取决于环境，这里假设 conftest 强制了中文)
|
|
# conftest.py force_chinese_language fixture sets language to zh-CN
|
|
# "breakthrough probability increased by 20.0%" -> zh-CN: "心境提升，突破概率增加 20.0%"
|
|
# Let's check for "20.0%" to be safe across languages if translation fails,
|
|
# or check the translation key if possible. But here we get the formatted string.
|
|
assert "20.0%" in events[0].content
|
|
|
|
# Check effect applied
|
|
# Need to check if temporary_effects has the entry
|
|
assert len(play_avatar.temporary_effects) == 1
|
|
effect = play_avatar.temporary_effects[0]
|
|
assert effect["source"] == "play_benefit"
|
|
assert effect["effects"]["extra_breakthrough_success_rate"] == 0.2
|
|
assert effect["duration"] == 1
|
|
|
|
@patch('src.classes.action.play.random.random')
|
|
def test_single_play_no_benefit(self, mock_random, play_avatar):
|
|
"""测试单人消遣未触发收益"""
|
|
# mock random >= 0.05
|
|
mock_random.return_value = 0.1
|
|
|
|
action = Reading(play_avatar, play_avatar.world)
|
|
|
|
events = asyncio.run(action.finish())
|
|
|
|
assert len(events) == 1
|
|
assert "20.0%" not in events[0].content
|
|
assert len(play_avatar.temporary_effects) == 0
|
|
|
|
@patch('src.classes.mutual_action.play.random.random')
|
|
def test_mutual_play_benefit(self, mock_random, play_avatar):
|
|
"""测试双人消遣触发收益"""
|
|
# Setup target avatar
|
|
target_avatar = MagicMock()
|
|
target_avatar.name = "Friend"
|
|
# MagicMock doesn't have temporary_effects list by default, but add_breakthrough_rate is called on it
|
|
|
|
# mock random < 0.05
|
|
mock_random.return_value = 0.01
|
|
|
|
action = TeaParty(play_avatar, play_avatar.world)
|
|
|
|
# Simulate feedback "Accept"
|
|
action._settle_feedback(target_avatar, "Accept")
|
|
|
|
# Check initiator benefit
|
|
assert len(play_avatar.temporary_effects) == 1
|
|
assert play_avatar.temporary_effects[0]["effects"]["extra_breakthrough_success_rate"] == 0.2
|
|
|
|
# Check target benefit
|
|
target_avatar.add_breakthrough_rate.assert_called_with(0.2)
|