fa81229f6f
* 滚木 * feat: 重构核心架构，增强类型安全与插件管理 本次提交对核心模块进行了深度重构，引入 Pydantic 增强配置管理的类型安全性，并全面优化了插件管理系统。 主要变更详情： 1. 核心架构与配置 - 重构配置加载模块：引入 Pydantic 模型 (`core/config_models.py`)，提供严格的配置项类型检查、验证及默认值管理。 - 统一模块结构：规范化模块导入路径，移除冗余的 `__init__.py` 文件，提升项目结构的清晰度。 - 性能优化：集成 Redis 缓存支持 (`RedisManager`)，有效降低高频 API 调用开销，提升响应速度。 2. 插件系统升级 - 实现热重载机制：新增插件文件变更监听功能，支持开发过程中自动重载插件，提升开发效率。 - 优化生命周期管理：改进插件加载与卸载逻辑，支持精确卸载指定插件及其关联的命令、事件处理器和定时任务。 3. 功能特性增强 - 新增媒体 API：引入 `MediaAPI` 模块，封装图片、语音等富媒体资源的获取与处理接口。 - 完善权限体系：重构权限管理系统，实现管理员与操作员的分级控制，支持更细粒度的命令权限校验。 4. 代码质量与稳定性 - 全面类型修复：解决 `mypy` 静态类型检查发现的大量类型错误（包括 `CommandManager`、`EventFactory` 及 `Bot` API 签名不匹配问题）。 - 增强错误处理：优化消息处理管道的异常捕获机制，完善关键路径的日志记录，提升系统运行稳定性。 * feat: 添加测试用例并优化代码结构 refactor(permission_manager): 调整初始化顺序和逻辑 fix(admin_manager): 修复初始化逻辑和目录创建问题 feat(ws): 优化Bot实例初始化条件 feat(message): 增强MessageSegment功能并添加测试 feat(events): 支持字符串格式的消息解析 test: 添加核心功能测试用例 refactor(plugin_manager): 改进插件路径处理 style: 清理无用导入和代码 chore: 更新依赖项
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import pytest
|
|
from models.message import MessageSegment
|
|
from models.objects import GroupInfo, StrangerInfo
|
|
|
|
class TestMessageSegment:
|
|
def test_text_segment(self):
|
|
seg = MessageSegment.text("Hello")
|
|
assert seg.type == "text"
|
|
assert seg.data["text"] == "Hello"
|
|
assert str(seg) == "Hello"
|
|
|
|
def test_at_segment(self):
|
|
seg = MessageSegment.at(123456)
|
|
assert seg.type == "at"
|
|
assert seg.data["qq"] == "123456"
|
|
assert str(seg) == "[CQ:at,qq=123456]"
|
|
|
|
def test_image_segment(self):
|
|
seg = MessageSegment.image("http://example.com/img.jpg", cache=False, proxy=False)
|
|
assert seg.type == "image"
|
|
assert seg.data["file"] == "http://example.com/img.jpg"
|
|
assert str(seg) == "[CQ:image,file=http://example.com/img.jpg,cache=0,proxy=0]"
|
|
|
|
def test_face_segment(self):
|
|
seg = MessageSegment.face(123)
|
|
assert seg.type == "face"
|
|
assert seg.data["id"] == "123"
|
|
assert str(seg) == "[CQ:face,id=123]"
|
|
|
|
def test_reply_segment(self):
|
|
seg = MessageSegment.reply(1001)
|
|
assert seg.type == "reply"
|
|
assert seg.data["id"] == "1001"
|
|
assert str(seg) == "[CQ:reply,id=1001]"
|
|
|
|
def test_add_segments(self):
|
|
seg1 = MessageSegment.text("Hello ")
|
|
seg2 = MessageSegment.at(123)
|
|
combined = seg1 + seg2
|
|
assert isinstance(combined, list)
|
|
assert len(combined) == 2
|
|
assert combined[0] == seg1
|
|
assert combined[1] == seg2
|
|
|
|
def test_add_segment_and_string(self):
|
|
seg = MessageSegment.at(123)
|
|
combined = seg + " Hello"
|
|
assert isinstance(combined, list)
|
|
assert len(combined) == 2
|
|
assert combined[0] == seg
|
|
assert combined[1].type == "text"
|
|
assert combined[1].data["text"] == " Hello"
|
|
|
|
class TestObjects:
|
|
def test_group_info(self):
|
|
data = {
|
|
"group_id": 123456,
|
|
"group_name": "Test Group",
|
|
"member_count": 10,
|
|
"max_member_count": 100
|
|
}
|
|
group = GroupInfo(**data)
|
|
assert group.group_id == 123456
|
|
assert group.group_name == "Test Group"
|
|
|
|
def test_stranger_info(self):
|
|
data = {
|
|
"user_id": 111111,
|
|
"nickname": "Stranger",
|
|
"sex": "male",
|
|
"age": 18
|
|
}
|
|
user = StrangerInfo(**data)
|
|
assert user.user_id == 111111
|
|
assert user.nickname == "Stranger"
|