K2cr2O1
caf5b06097
feat(ws_pool): 新增 WebSocket 连接池实现 perf(json): 使用 orjson 替代标准 json 库提升性能 style: 清理未使用的导入和冗余代码 docs: 更新架构文档和开发规范 test: 添加 WebSocket 连接池测试用例 fix(plugins): 修复自动审批插件 API 调用参数格式
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
import pytest
|
|
from core.config_loader import Config
|
|
from core.config_models import ConfigModel, NapCatWSModel, BotModel, RedisModel, DockerModel
|
|
|
|
|
|
class TestConfigLoader:
|
|
def test_config_initialization(self, tmp_path):
|
|
"""测试配置加载器初始化。"""
|
|
config_file = tmp_path / "config.toml"
|
|
config_file.write_text("""
|
|
[napcat_ws]
|
|
uri = "ws://localhost:3560"
|
|
token = "test_token"
|
|
|
|
[bot]
|
|
command = ["/"]
|
|
ignore_self_message = true
|
|
permission_denied_message = "权限不足，需要 {permission_name} 权限"
|
|
|
|
[redis]
|
|
host = "localhost"
|
|
port = 6379
|
|
db = 0
|
|
password = ""
|
|
|
|
[docker]
|
|
base_url = "unix:///var/run/docker.sock"
|
|
sandbox_image = "python-sandbox:latest"
|
|
timeout = 10
|
|
concurrency_limit = 5
|
|
tls_verify = false
|
|
""", encoding='utf-8')
|
|
config = Config(str(config_file))
|
|
assert config.path == config_file
|
|
assert isinstance(config._model, ConfigModel)
|
|
|
|
def test_config_properties(self, tmp_path):
|
|
"""测试配置属性访问。"""
|
|
config_file = tmp_path / "config.toml"
|
|
config_file.write_text("""
|
|
[napcat_ws]
|
|
uri = "ws://localhost:3560"
|
|
token = "test_token"
|
|
reconnect_interval = 5
|
|
|
|
[bot]
|
|
command = ["/"]
|
|
ignore_self_message = true
|
|
permission_denied_message = "权限不足，需要 {permission_name} 权限"
|
|
|
|
[redis]
|
|
host = "localhost"
|
|
port = 6379
|
|
db = 0
|
|
password = ""
|
|
|
|
[docker]
|
|
base_url = "unix:///var/run/docker.sock"
|
|
sandbox_image = "python-sandbox:latest"
|
|
timeout = 10
|
|
concurrency_limit = 5
|
|
tls_verify = false
|
|
""", encoding='utf-8')
|
|
config = Config(str(config_file))
|
|
assert isinstance(config.napcat_ws, NapCatWSModel)
|
|
assert config.napcat_ws.uri == "ws://localhost:3560"
|
|
assert config.napcat_ws.token == "test_token"
|
|
assert config.napcat_ws.reconnect_interval == 5
|
|
assert isinstance(config.bot, BotModel)
|
|
assert config.bot.command == ["/"]
|
|
assert config.bot.ignore_self_message is True
|
|
assert config.bot.permission_denied_message == "权限不足，需要 {permission_name} 权限"
|
|
assert isinstance(config.redis, RedisModel)
|
|
assert config.redis.host == "localhost"
|
|
assert config.redis.port == 6379
|
|
assert config.redis.db == 0
|
|
assert config.redis.password == ""
|
|
assert isinstance(config.docker, DockerModel)
|
|
assert config.docker.base_url == "unix:///var/run/docker.sock"
|
|
assert config.docker.sandbox_image == "python-sandbox:latest"
|
|
assert config.docker.timeout == 10
|
|
assert config.docker.concurrency_limit == 5
|
|
assert config.docker.tls_verify is False
|
|
|
|
def test_config_file_not_found(self, tmp_path):
|
|
"""测试配置文件不存在时的错误处理。"""
|
|
config_file = tmp_path / "non_existent_config.toml"
|
|
with pytest.raises(FileNotFoundError):
|
|
Config(str(config_file))
|
|
|
|
def test_config_invalid_format(self, tmp_path):
|
|
"""测试配置文件格式错误时的错误处理。"""
|
|
config_file = tmp_path / "invalid_config.toml"
|
|
config_file.write_text("invalid toml format", encoding='utf-8')
|
|
with pytest.raises(Exception):
|
|
Config(str(config_file))
|
|
|
|
def test_config_validation_error(self, tmp_path):
|
|
"""测试配置验证失败时的错误处理。"""
|
|
config_file = tmp_path / "invalid_config.toml"
|
|
config_file.write_text("""
|
|
[napcat_ws]
|
|
uri = "ws://localhost:3560"
|
|
|
|
[bot]
|
|
command = ["/"]
|
|
ignore_self_message = true
|
|
permission_denied_message = "权限不足，需要 {permission_name} 权限"
|
|
|
|
[redis]
|
|
host = "localhost"
|
|
port = 6379
|
|
db = 0
|
|
password = ""
|
|
|
|
[docker]
|
|
base_url = "unix:///var/run/docker.sock"
|
|
sandbox_image = "python-sandbox:latest"
|
|
timeout = 10
|
|
concurrency_limit = 5
|
|
tls_verify = false
|
|
""", encoding='utf-8')
|
|
with pytest.raises(Exception):
|
|
Config(str(config_file)) |