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重连相关测试
250 lines
8.9 KiB
Python
250 lines
8.9 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
|
|
class TestWeChatAccountPlugin:
|
|
def test_plugin_attributes(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
assert plugin.plugin_id == "wechat-account"
|
|
assert plugin.plugin_type == "account"
|
|
|
|
def test_init_with_config(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
config = {"corp_id": "test_corp", "corp_secret": "test_secret"}
|
|
plugin = WeChatAccountPlugin(config)
|
|
assert plugin._config == config
|
|
|
|
def test_init_with_no_config(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
assert plugin._config == {}
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
registry = AsyncMock()
|
|
await plugin.register(registry)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_unregister(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
registry = AsyncMock()
|
|
await plugin.unregister(registry)
|
|
|
|
def test_detect_legacy_state_migrations_wechat_token(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
old_config = {"wechat_token": "legacy_token_value"}
|
|
migrated = plugin.detect_legacy_state_migrations(old_config)
|
|
|
|
assert "corp_secret" in migrated
|
|
assert migrated["corp_secret"] == "legacy_token_value"
|
|
assert "wechat_token" not in migrated
|
|
|
|
def test_detect_legacy_state_migrations_no_legacy(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
old_config = {"corp_secret": "modern_value", "corp_id": "test"}
|
|
migrated = plugin.detect_legacy_state_migrations(old_config)
|
|
|
|
assert migrated["corp_secret"] == "modern_value"
|
|
assert "wechat_token" not in migrated
|
|
|
|
def test_detect_legacy_state_migrations_preserves_other_keys(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatAccountPlugin
|
|
|
|
plugin = WeChatAccountPlugin()
|
|
old_config = {"wechat_token": "old", "corp_id": "test_corp", "agent_id": "1000001"}
|
|
migrated = plugin.detect_legacy_state_migrations(old_config)
|
|
|
|
assert migrated["corp_id"] == "test_corp"
|
|
assert migrated["agent_id"] == "1000001"
|
|
assert migrated["corp_secret"] == "old"
|
|
|
|
|
|
class TestWeChatSetupPlugin:
|
|
def test_plugin_attributes(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
plugin = WeChatSetupPlugin()
|
|
assert plugin.plugin_id == "wechat-setup"
|
|
assert plugin.plugin_type == "setup"
|
|
|
|
def test_init_with_config(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
config = {"app_id": "wx_test", "app_secret": "test_secret"}
|
|
plugin = WeChatSetupPlugin(config)
|
|
assert plugin._config == config
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_register(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
plugin = WeChatSetupPlugin()
|
|
registry = AsyncMock()
|
|
await plugin.register(registry)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_unregister(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
plugin = WeChatSetupPlugin()
|
|
registry = AsyncMock()
|
|
await plugin.unregister(registry)
|
|
|
|
def test_detect_legacy_state_migrations_aes_key(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
plugin = WeChatSetupPlugin()
|
|
old_config = {"aes_key": "legacy_aes_key_value"}
|
|
migrated = plugin.detect_legacy_state_migrations(old_config)
|
|
|
|
assert "encoding_aes_key" in migrated
|
|
assert migrated["encoding_aes_key"] == "legacy_aes_key_value"
|
|
assert "aes_key" not in migrated
|
|
|
|
def test_detect_legacy_state_migrations_no_legacy(self):
|
|
from yuxi.channels.adapters.wechat.dual_plugin import WeChatSetupPlugin
|
|
|
|
plugin = WeChatSetupPlugin()
|
|
old_config = {"encoding_aes_key": "modern_value", "token": "test_token"}
|
|
migrated = plugin.detect_legacy_state_migrations(old_config)
|
|
|
|
assert migrated["encoding_aes_key"] == "modern_value"
|
|
assert "aes_key" not in migrated
|
|
|
|
|
|
class TestSetupContract:
|
|
def test_get_setup_contract(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
assert contract["channel_id"] == "wechat"
|
|
assert "setup_steps" in contract
|
|
assert "health_checks" in contract
|
|
|
|
def test_setup_steps_count(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
steps = contract["setup_steps"]
|
|
assert len(steps) == 5
|
|
|
|
def test_mode_selection_step(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
mode_step = contract["setup_steps"][0]
|
|
assert mode_step["id"] == "mode_selection"
|
|
assert len(mode_step["options"]) == 3
|
|
|
|
modes = {o["id"] for o in mode_step["options"]}
|
|
assert modes == {"wecom", "mp", "personal"}
|
|
|
|
def test_wecom_option_required_fields(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
wecom_option = contract["setup_steps"][0]["options"][0]
|
|
assert wecom_option["required_fields"] == ["corp_id", "corp_secret", "agent_id"]
|
|
|
|
def test_mp_option_required_fields(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
mp_option = contract["setup_steps"][0]["options"][1]
|
|
assert mp_option["required_fields"] == ["app_id", "app_secret"]
|
|
|
|
def test_personal_option_required_fields(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
personal_option = contract["setup_steps"][0]["options"][2]
|
|
assert personal_option["required_fields"] == ["bridge_url"]
|
|
|
|
def test_health_checks_count(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import get_setup_contract
|
|
|
|
contract = get_setup_contract()
|
|
checks = contract["health_checks"]
|
|
assert len(checks) == 3
|
|
|
|
check_ids = {c["id"] for c in checks}
|
|
assert "wecom_token_valid" in check_ids
|
|
assert "mp_token_valid" in check_ids
|
|
assert "bridge_healthy" in check_ids
|
|
|
|
def test_contract_returns_copy(self):
|
|
from yuxi.channels.adapters.wechat.setup_contract import (
|
|
WECHAT_SETUP_CONTRACT,
|
|
get_setup_contract,
|
|
)
|
|
|
|
contract1 = get_setup_contract()
|
|
contract2 = get_setup_contract()
|
|
assert contract1 == contract2
|
|
assert contract1 is not contract2
|
|
|
|
|
|
class TestDirectoryContract:
|
|
def test_get_directory_contract(self):
|
|
from yuxi.channels.adapters.wechat.directory_contract import get_directory_contract
|
|
|
|
contract = get_directory_contract()
|
|
assert contract["channel_id"] == "wechat"
|
|
assert "listing_methods" in contract
|
|
assert "identity_format" in contract
|
|
|
|
def test_listing_methods_peers(self):
|
|
from yuxi.channels.adapters.wechat.directory_contract import get_directory_contract
|
|
|
|
contract = get_directory_contract()
|
|
peers = contract["listing_methods"]["peers"]
|
|
|
|
assert "wecom" in peers
|
|
assert "mp" in peers
|
|
assert "personal" in peers
|
|
|
|
def test_listing_methods_groups(self):
|
|
from yuxi.channels.adapters.wechat.directory_contract import get_directory_contract
|
|
|
|
contract = get_directory_contract()
|
|
groups = contract["listing_methods"]["groups"]
|
|
|
|
assert "wecom" in groups
|
|
assert "mp" in groups
|
|
assert "personal" in groups
|
|
|
|
def test_identity_format(self):
|
|
from yuxi.channels.adapters.wechat.directory_contract import get_directory_contract
|
|
|
|
contract = get_directory_contract()
|
|
identity = contract["identity_format"]
|
|
|
|
assert identity["user_id"] == "wx:openid"
|
|
assert identity["group_id"] == "wx:chat_id"
|
|
|
|
def test_contract_returns_copy(self):
|
|
from yuxi.channels.adapters.wechat.directory_contract import (
|
|
WECHAT_DIRECTORY_CONTRACT,
|
|
get_directory_contract,
|
|
)
|
|
|
|
contract1 = get_directory_contract()
|
|
contract2 = get_directory_contract()
|
|
assert contract1 == contract2
|
|
assert contract1 is not contract2 |