Kris
3264900bc9
新增了Twitch、Telegram、Discord、Slack、Mattermost、WeChat、Zalo等多渠道的单元测试用例，覆盖了令牌处理、速率限制、消息去重、会话解析、格式转换、安全策略等模块 同时在测试配置中添加了测试用的OpenAI API密钥环境变量
135 lines
5.2 KiB
Python
135 lines
5.2 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from yuxi.channels.models import DeliveryResult
|
|
|
|
|
|
class TestWeChatGroupAdmin:
|
|
@pytest.fixture
|
|
def admin(self):
|
|
from yuxi.channels.adapters.wechat.group_admin import WeChatGroupAdmin
|
|
|
|
return WeChatGroupAdmin()
|
|
|
|
@pytest.fixture
|
|
def mock_client(self):
|
|
client = MagicMock()
|
|
client.get_access_token = AsyncMock(return_value="test_token")
|
|
return client
|
|
|
|
@pytest.fixture
|
|
def mock_http(self):
|
|
return AsyncMock()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_channel_info_success(self, admin, mock_client, mock_http):
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {
|
|
"errcode": 0,
|
|
"chat_info": {"chatid": "chat123", "name": "Test Group", "userlist": ["user1", "user2"]},
|
|
}
|
|
mock_http.get = AsyncMock(return_value=mock_resp)
|
|
|
|
info = await admin.get_channel_info(mock_client, mock_http, "chat123")
|
|
assert info["chatid"] == "chat123"
|
|
assert info["name"] == "Test Group"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_channel_info_error(self, admin, mock_client, mock_http):
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"errcode": 60011, "errmsg": "Invalid userid list"}
|
|
mock_http.get = AsyncMock(return_value=mock_resp)
|
|
|
|
info = await admin.get_channel_info(mock_client, mock_http, "chat_nonexist")
|
|
assert info == {}
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_rename_group_success(self, admin, mock_client, mock_http):
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"errcode": 0, "errmsg": "ok"}
|
|
mock_http.post = AsyncMock(return_value=mock_resp)
|
|
|
|
result = await admin.rename_group(mock_client, mock_http, "chat123", "New Name")
|
|
assert result.success is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_rename_group_error(self, admin, mock_client, mock_http):
|
|
mock_resp = MagicMock()
|
|
mock_resp.json.return_value = {"errcode": 60011, "errmsg": "Invalid chatid"}
|
|
mock_http.post = AsyncMock(return_value=mock_resp)
|
|
|
|
result = await admin.rename_group(mock_client, mock_http, "chat123", "New Name")
|
|
assert result.success is False
|
|
assert "Invalid chatid" in result.error
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_participant_success(self, admin, mock_client, mock_http):
|
|
get_resp = MagicMock()
|
|
get_resp.json.return_value = {
|
|
"errcode": 0,
|
|
"chat_info": {"chatid": "chat123", "userlist": ["user1"]},
|
|
}
|
|
mock_http.get = AsyncMock(return_value=get_resp)
|
|
|
|
post_resp = MagicMock()
|
|
post_resp.json.return_value = {"errcode": 0, "errmsg": "ok"}
|
|
mock_http.post = AsyncMock(return_value=post_resp)
|
|
|
|
result = await admin.add_participant(mock_client, mock_http, "chat123", "user2")
|
|
assert result.success is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_remove_participant_success(self, admin, mock_client, mock_http):
|
|
post_resp = MagicMock()
|
|
post_resp.json.return_value = {"errcode": 0, "errmsg": "ok"}
|
|
mock_http.post = AsyncMock(return_value=post_resp)
|
|
|
|
result = await admin.remove_participant(mock_client, mock_http, "chat123", "user2")
|
|
assert result.success is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_group_icon_unsupported(self, admin, mock_client, mock_http):
|
|
result = await admin.set_group_icon(mock_client, mock_http, "chat123", b"icon_data")
|
|
assert result.success is False
|
|
assert "not support" in result.error.lower()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_leave_group_unsupported(self, admin, mock_client, mock_http):
|
|
result = await admin.leave_group(mock_client, mock_http, "chat123")
|
|
assert result.success is False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_members_success(self, admin, mock_client, mock_http):
|
|
get_resp = MagicMock()
|
|
get_resp.json.return_value = {
|
|
"errcode": 0,
|
|
"chat_info": {"chatid": "chat123", "userlist": ["user1", "user2", "user3"]},
|
|
}
|
|
mock_http.get = AsyncMock(return_value=get_resp)
|
|
|
|
members = await admin.list_members(mock_client, mock_http, "chat123")
|
|
assert len(members) == 3
|
|
assert members[0] == {"user_id": "user1"}
|
|
|
|
|
|
class TestGroupAdminCapabilities:
|
|
def test_group_admin_actions_are_registered(self):
|
|
from yuxi.channels.adapters.wechat.message_actions import WECHAT_MESSAGE_ACTIONS, WeChatMessageActionAdapter
|
|
|
|
adapter = WeChatMessageActionAdapter()
|
|
all_actions = adapter.list_all_actions()
|
|
|
|
assert "renameGroup" in all_actions
|
|
assert all_actions["renameGroup"]["status"] == "implemented"
|
|
assert "addParticipant" in all_actions
|
|
assert all_actions["addParticipant"]["status"] == "implemented"
|
|
assert "removeParticipant" in all_actions
|
|
assert all_actions["removeParticipant"]["status"] == "implemented"
|
|
|
|
supported = adapter.list_supported_actions()
|
|
assert "renameGroup" in supported
|
|
assert "addParticipant" in supported
|
|
assert "removeParticipant" in supported |