feat: 添加测试用例并优化代码结构
refactor(permission_manager): 调整初始化顺序和逻辑 fix(admin_manager): 修复初始化逻辑和目录创建问题 feat(ws): 优化Bot实例初始化条件 feat(message): 增强MessageSegment功能并添加测试 feat(events): 支持字符串格式的消息解析 test: 添加核心功能测试用例 refactor(plugin_manager): 改进插件路径处理 style: 清理无用导入和代码 chore: 更新依赖项
This commit is contained in:
194
tests/test_event_handler.py
Normal file
194
tests/test_event_handler.py
Normal file
@@ -0,0 +1,194 @@
|
||||
import pytest
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
from core.handlers.event_handler import MessageHandler, NoticeHandler, RequestHandler
|
||||
from models.events.message import GroupMessageEvent
|
||||
from models.events.notice import GroupIncreaseNoticeEvent
|
||||
from models.events.request import FriendRequestEvent
|
||||
|
||||
@pytest.fixture
|
||||
def mock_bot():
|
||||
bot = AsyncMock()
|
||||
return bot
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_message_handler_run_handler_injection(mock_bot):
|
||||
"""测试参数注入"""
|
||||
handler = MessageHandler(prefixes=("/",))
|
||||
|
||||
# 1. 测试注入 bot 和 event
|
||||
async def func1(bot, event):
|
||||
assert bot == mock_bot
|
||||
assert event.user_id == 123
|
||||
return True
|
||||
|
||||
event = MagicMock(spec=GroupMessageEvent)
|
||||
event.user_id = 123
|
||||
|
||||
result = await handler._run_handler(func1, mock_bot, event)
|
||||
assert result is True
|
||||
|
||||
# 2. 测试注入 args
|
||||
async def func2(args):
|
||||
assert args == ["arg1", "arg2"]
|
||||
return True
|
||||
|
||||
result = await handler._run_handler(func2, mock_bot, event, args=["arg1", "arg2"])
|
||||
assert result is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_message_handler_command_parsing(mock_bot):
|
||||
"""测试命令解析"""
|
||||
handler = MessageHandler(prefixes=("/",))
|
||||
|
||||
mock_func = AsyncMock()
|
||||
handler.commands["test"] = {
|
||||
"func": mock_func,
|
||||
"permission": None,
|
||||
"override_permission_check": False,
|
||||
"plugin_name": "test_plugin"
|
||||
}
|
||||
|
||||
event = MagicMock(spec=GroupMessageEvent)
|
||||
event.raw_message = "/test arg1 arg2"
|
||||
event.user_id = 123
|
||||
|
||||
# Mock permission manager
|
||||
with patch("core.managers.permission_manager.PermissionManager.check_permission", new_callable=AsyncMock) as mock_perm:
|
||||
mock_perm.return_value = True
|
||||
|
||||
await handler.handle(mock_bot, event)
|
||||
|
||||
mock_func.assert_called_once()
|
||||
# 验证 args 参数是否正确传递
|
||||
call_args = mock_func.call_args
|
||||
if "args" in call_args.kwargs:
|
||||
assert call_args.kwargs["args"] == ["arg1", "arg2"]
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_notice_handler(mock_bot):
|
||||
"""测试通知事件分发"""
|
||||
handler = NoticeHandler()
|
||||
|
||||
mock_func = AsyncMock()
|
||||
handler.handlers.append({
|
||||
"type": "group_increase",
|
||||
"func": mock_func,
|
||||
"plugin_name": "test_plugin"
|
||||
})
|
||||
|
||||
event = MagicMock(spec=GroupIncreaseNoticeEvent)
|
||||
event.notice_type = "group_increase"
|
||||
|
||||
await handler.handle(mock_bot, event)
|
||||
|
||||
mock_func.assert_called_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_sync_handler_execution(mock_bot):
|
||||
"""测试同步处理函数的执行"""
|
||||
handler = MessageHandler(prefixes=("/",))
|
||||
|
||||
def sync_func(event):
|
||||
return True
|
||||
|
||||
event = MagicMock(spec=GroupMessageEvent)
|
||||
|
||||
# 同步函数应该在线程池中运行
|
||||
result = await handler._run_handler(sync_func, mock_bot, event)
|
||||
assert result is True
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_message_handler_management(mock_bot):
|
||||
"""测试消息处理器的管理(注册、卸载、清空)"""
|
||||
handler = MessageHandler(prefixes=("/",))
|
||||
|
||||
# 测试 on_message 装饰器
|
||||
@handler.on_message()
|
||||
async def msg_handler(event):
|
||||
pass
|
||||
|
||||
assert len(handler.message_handlers) == 1
|
||||
|
||||
# 测试 command 装饰器
|
||||
@handler.command("cmd1", "cmd2")
|
||||
async def cmd_handler(event):
|
||||
pass
|
||||
|
||||
assert len(handler.commands) == 2
|
||||
assert "cmd1" in handler.commands
|
||||
assert "cmd2" in handler.commands
|
||||
|
||||
# 测试 unregister_by_plugin_name
|
||||
# 直接从已注册的处理器中获取 plugin_name
|
||||
if handler.message_handlers:
|
||||
plugin_name = handler.message_handlers[0]["plugin_name"]
|
||||
handler.unregister_by_plugin_name(plugin_name)
|
||||
|
||||
assert len(handler.message_handlers) == 0
|
||||
assert len(handler.commands) == 0
|
||||
|
||||
# 测试 clear
|
||||
handler.commands["cmd"] = {}
|
||||
handler.message_handlers.append({})
|
||||
handler.clear()
|
||||
assert len(handler.commands) == 0
|
||||
assert len(handler.message_handlers) == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_request_handler(mock_bot):
|
||||
"""测试请求事件处理器"""
|
||||
handler = RequestHandler()
|
||||
|
||||
mock_func = AsyncMock()
|
||||
|
||||
# 测试 register 装饰器
|
||||
@handler.register("friend")
|
||||
async def req_handler(event):
|
||||
await mock_func(event)
|
||||
|
||||
assert len(handler.handlers) == 1
|
||||
|
||||
event = MagicMock(spec=FriendRequestEvent)
|
||||
event.request_type = "friend"
|
||||
|
||||
await handler.handle(mock_bot, event)
|
||||
mock_func.assert_called_once()
|
||||
|
||||
# 测试 unregister 和 clear
|
||||
import inspect
|
||||
module = inspect.getmodule(req_handler)
|
||||
plugin_name = module.__name__
|
||||
|
||||
handler.unregister_by_plugin_name(plugin_name)
|
||||
assert len(handler.handlers) == 0
|
||||
|
||||
handler.handlers.append({})
|
||||
handler.clear()
|
||||
assert len(handler.handlers) == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_permission_denied(mock_bot):
|
||||
"""测试权限不足的情况"""
|
||||
handler = MessageHandler(prefixes=("/",))
|
||||
|
||||
mock_func = AsyncMock()
|
||||
handler.commands["admin_cmd"] = {
|
||||
"func": mock_func,
|
||||
"permission": "ADMIN", # 假设 Permission.ADMIN
|
||||
"override_permission_check": False,
|
||||
"plugin_name": "test_plugin"
|
||||
}
|
||||
|
||||
event = MagicMock(spec=GroupMessageEvent)
|
||||
event.raw_message = "/admin_cmd"
|
||||
event.user_id = 123
|
||||
|
||||
# Mock permission manager returning False
|
||||
with patch("core.managers.permission_manager.PermissionManager.check_permission", new_callable=AsyncMock) as mock_perm:
|
||||
mock_perm.return_value = False
|
||||
|
||||
await handler.handle(mock_bot, event)
|
||||
|
||||
mock_func.assert_not_called()
|
||||
# 应该发送拒绝消息
|
||||
mock_bot.send.assert_called_once()
|
||||
Reference in New Issue
Block a user