Kris
69fe97a90d
1. 移除Telegram格式化测试中未使用的导入项 2. 修复Teams测试用例，添加monkeypatch参数并配置通配符开关 3. 更新钉钉适配器测试，替换弃用的流属性检查 4. 修正Twitch规范化测试，更新ROOMSTATE测试逻辑 5. 重构会话映射测试，完善数据库执行结果模拟 6. 格式化Slack块构建测试的长参数调用 7. 修复LINE适配器测试，更新能力断言和异步锁使用 8. 修正Slack会话解析测试，修复聊天类型判断错误 9. 更新能力测试，补充缺失的字段检查 10. 修复Matrix适配器测试，修正位置参数和配置校验逻辑 11. 为飞书分析模块测试添加跳过标记 12. 新增微信能力、限流、链接格式、会话路由等模块的单元测试 13. 修复Twitch适配器导入路径和测试断言 14. 新增Discord Webhook、Nextcloud Talk、Signal多账户等模块的单元测试 15. 修复Manager阶段测试的导入路径 16. 新增iMessage异常和命令处理的单元测试 17. 新增Nostr健康检查和相关模块的单元测试 18. 新增Signal守护进程和SSE重连相关测试
468 lines
15 KiB
Python
468 lines
15 KiB
Python
"""MSTeams config_resolver / route_config / tool_policy 单元测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from yuxi.channels.adapters.msteams.config_resolver import (
|
|
resolve_nested_config,
|
|
resolve_reply_style,
|
|
resolve_require_mention,
|
|
)
|
|
from yuxi.channels.adapters.msteams.route_config import resolve_route_config
|
|
from yuxi.channels.adapters.msteams.tool_policy import ToolPolicy, resolve_tool_policy
|
|
|
|
|
|
class TestResolveNestedConfig:
|
|
def test_empty_config(self):
|
|
result = resolve_nested_config({}, "team-1", "ch-1")
|
|
assert result == {}
|
|
|
|
def test_exact_team_match(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {"reply_style": "thread"},
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "team-1")
|
|
assert result["team_config"]["reply_style"] == "thread"
|
|
|
|
def test_wildcard_team_match(self):
|
|
config = {
|
|
"teams": {
|
|
"*": {"reply_style": "thread"},
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "unknown-team")
|
|
assert result["team_config"]["reply_style"] == "thread"
|
|
|
|
def test_no_team_match(self):
|
|
config = {"teams": {"team-1": {"reply_style": "thread"}}}
|
|
result = resolve_nested_config(config, "team-2")
|
|
assert result == {}
|
|
|
|
def test_channel_config_in_team(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {
|
|
"reply_style": "thread",
|
|
"channels": {"ch-1": {"require_mention": True}},
|
|
}
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "team-1", "ch-1")
|
|
assert result["team_config"]["reply_style"] == "thread"
|
|
assert result["channel_config"]["require_mention"] is True
|
|
|
|
def test_channel_wildcard(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {
|
|
"channels": {"*": {"require_mention": True}},
|
|
}
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "team-1", "ch-unknown")
|
|
assert result["channel_config"]["require_mention"] is True
|
|
|
|
def test_channel_not_found_no_wildcard(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {
|
|
"channels": {"ch-1": {"require_mention": True}},
|
|
}
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "team-1", "ch-2")
|
|
assert result.get("channel_config") is None
|
|
|
|
def test_no_channel_id(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {
|
|
"reply_style": "thread",
|
|
"channels": {"ch-1": {}},
|
|
}
|
|
}
|
|
}
|
|
result = resolve_nested_config(config, "team-1")
|
|
assert "channel_config" not in result
|
|
assert result["team_config"]["reply_style"] == "thread"
|
|
|
|
|
|
class TestResolveReplyStyle:
|
|
def test_channel_override(self):
|
|
config = {
|
|
"reply_style": "inline",
|
|
"teams": {
|
|
"team-1": {
|
|
"reply_style": "thread",
|
|
"channels": {"ch-1": {"reply_style": "new_thread"}},
|
|
}
|
|
},
|
|
}
|
|
result = resolve_reply_style(config, "team-1", "ch-1")
|
|
assert result == "new_thread"
|
|
|
|
def test_team_override(self):
|
|
config = {
|
|
"reply_style": "inline",
|
|
"teams": {"team-1": {"reply_style": "thread"}},
|
|
}
|
|
result = resolve_reply_style(config, "team-1", "ch-1")
|
|
assert result == "thread"
|
|
|
|
def test_global_default(self):
|
|
config = {"reply_style": "inline"}
|
|
result = resolve_reply_style(config, "team-1")
|
|
assert result == "inline"
|
|
|
|
def test_fallback_default(self):
|
|
result = resolve_reply_style({}, "team-1")
|
|
assert result == "thread"
|
|
|
|
|
|
class TestResolveRequireMention:
|
|
def test_channel_override(self):
|
|
config = {
|
|
"teams": {
|
|
"team-1": {
|
|
"require_mention": False,
|
|
"channels": {"ch-1": {"require_mention": True}},
|
|
}
|
|
},
|
|
}
|
|
result = resolve_require_mention(config, "team-1", "ch-1")
|
|
assert result is True
|
|
|
|
def test_team_override(self):
|
|
config = {
|
|
"require_mention": False,
|
|
"teams": {"team-1": {"require_mention": True}},
|
|
}
|
|
result = resolve_require_mention(config, "team-1")
|
|
assert result is True
|
|
|
|
def test_global_fallback(self):
|
|
config = {"require_mention": True}
|
|
result = resolve_require_mention(config, "team-1")
|
|
assert result is True
|
|
|
|
def test_none_when_not_configured(self):
|
|
result = resolve_require_mention({}, "team-1")
|
|
assert result is None
|
|
|
|
|
|
class TestResolveRouteConfig:
|
|
def test_no_routes_returns_original(self):
|
|
config = {"app_id": "test"}
|
|
result = resolve_route_config(config)
|
|
assert result == config
|
|
|
|
def test_empty_routes(self):
|
|
config = {"app_id": "test", "routes": []}
|
|
result = resolve_route_config(config)
|
|
assert result == config
|
|
|
|
def test_exact_team_id_match(self):
|
|
config = {
|
|
"app_id": "global",
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"config": {"app_id": "team-override"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="team-1")
|
|
assert result["app_id"] == "team-override"
|
|
|
|
def test_team_name_match(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"name": "My Team"},
|
|
"config": {"streaming_mode": "off"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_name="My Team")
|
|
assert result["streaming_mode"] == "off"
|
|
|
|
def test_team_name_case_insensitive(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"name": "My Team"},
|
|
"config": {"streaming_mode": "off"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_name="my team")
|
|
assert result["streaming_mode"] == "off"
|
|
|
|
def test_team_slug_match(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"slug": "my-team"},
|
|
"config": {"reply_style": "thread"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_name="My Team")
|
|
assert result["reply_style"] == "thread"
|
|
|
|
def test_slug_no_match(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"slug": "other-team"},
|
|
"config": {"reply_style": "thread"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_name="My Team")
|
|
assert "reply_style" not in result
|
|
|
|
def test_team_and_channel_match(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"match_channel": {"id": "ch-1"},
|
|
"config": {"require_mention": False},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="team-1", channel_id="ch-1")
|
|
assert result["require_mention"] is False
|
|
|
|
def test_channel_mismatch(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"match_channel": {"id": "ch-1"},
|
|
"config": {"require_mention": False},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="team-1", channel_id="ch-2")
|
|
assert result == config
|
|
|
|
def test_wildcard_route(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "*"},
|
|
"config": {"reply_style": "thread"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="any-team")
|
|
assert result["reply_style"] == "thread"
|
|
|
|
def test_first_match_wins(self):
|
|
config = {
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"config": {"reply_style": "first"},
|
|
},
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"config": {"reply_style": "second"},
|
|
},
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="team-1")
|
|
assert result["reply_style"] == "first"
|
|
|
|
def test_merged_config_preserves_original(self):
|
|
config = {
|
|
"app_id": "test",
|
|
"routes": [
|
|
{
|
|
"match_team": {"id": "team-1"},
|
|
"config": {"reply_style": "thread"},
|
|
}
|
|
],
|
|
}
|
|
result = resolve_route_config(config, team_id="team-1")
|
|
assert result["app_id"] == "test"
|
|
assert result["reply_style"] == "thread"
|
|
|
|
|
|
class TestToolPolicyBasic:
|
|
def test_default_allows_all(self):
|
|
policy = ToolPolicy()
|
|
assert policy.is_tool_allowed("any_tool") is True
|
|
assert policy.is_tool_allowed("web_search") is True
|
|
|
|
def test_global_allow_restricts(self):
|
|
policy = ToolPolicy({"tools_allow": ["web_search", "code_interpreter"]})
|
|
assert policy.is_tool_allowed("web_search") is True
|
|
assert policy.is_tool_allowed("file_upload") is False
|
|
|
|
def test_global_deny_blocks(self):
|
|
policy = ToolPolicy({"tools_deny": ["web_search"]})
|
|
assert policy.is_tool_allowed("web_search") is False
|
|
assert policy.is_tool_allowed("code_interpreter") is True
|
|
|
|
def test_global_deny_overrides_allow(self):
|
|
policy = ToolPolicy({
|
|
"tools_allow": ["web_search"],
|
|
"tools_deny": ["web_search"],
|
|
})
|
|
assert policy.is_tool_allowed("web_search") is False
|
|
|
|
|
|
class TestToolPolicySender:
|
|
def test_sender_allow(self):
|
|
policy = ToolPolicy({
|
|
"tools_by_sender": [{"sender_id": "user-1", "allow": ["web_search"]}],
|
|
})
|
|
assert policy.is_tool_allowed("web_search", sender_id="user-1") is True
|
|
assert policy.is_tool_allowed("code_interpreter", sender_id="user-1") is False
|
|
|
|
def test_sender_deny(self):
|
|
policy = ToolPolicy({
|
|
"tools_by_sender": [{"sender_id": "user-1", "allow": ["web_search", "code"], "deny": ["web_search"]}],
|
|
})
|
|
assert policy.is_tool_allowed("web_search", sender_id="user-1") is False
|
|
assert policy.is_tool_allowed("code", sender_id="user-1") is True
|
|
|
|
def test_sender_not_matched_falls_back(self):
|
|
policy = ToolPolicy({
|
|
"tools_allow": ["web_search"],
|
|
"tools_by_sender": [{"sender_id": "user-1", "allow": ["code_interpreter"]}],
|
|
})
|
|
assert policy.is_tool_allowed("web_search", sender_id="user-2") is True
|
|
assert policy.is_tool_allowed("code_interpreter", sender_id="user-2") is False
|
|
|
|
|
|
class TestToolPolicyTeamChannel:
|
|
def test_team_allow(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {"tools": {"allow": ["web_search"]}},
|
|
},
|
|
})
|
|
assert policy.is_tool_allowed("web_search", team_id="team-1") is True
|
|
assert policy.is_tool_allowed("code_interpreter", team_id="team-1") is False
|
|
|
|
def test_team_deny(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {"tools": {"deny": ["web_search"]}},
|
|
},
|
|
})
|
|
assert policy.is_tool_allowed("web_search", team_id="team-1") is False
|
|
|
|
def test_channel_allow_override_team(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {
|
|
"tools": {"allow": ["code"]},
|
|
"channels": {
|
|
"ch-1": {"tools": {"allow": ["web_search"]}},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
assert policy.is_tool_allowed("web_search", team_id="team-1", channel_id="ch-1") is False
|
|
assert policy.is_tool_allowed("code", team_id="team-1", channel_id="ch-1") is False
|
|
|
|
|
|
class TestToolPolicyGetAllowed:
|
|
def test_global_allow_list(self):
|
|
policy = ToolPolicy({"tools_allow": ["tool_a", "tool_b"]})
|
|
allowed = policy.get_allowed_tools()
|
|
assert allowed == ["tool_a", "tool_b"]
|
|
|
|
def test_no_allow_list_returns_none(self):
|
|
policy = ToolPolicy()
|
|
assert policy.get_allowed_tools() is None
|
|
|
|
def test_sender_allow(self):
|
|
policy = ToolPolicy({
|
|
"tools_by_sender": [{"sender_id": "user-1", "allow": ["tool_a", "tool_b"]}],
|
|
})
|
|
allowed = policy.get_allowed_tools(sender_id="user-1")
|
|
assert allowed == ["tool_a", "tool_b"]
|
|
|
|
def test_channel_overrides_team(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {
|
|
"tools": {"allow": ["team_tool"]},
|
|
"channels": {
|
|
"ch-1": {"tools": {"allow": ["channel_tool"]}},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
allowed = policy.get_allowed_tools(team_id="team-1", channel_id="ch-1")
|
|
assert allowed == ["channel_tool"]
|
|
|
|
def test_team_fallback_when_no_channel(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {"tools": {"allow": ["team_tool"]}},
|
|
},
|
|
})
|
|
allowed = policy.get_allowed_tools(team_id="team-1")
|
|
assert allowed == ["team_tool"]
|
|
|
|
def test_deny_excludes_from_allow(self):
|
|
policy = ToolPolicy({
|
|
"teams": {
|
|
"team-1": {
|
|
"tools": {"allow": ["tool_a", "tool_b"], "deny": ["tool_b"]},
|
|
},
|
|
},
|
|
})
|
|
allowed = policy.get_allowed_tools(team_id="team-1")
|
|
assert "tool_b" not in allowed
|
|
assert "tool_a" in allowed
|
|
|
|
|
|
class TestToolPolicyGetDenied:
|
|
def test_empty_config_returns_empty(self):
|
|
policy = ToolPolicy()
|
|
denied = policy.get_denied_tools()
|
|
assert denied == set()
|
|
|
|
def test_global_deny(self):
|
|
policy = ToolPolicy({"tools_deny": ["tool_a", "tool_b"]})
|
|
denied = policy.get_denied_tools()
|
|
assert denied == {"tool_a", "tool_b"}
|
|
|
|
def test_combined_deny(self):
|
|
policy = ToolPolicy({
|
|
"tools_deny": ["global_deny"],
|
|
"teams": {
|
|
"team-1": {
|
|
"tools": {"deny": ["team_deny"]},
|
|
"channels": {
|
|
"ch-1": {"tools": {"deny": ["channel_deny"]}},
|
|
},
|
|
},
|
|
},
|
|
})
|
|
denied = policy.get_denied_tools(team_id="team-1", channel_id="ch-1")
|
|
assert "global_deny" in denied
|
|
assert "team_deny" in denied
|
|
assert "channel_deny" in denied
|
|
|
|
|
|
class TestResolveToolPolicy:
|
|
def test_allow(self):
|
|
assert resolve_tool_policy({"tools_allow": ["web_search"]}, "web_search") is True
|
|
|
|
def test_deny(self):
|
|
assert resolve_tool_policy({"tools_deny": ["web_search"]}, "web_search") is False
|
|
|
|
def test_allow_missing(self):
|
|
assert resolve_tool_policy({"tools_allow": ["web_search"]}, "code_interpreter") is False |