feat: 添加测试覆盖率并修复相关问题
refactor(redis_manager): 移除冗余的ConnectionError处理 refactor(event_handler): 优化Bot类型注解 refactor(factory): 移除未使用的GroupCardNoticeEvent test: 添加全面的单元测试覆盖 - 添加test_import.py测试模块导入 - 添加test_debug.py测试插件加载调试 - 添加test_plugin_error.py测试错误处理 - 添加test_config_loader.py测试配置加载 - 添加test_redis_manager.py测试Redis管理 - 添加test_bot.py测试Bot功能 - 扩展test_models.py测试消息模型 - 添加test_plugin_manager_coverage.py测试插件管理 - 添加test_executor.py测试代码执行器 - 添加test_ws.py测试WebSocket - 添加test_api.py测试API接口 - 添加test_core_managers.py测试核心管理模块 fix(plugin_manager): 修复插件加载日志变量问题 覆盖率已到达86%（忽略插件）
This commit is contained in:
@@ -1,141 +1,430 @@
|
||||
import pytest
|
||||
from models.events.factory import EventFactory, EventType
|
||||
from models.events.factory import EventFactory
|
||||
from models.events.base import EventType
|
||||
from models.events.message import GroupMessageEvent, PrivateMessageEvent
|
||||
from models.events.notice import GroupIncreaseNoticeEvent
|
||||
from models.events.request import FriendRequestEvent
|
||||
from models.events.meta import HeartbeatEvent
|
||||
from models.message import MessageSegment
|
||||
from models.events.notice import (
|
||||
FriendAddNoticeEvent, FriendRecallNoticeEvent, GroupRecallNoticeEvent,
|
||||
GroupIncreaseNoticeEvent, GroupDecreaseNoticeEvent, GroupAdminNoticeEvent,
|
||||
GroupBanNoticeEvent, GroupUploadNoticeEvent, PokeNotifyEvent,
|
||||
LuckyKingNotifyEvent, HonorNotifyEvent, GroupCardNoticeEvent,
|
||||
OfflineFileNoticeEvent, ClientStatusNoticeEvent, EssenceNoticeEvent,
|
||||
NotifyNoticeEvent
|
||||
)
|
||||
from models.events.request import FriendRequestEvent, GroupRequestEvent
|
||||
from models.events.meta import HeartbeatEvent, LifeCycleEvent
|
||||
|
||||
|
||||
class TestEventFactory:
|
||||
def test_create_group_message_event_list(self):
|
||||
"""测试创建群消息事件 (message 为列表格式)"""
|
||||
def test_create_private_message_event(self):
|
||||
"""测试创建私聊消息事件。"""
|
||||
data = {
|
||||
"post_type": "message",
|
||||
"message_type": "group",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456,
|
||||
"sub_type": "normal",
|
||||
"message_id": 1001,
|
||||
"user_id": 111111,
|
||||
"group_id": 222222,
|
||||
"message": [
|
||||
{"type": "text", "data": {"text": "Hello"}}
|
||||
],
|
||||
"post_type": EventType.MESSAGE,
|
||||
"message_type": "private",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"message_id": 123,
|
||||
"user_id": 20000,
|
||||
"message": [{"type": "text", "data": {"text": "Hello"}}],
|
||||
"raw_message": "Hello",
|
||||
"font": 0,
|
||||
"sender": {
|
||||
"user_id": 111111,
|
||||
"nickname": "User",
|
||||
"role": "member"
|
||||
}
|
||||
"font": 12,
|
||||
"sender": {"user_id": 20000, "nickname": "TestUser"}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupMessageEvent)
|
||||
assert event.group_id == 222222
|
||||
assert isinstance(event, PrivateMessageEvent)
|
||||
assert event.message_type == "private"
|
||||
assert event.user_id == 20000
|
||||
assert len(event.message) == 1
|
||||
assert event.message[0].type == "text"
|
||||
assert event.message[0].data["text"] == "Hello"
|
||||
|
||||
def test_create_group_message_event_str(self):
|
||||
"""测试创建群消息事件 (message 为字符串格式)"""
|
||||
def test_create_group_message_event(self):
|
||||
"""测试创建群消息事件。"""
|
||||
data = {
|
||||
"post_type": "message",
|
||||
"post_type": EventType.MESSAGE,
|
||||
"message_type": "group",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456,
|
||||
"sub_type": "normal",
|
||||
"message_id": 1002,
|
||||
"user_id": 111111,
|
||||
"group_id": 222222,
|
||||
"message": "Hello World",
|
||||
"raw_message": "Hello World",
|
||||
"font": 0,
|
||||
"sender": {
|
||||
"user_id": 111111,
|
||||
"nickname": "User"
|
||||
}
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"message_id": 123,
|
||||
"user_id": 20000,
|
||||
"group_id": 30000,
|
||||
"message": [{"type": "text", "data": {"text": "Hello"}}],
|
||||
"raw_message": "Hello",
|
||||
"font": 12,
|
||||
"sender": {"user_id": 20000, "nickname": "TestUser", "role": "member"}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupMessageEvent)
|
||||
assert len(event.message) == 1
|
||||
assert event.message[0].type == "text"
|
||||
assert event.message[0].data["text"] == "Hello World"
|
||||
assert event.message_type == "group"
|
||||
assert event.group_id == 30000
|
||||
assert event.user_id == 20000
|
||||
|
||||
def test_create_private_message_event(self):
|
||||
"""测试创建私聊消息事件"""
|
||||
def test_create_group_message_with_anonymous(self):
|
||||
"""测试创建匿名群消息事件。"""
|
||||
data = {
|
||||
"post_type": "message",
|
||||
"message_type": "private",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456,
|
||||
"sub_type": "friend",
|
||||
"message_id": 2001,
|
||||
"user_id": 333333,
|
||||
"message": "Private Msg",
|
||||
"raw_message": "Private Msg",
|
||||
"font": 0,
|
||||
"sender": {
|
||||
"user_id": 333333,
|
||||
"nickname": "Friend"
|
||||
}
|
||||
"post_type": EventType.MESSAGE,
|
||||
"message_type": "group",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"message_id": 123,
|
||||
"user_id": 20000,
|
||||
"group_id": 30000,
|
||||
"anonymous": {"id": 12345, "name": "Anonymous", "flag": "flag123"},
|
||||
"message": [{"type": "text", "data": {"text": "Hello"}}],
|
||||
"raw_message": "Hello",
|
||||
"font": 12,
|
||||
"sender": {"user_id": 20000, "nickname": "TestUser", "role": "member"}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, PrivateMessageEvent)
|
||||
assert event.user_id == 333333
|
||||
assert isinstance(event, GroupMessageEvent)
|
||||
assert event.anonymous is not None
|
||||
assert event.anonymous.id == 12345
|
||||
assert event.anonymous.name == "Anonymous"
|
||||
assert event.anonymous.flag == "flag123"
|
||||
|
||||
def test_create_notice_event(self):
|
||||
"""测试创建通知事件 (群成员增加)"""
|
||||
def test_create_friend_add_notice(self):
|
||||
"""测试创建好友添加通知事件。"""
|
||||
data = {
|
||||
"post_type": "notice",
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "friend_add",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"user_id": 20000
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, FriendAddNoticeEvent)
|
||||
assert event.notice_type == "friend_add"
|
||||
assert event.user_id == 20000
|
||||
|
||||
def test_create_friend_recall_notice(self):
|
||||
"""测试创建好友消息撤回通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "friend_recall",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"user_id": 20000,
|
||||
"message_id": 123
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, FriendRecallNoticeEvent)
|
||||
assert event.notice_type == "friend_recall"
|
||||
assert event.message_id == 123
|
||||
|
||||
def test_create_group_recall_notice(self):
|
||||
"""测试创建群消息撤回通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_recall",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"operator_id": 40000,
|
||||
"message_id": 123
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupRecallNoticeEvent)
|
||||
assert event.notice_type == "group_recall"
|
||||
assert event.group_id == 30000
|
||||
assert event.operator_id == 40000
|
||||
|
||||
def test_create_group_increase_notice(self):
|
||||
"""测试创建群成员增加通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_increase",
|
||||
"sub_type": "approve",
|
||||
"group_id": 222222,
|
||||
"operator_id": 444444,
|
||||
"user_id": 555555,
|
||||
"time": 1600000000,
|
||||
"self_id": 123456
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"operator_id": 40000,
|
||||
"sub_type": "approve"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupIncreaseNoticeEvent)
|
||||
assert event.group_id == 222222
|
||||
assert event.user_id == 555555
|
||||
assert event.notice_type == "group_increase"
|
||||
assert event.sub_type == "approve"
|
||||
|
||||
def test_create_request_event(self):
|
||||
"""测试创建请求事件 (加好友)"""
|
||||
def test_create_group_decrease_notice(self):
|
||||
"""测试创建群成员减少通知事件。"""
|
||||
data = {
|
||||
"post_type": "request",
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_decrease",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"operator_id": 40000,
|
||||
"sub_type": "kick"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupDecreaseNoticeEvent)
|
||||
assert event.notice_type == "group_decrease"
|
||||
assert event.sub_type == "kick"
|
||||
|
||||
def test_create_group_admin_notice(self):
|
||||
"""测试创建群管理员变更通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_admin",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"sub_type": "set"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupAdminNoticeEvent)
|
||||
assert event.notice_type == "group_admin"
|
||||
assert event.sub_type == "set"
|
||||
|
||||
def test_create_group_ban_notice(self):
|
||||
"""测试创建群成员禁言通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_ban",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"operator_id": 40000,
|
||||
"duration": 3600,
|
||||
"sub_type": "ban"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupBanNoticeEvent)
|
||||
assert event.notice_type == "group_ban"
|
||||
assert event.duration == 3600
|
||||
|
||||
def test_create_group_upload_notice(self):
|
||||
"""测试创建群文件上传通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_upload",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"file": {"id": "file123", "name": "test.txt", "size": 1024, "busid": 1}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupUploadNoticeEvent)
|
||||
assert event.notice_type == "group_upload"
|
||||
assert event.file.name == "test.txt"
|
||||
assert event.file.size == 1024
|
||||
|
||||
def test_create_poke_notify_event(self):
|
||||
"""测试创建戳一戳通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "notify",
|
||||
"sub_type": "poke",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"target_id": 40000
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, PokeNotifyEvent)
|
||||
assert event.notice_type == "notify"
|
||||
assert event.sub_type == "poke"
|
||||
|
||||
def test_create_lucky_king_notify_event(self):
|
||||
"""测试创建运气王通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "notify",
|
||||
"sub_type": "lucky_king",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"target_id": 40000
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, LuckyKingNotifyEvent)
|
||||
assert event.sub_type == "lucky_king"
|
||||
|
||||
def test_create_honor_notify_event(self):
|
||||
"""测试创建荣誉变更通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "notify",
|
||||
"sub_type": "honor",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"honor_type": "talkative"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, HonorNotifyEvent)
|
||||
assert event.sub_type == "honor"
|
||||
assert event.honor_type == "talkative"
|
||||
|
||||
def test_create_unknown_notify_event(self):
|
||||
"""测试创建未知类型的通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "notify",
|
||||
"sub_type": "unknown",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"user_id": 20000
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, NotifyNoticeEvent)
|
||||
assert event.notice_type == "notify"
|
||||
assert event.sub_type == "unknown"
|
||||
|
||||
def test_create_group_card_notice(self):
|
||||
"""测试创建群名片变更通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "group_card",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"card_new": "NewCard",
|
||||
"card_old": "OldCard"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupCardNoticeEvent)
|
||||
assert event.notice_type == "group_card"
|
||||
assert event.card_new == "NewCard"
|
||||
assert event.card_old == "OldCard"
|
||||
|
||||
def test_create_offline_file_notice(self):
|
||||
"""测试创建离线文件通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "offline_file",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"user_id": 20000,
|
||||
"file": {"name": "test.txt", "size": 1024, "url": "http://example.com/test.txt"}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, OfflineFileNoticeEvent)
|
||||
assert event.notice_type == "offline_file"
|
||||
assert event.file.name == "test.txt"
|
||||
|
||||
def test_create_client_status_notice(self):
|
||||
"""测试创建客户端状态通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "client_status",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"client": {"online": True, "status": "normal"}
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, ClientStatusNoticeEvent)
|
||||
assert event.notice_type == "client_status"
|
||||
assert event.client.online is True
|
||||
|
||||
def test_create_essence_notice(self):
|
||||
"""测试创建精华消息通知事件。"""
|
||||
data = {
|
||||
"post_type": EventType.NOTICE,
|
||||
"notice_type": "essence",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"sender_id": 20000,
|
||||
"operator_id": 40000,
|
||||
"message_id": 123,
|
||||
"sub_type": "add"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, EssenceNoticeEvent)
|
||||
assert event.notice_type == "essence"
|
||||
assert event.sub_type == "add"
|
||||
|
||||
def test_create_friend_request_event(self):
|
||||
"""测试创建好友请求事件。"""
|
||||
data = {
|
||||
"post_type": EventType.REQUEST,
|
||||
"request_type": "friend",
|
||||
"user_id": 666666,
|
||||
"comment": "Add me",
|
||||
"flag": "flag_123",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"user_id": 20000,
|
||||
"comment": "Hello",
|
||||
"flag": "flag123"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, FriendRequestEvent)
|
||||
assert event.user_id == 666666
|
||||
assert event.comment == "Add me"
|
||||
assert event.request_type == "friend"
|
||||
assert event.comment == "Hello"
|
||||
|
||||
def test_create_meta_event(self):
|
||||
"""测试创建元事件 (心跳)"""
|
||||
def test_create_group_request_event(self):
|
||||
"""测试创建群请求事件。"""
|
||||
data = {
|
||||
"post_type": "meta_event",
|
||||
"post_type": EventType.REQUEST,
|
||||
"request_type": "group",
|
||||
"sub_type": "add",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"group_id": 30000,
|
||||
"user_id": 20000,
|
||||
"comment": "Hello",
|
||||
"flag": "flag123"
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, GroupRequestEvent)
|
||||
assert event.request_type == "group"
|
||||
assert event.sub_type == "add"
|
||||
|
||||
def test_create_heartbeat_event(self):
|
||||
"""测试创建心跳元事件。"""
|
||||
data = {
|
||||
"post_type": EventType.META,
|
||||
"meta_event_type": "heartbeat",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456,
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"status": {"online": True, "good": True},
|
||||
"interval": 5000
|
||||
"interval": 1000
|
||||
}
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, HeartbeatEvent)
|
||||
assert event.interval == 5000
|
||||
assert event.meta_event_type == "heartbeat"
|
||||
assert event.status.online is True
|
||||
assert event.interval == 1000
|
||||
|
||||
def test_unknown_event_type(self):
|
||||
"""测试未知事件类型"""
|
||||
def test_create_lifecycle_event(self):
|
||||
"""测试创建生命周期元事件。"""
|
||||
data = {
|
||||
"post_type": "unknown_type",
|
||||
"time": 1600000000,
|
||||
"self_id": 123456
|
||||
"post_type": EventType.META,
|
||||
"meta_event_type": "lifecycle",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"sub_type": "enable"
|
||||
}
|
||||
with pytest.raises(ValueError, match="Unknown event type"):
|
||||
event = EventFactory.create_event(data)
|
||||
assert isinstance(event, LifeCycleEvent)
|
||||
assert event.meta_event_type == "lifecycle"
|
||||
assert event.sub_type == "enable"
|
||||
|
||||
def test_create_unknown_event_type(self):
|
||||
"""测试创建未知类型事件时抛出异常。"""
|
||||
data = {
|
||||
"post_type": "unknown",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000
|
||||
}
|
||||
with pytest.raises(ValueError, match="Unknown event type: unknown"):
|
||||
EventFactory.create_event(data)
|
||||
|
||||
def test_create_unknown_message_type(self):
|
||||
"""测试创建未知消息类型时抛出异常。"""
|
||||
data = {
|
||||
"post_type": EventType.MESSAGE,
|
||||
"message_type": "unknown",
|
||||
"time": 1234567890,
|
||||
"self_id": 10000,
|
||||
"message": "Hello"
|
||||
}
|
||||
with pytest.raises(ValueError, match="Unknown message type: unknown"):
|
||||
EventFactory.create_event(data)
|
||||
|
||||
Reference in New Issue
Block a user