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重连相关测试
504 lines
19 KiB
Python
504 lines
19 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from yuxi.channels.models import DeliveryResult
|
|
|
|
|
|
class TestMessageActionAdapterWithAdapter:
|
|
def test_get_handler_returns_none_without_adapter(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
adapter = WeChatMessageActionAdapter()
|
|
assert adapter.get_handler("send") is None
|
|
assert adapter.get_handler("reply") is None
|
|
assert adapter.get_handler("react") is None
|
|
|
|
def test_get_handler_returns_callable_with_adapter(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
|
|
handler = action_adapter.get_handler("send")
|
|
assert handler is not None
|
|
assert callable(handler)
|
|
|
|
handler = action_adapter.get_handler("reply")
|
|
assert handler is not None
|
|
assert callable(handler)
|
|
|
|
handler = action_adapter.get_handler("sendAttachment")
|
|
assert handler is not None
|
|
assert callable(handler)
|
|
|
|
def test_get_handler_returns_none_for_unsupported_actions(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
|
|
assert action_adapter.get_handler("react") is None
|
|
assert action_adapter.get_handler("edit") is None
|
|
assert action_adapter.get_handler("unsend") is None
|
|
assert action_adapter.get_handler("delete") is None
|
|
|
|
def test_is_supported_works_without_adapter(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
assert WeChatMessageActionAdapter.is_supported("send") is True
|
|
assert WeChatMessageActionAdapter.is_supported("reply") is True
|
|
assert WeChatMessageActionAdapter.is_supported("react") is False
|
|
assert WeChatMessageActionAdapter.is_supported("edit") is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_handler_calls_adapter_send(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
mock_wechat.send = AsyncMock(return_value=DeliveryResult(success=True))
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("send")
|
|
|
|
result = await handler(chat_id="test_chat", content="Hello World", channel_user_id="user_123")
|
|
|
|
assert result["success"] is True
|
|
mock_wechat.send.assert_called_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_handler_fails_without_chat_id(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("send")
|
|
|
|
result = await handler(content="Hello World")
|
|
assert result["success"] is False
|
|
assert "missing chat_id" in result["error"]
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reply_handler_calls_adapter_send_with_reply_metadata(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
mock_wechat.send = AsyncMock(return_value=DeliveryResult(success=True))
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("reply")
|
|
|
|
result = await handler(
|
|
chat_id="test_chat",
|
|
content="Reply message",
|
|
channel_user_id="user_123",
|
|
reply_to_msg_id="msg_456",
|
|
reply_to_channel_user_id="sender_789",
|
|
chat_type="direct",
|
|
)
|
|
|
|
assert result["success"] is True
|
|
mock_wechat.send.assert_called_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_download_file_handler_calls_download_media(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_data = b"fake_media_data"
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
mock_wechat.download_media = AsyncMock(return_value=mock_data)
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("download-file")
|
|
|
|
result = await handler(file_id="media_123")
|
|
assert result["success"] is True
|
|
assert result["data"] == mock_data
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_download_file_handler_fails_without_file_id(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("download-file")
|
|
|
|
result = await handler()
|
|
assert result["success"] is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_handler_calls_read_message(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
mock_wechat = MagicMock()
|
|
mock_wechat.channel_id = "wechat"
|
|
mock_wechat.channel_type = "wechat"
|
|
mock_wechat.read_message = AsyncMock(return_value={"msg_id": "123", "available": True})
|
|
|
|
action_adapter = WeChatMessageActionAdapter(mock_wechat)
|
|
handler = action_adapter.get_handler("read")
|
|
|
|
result = await handler(msg_id="123")
|
|
assert result["available"] is True
|
|
|
|
def test_list_supported_actions_static(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
supported = WeChatMessageActionAdapter.list_supported_actions()
|
|
assert "send" in supported
|
|
assert "reply" in supported
|
|
assert "read" in supported
|
|
assert "react" not in supported
|
|
assert "edit" not in supported
|
|
|
|
def test_list_all_actions_static(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WeChatMessageActionAdapter
|
|
|
|
all_actions = WeChatMessageActionAdapter.list_all_actions()
|
|
assert len(all_actions) == 56
|
|
assert "send" in all_actions
|
|
assert "react" in all_actions
|
|
|
|
|
|
class TestWeChatHandleAction:
|
|
@pytest.mark.asyncio
|
|
async def test_handle_action_delegates_to_get_handler(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._mode = "wecom"
|
|
|
|
mock_client = MagicMock()
|
|
mock_client.get_access_token = AsyncMock(return_value="test_token")
|
|
adapter._wecom_client = mock_client
|
|
adapter._http_client = AsyncMock()
|
|
|
|
from yuxi.channels.adapters.wechat.wecom import WeComMonitor
|
|
|
|
class FakeMonitor:
|
|
async def start(self, cb):
|
|
pass
|
|
|
|
async def stop(self):
|
|
pass
|
|
|
|
adapter._monitor = FakeMonitor()
|
|
|
|
adapter._message_actions = MagicMock()
|
|
mock_handler = AsyncMock(return_value={"success": True})
|
|
adapter._message_actions.get_handler.return_value = mock_handler
|
|
adapter._message_actions.is_supported.return_value = True
|
|
|
|
from yuxi.channels.protocols.actions import ActionContext
|
|
|
|
ctx = ActionContext(
|
|
action="send",
|
|
channel_id="wechat",
|
|
chat_id="test_chat",
|
|
msg_id="msg_001",
|
|
args={"content": "Hello", "channel_user_id": "user_1"},
|
|
)
|
|
result = await adapter.handle_action(ctx)
|
|
|
|
assert result.success is True
|
|
adapter._message_actions.get_handler.assert_called_once_with("send")
|
|
mock_handler.assert_called_once()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_action_returns_error_for_unsupported_action(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._message_actions = MagicMock()
|
|
adapter._message_actions.get_handler.return_value = None
|
|
adapter._message_actions.is_supported.return_value = False
|
|
|
|
from yuxi.channels.protocols.actions import ActionContext
|
|
|
|
ctx = ActionContext(
|
|
action="react",
|
|
channel_id="wechat",
|
|
chat_id="test_chat",
|
|
msg_id="msg_001",
|
|
args={"emoji": "👍"},
|
|
)
|
|
result = await adapter.handle_action(ctx)
|
|
|
|
assert result.success is False
|
|
assert "not supported" in result.error
|
|
|
|
def test_supports_action(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._message_actions = MagicMock()
|
|
adapter._message_actions.is_supported.side_effect = lambda action: action == "send"
|
|
|
|
assert adapter.supports_action("send") is True
|
|
assert adapter.supports_action("react") is False
|
|
|
|
def test_resolve_execution_mode(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
assert adapter.resolve_execution_mode("send") == "direct"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_handle_action_handler_exception(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._message_actions = MagicMock()
|
|
async def failing_handler(**kwargs):
|
|
raise RuntimeError("boom")
|
|
adapter._message_actions.get_handler.return_value = failing_handler
|
|
adapter._message_actions.is_supported.return_value = True
|
|
|
|
from yuxi.channels.protocols.actions import ActionContext
|
|
|
|
ctx = ActionContext(
|
|
action="send",
|
|
channel_id="wechat",
|
|
chat_id="test_chat",
|
|
msg_id="msg_001",
|
|
args={},
|
|
)
|
|
result = await adapter.handle_action(ctx)
|
|
|
|
assert result.success is False
|
|
assert "boom" in result.error
|
|
|
|
|
|
class TestWeChatAuthAdapter:
|
|
def test_get_dm_exposure(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
assert auth.get_dm_exposure({}) == "pairing"
|
|
assert auth.get_dm_exposure({"dm_policy": "allowlist"}) == "allowlist"
|
|
assert auth.get_dm_exposure({"dm_policy": "open"}) == "open"
|
|
|
|
def test_get_authorization_url_with_config(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
assert auth.get_authorization_url({}) is None
|
|
assert auth.get_authorization_url({"wecom_auth_url": "https://example.com/auth"}) == "https://example.com/auth"
|
|
|
|
def test_build_wecom_oauth_url(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
url = auth.build_wecom_oauth_url("corp_123", "https://myapp.com/callback", state="state_456")
|
|
|
|
assert "open.weixin.qq.com" in url
|
|
assert "appid=corp_123" in url
|
|
assert "redirect_uri=https://myapp.com/callback" in url
|
|
assert "state=state_456" in url
|
|
assert url.endswith("#wechat_redirect")
|
|
|
|
def test_build_wecom_oauth_url_without_state(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
url = auth.build_wecom_oauth_url("corp_123", "https://myapp.com/callback")
|
|
|
|
assert "state" not in url or url.endswith("#wechat_redirect")
|
|
|
|
def test_build_mp_oauth_url(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
url = auth.build_mp_oauth_url("wx_app_123", "https://myapp.com/callback", state="state_789", scope="snsapi_userinfo")
|
|
|
|
assert "open.weixin.qq.com" in url
|
|
assert "appid=wx_app_123" in url
|
|
assert "scope=snsapi_userinfo" in url
|
|
assert "state=state_789" in url
|
|
|
|
def test_build_mp_oauth_url_default_scope(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
url = auth.build_mp_oauth_url("wx_app_123", "https://myapp.com/callback")
|
|
|
|
assert "scope=snsapi_userinfo" in url
|
|
|
|
def test_get_oauth_enabled(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
assert auth.get_oauth_enabled({}) is False
|
|
assert auth.get_oauth_enabled({"wecom_oauth_enabled": True}) is True
|
|
assert auth.get_oauth_enabled({"mp_oauth_enabled": True}) is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_exchange_wecom_code_success(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
mock_client = AsyncMock()
|
|
mock_client_cls.return_value.__aenter__.return_value = mock_client
|
|
|
|
token_resp = MagicMock()
|
|
token_resp.json.return_value = {"access_token": "token_abc", "expires_in": 7200}
|
|
user_resp = MagicMock()
|
|
user_resp.json.return_value = {"errcode": 0, "UserId": "user_123"}
|
|
|
|
mock_client.get.side_effect = [token_resp, user_resp]
|
|
|
|
result = await auth.exchange_wecom_code("auth_code", "corp_123", "corp_secret_456")
|
|
|
|
assert result["success"] is True
|
|
assert result["access_token"] == "token_abc"
|
|
assert result["user_info"]["UserId"] == "user_123"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_exchange_wecom_code_token_failure(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
mock_client = AsyncMock()
|
|
mock_client_cls.return_value.__aenter__.return_value = mock_client
|
|
|
|
token_resp = MagicMock()
|
|
token_resp.json.return_value = {"errcode": 40001, "errmsg": "invalid credential"}
|
|
|
|
mock_client.get.return_value = token_resp
|
|
|
|
result = await auth.exchange_wecom_code("auth_code", "corp_123", "corp_secret_456")
|
|
|
|
assert result["success"] is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_exchange_mp_code_success(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
mock_client = AsyncMock()
|
|
mock_client_cls.return_value.__aenter__.return_value = mock_client
|
|
|
|
token_resp = MagicMock()
|
|
token_resp.json.return_value = {
|
|
"access_token": "mp_token_abc",
|
|
"openid": "openid_123",
|
|
"refresh_token": "refresh_xyz",
|
|
}
|
|
userinfo_resp = MagicMock()
|
|
userinfo_resp.json.return_value = {"openid": "openid_123", "nickname": "TestUser"}
|
|
|
|
mock_client.get.side_effect = [token_resp, userinfo_resp]
|
|
|
|
result = await auth.exchange_mp_code("auth_code", "wx_app_123", "app_secret_456")
|
|
|
|
assert result["success"] is True
|
|
assert result["access_token"] == "mp_token_abc"
|
|
assert result["openid"] == "openid_123"
|
|
assert result["refresh_token"] == "refresh_xyz"
|
|
assert result["user_info"]["nickname"] == "TestUser"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_exchange_mp_code_failure(self):
|
|
from yuxi.channels.adapters.wechat.auth_adapter import WeChatAuthAdapter
|
|
|
|
auth = WeChatAuthAdapter()
|
|
|
|
with patch("httpx.AsyncClient") as mock_client_cls:
|
|
mock_client = AsyncMock()
|
|
mock_client_cls.return_value.__aenter__.return_value = mock_client
|
|
|
|
token_resp = MagicMock()
|
|
token_resp.json.return_value = {"errcode": 40029, "errmsg": "invalid code"}
|
|
|
|
mock_client.get.return_value = token_resp
|
|
|
|
result = await auth.exchange_mp_code("bad_code", "wx_app_123", "app_secret_456")
|
|
|
|
assert result["success"] is False
|
|
|
|
|
|
class TestWeChatSubAccount:
|
|
def test_sub_account_tracking_fields_in_init(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
|
|
assert isinstance(adapter._sub_monitors, dict)
|
|
assert len(adapter._sub_monitors) == 0
|
|
assert isinstance(adapter._sub_clients, dict)
|
|
assert len(adapter._sub_clients) == 0
|
|
assert isinstance(adapter._sub_polling_tasks, dict)
|
|
assert len(adapter._sub_polling_tasks) == 0
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_connect_sub_accounts_skips_when_only_default(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._account_mgr = MagicMock()
|
|
adapter._account_mgr.list_account_ids.return_value = ["default"]
|
|
|
|
await adapter._connect_sub_accounts()
|
|
|
|
assert len(adapter._sub_monitors) == 0
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_connect_sub_accounts_skips_disabled(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={})
|
|
adapter._account_mgr = MagicMock()
|
|
adapter._account_mgr.list_account_ids.return_value = ["default", "sub1"]
|
|
|
|
mock_account = MagicMock()
|
|
mock_account.enabled = False
|
|
mock_account.mode = "wecom"
|
|
adapter._account_mgr.resolve_account.return_value = mock_account
|
|
|
|
await adapter._connect_sub_accounts()
|
|
|
|
assert len(adapter._sub_monitors) == 0
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_connect_sub_accounts_respects_limit(self):
|
|
from yuxi.channels.adapters.wechat.adapter import WeChatAdapter
|
|
|
|
adapter = WeChatAdapter(config={"max_sub_accounts": 2})
|
|
adapter._account_mgr = MagicMock()
|
|
adapter._account_mgr.list_account_ids.return_value = [
|
|
"default", "sub1", "sub2", "sub3", "sub4"
|
|
]
|
|
|
|
mock_account = MagicMock()
|
|
mock_account.enabled = True
|
|
mock_account.mode = "mp"
|
|
adapter._account_mgr.resolve_account.return_value = mock_account
|
|
|
|
await adapter._connect_sub_accounts()
|
|
|
|
adapter._account_mgr.resolve_account.assert_called()
|
|
call_count = adapter._account_mgr.resolve_account.call_count
|
|
assert call_count <= 2 |