100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
"""
|
||
MCP 工具测试
|
||
验证 psycho_screen 函数在真实 API 下的行为
|
||
"""
|
||
|
||
import os
|
||
import pytest
|
||
from psycho_screener.mcp_tool import psycho_screen
|
||
|
||
|
||
@pytest.fixture
|
||
def api_key():
|
||
key = os.environ.get("MINIMAX_API_KEY", "")
|
||
if not key:
|
||
pytest.skip("MINIMAX_API_KEY not set")
|
||
return key
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 真实 API 测试
|
||
# ---------------------------------------------------------------------------
|
||
|
||
class TestPsychoScreenTool:
|
||
"""测试 MCP 工具函数 psycho_screen"""
|
||
|
||
def test_bullying_conversation(self, api_key):
|
||
"""霸凌场景：多轮对话，孩子最新消息包含霸凌内容"""
|
||
messages = [
|
||
{"role": "system", "content": "你是一个友好的小智AI玩偶..."},
|
||
{"role": "user", "content": "今天幼儿园有个小朋友抢了我的玩具"},
|
||
{"role": "assistant", "content": "哎呀，那真是太过分了！"},
|
||
{"role": "user", "content": "他还不让我告诉老师，他说如果我告诉老师就会打我。我好害怕。"},
|
||
]
|
||
|
||
result = psycho_screen(messages, include_prefix=True)
|
||
|
||
print(f"\n[tool] bullying result: {result}")
|
||
assert result["detected"] is True
|
||
assert result["category"] == "bullying"
|
||
assert result["severity"] in ("medium", "high")
|
||
assert "已发现特定心理问题" in result["prefix"]
|
||
|
||
def test_normal_conversation(self, api_key):
|
||
"""正常对话：全程无异常"""
|
||
messages = [
|
||
{"role": "system", "content": "你是一个友好的小智AI玩偶..."},
|
||
{"role": "user", "content": "今天我画了一幅画，是一只大恐龙！"},
|
||
{"role": "assistant", "content": "哇，好厉害！"},
|
||
{"role": "user", "content": "是绿色的！晚上妈妈还做了红烧肉，好开心！"},
|
||
]
|
||
|
||
result = psycho_screen(messages, include_prefix=False)
|
||
|
||
print(f"\n[tool] normal result: {result}")
|
||
assert result["detected"] is False
|
||
assert result["category"] == "none"
|
||
assert result["severity"] == "none"
|
||
|
||
def test_no_prefix_when_not_detected(self, api_key):
|
||
"""未检测到问题时，prefix 应为空"""
|
||
messages = [
|
||
{"role": "user", "content": "今天天气真好呀！"},
|
||
]
|
||
|
||
result = psycho_screen(messages, include_prefix=True)
|
||
assert result["detected"] is False
|
||
assert result.get("prefix", "") == ""
|
||
|
||
def test_empty_messages(self, api_key):
|
||
"""空消息列表"""
|
||
result = psycho_screen([], include_prefix=True)
|
||
assert result["detected"] is False
|
||
assert result["summary"] == "无儿童对话内容可分析"
|
||
|
||
def test_messages_without_child_content(self, api_key):
|
||
"""只有 system 消息，无 user 消息"""
|
||
messages = [
|
||
{"role": "system", "content": "你是一个友好的小智AI玩偶..."},
|
||
]
|
||
|
||
result = psycho_screen(messages, include_prefix=True)
|
||
assert result["detected"] is False
|
||
|
||
def test_api_key_missing(self):
|
||
"""API key 未配置时的行为"""
|
||
# 临时清除 API key
|
||
original = os.environ.pop("MINIMAX_API_KEY", None)
|
||
|
||
result = psycho_screen(
|
||
[{"role": "user", "content": "测试"}],
|
||
include_prefix=True
|
||
)
|
||
|
||
# 恢复原值
|
||
if original:
|
||
os.environ["MINIMAX_API_KEY"] = original
|
||
|
||
assert result["detected"] is False
|
||
assert "error" in result
|