mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
169a856a3a
Coverage improvements: - agent/tool/core: 150+ tests (FunctionTool, ToolSet, ToolSchema, run_context, messages, registry) - provider: 152 tests (entities, Provider ABC, register, manager) - star module: 58 tests (Context, PluginManager, StarHandler) - platform: 71 tests (Platform ABC, manager, MessageSession) - pipeline: 130 tests (process/respond stages, scheduler, bootstrap, conversation_mgr) - kb+skills: 96 tests (mgr, helpers, chunkers, skill_manager) - config+builtins: 92 tests (astrbot_config, config_mgr, default, LTM, star main) - cron+db+event_bus: 89 tests (cron events/manager, PO models, BaseDatabase, EventBus) - dashboard routes: expanded smoke tests for all route modules
212 lines
7.0 KiB
Python
212 lines
7.0 KiB
Python
"""Mock-based unit tests for default config constants and structure."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.config.default import (
|
|
DB_PATH,
|
|
DEFAULT_CONFIG,
|
|
DEFAULT_VALUE_MAP,
|
|
VERSION,
|
|
WEBHOOK_SUPPORTED_PLATFORMS,
|
|
)
|
|
|
|
|
|
class TestVersionConstant:
|
|
"""Tests for the VERSION constant."""
|
|
|
|
def test_version_is_string(self):
|
|
assert isinstance(VERSION, str)
|
|
|
|
def test_version_format(self):
|
|
parts = VERSION.split(".")
|
|
assert len(parts) == 3
|
|
for p in parts:
|
|
assert p.isdigit()
|
|
|
|
|
|
class TestDBPath:
|
|
"""Tests for DB_PATH."""
|
|
|
|
@patch("astrbot.core.config.default.get_astrbot_data_path", return_value="/data")
|
|
def test_db_path_uses_data_path(self, mock_get_path):
|
|
"""DB_PATH should join data path with the database filename."""
|
|
# Reimport to pick up the patched value
|
|
import importlib
|
|
|
|
from astrbot.core.config import default as default_mod
|
|
|
|
importlib.reload(default_mod)
|
|
assert "data_v4.db" in default_mod.DB_PATH
|
|
|
|
def test_db_path_ends_with_db(self):
|
|
assert DB_PATH.endswith(".db")
|
|
|
|
|
|
class TestDEFAULT_VALUE_MAP:
|
|
"""Tests for DEFAULT_VALUE_MAP structure."""
|
|
|
|
def test_contains_expected_types(self):
|
|
expected = {"int", "float", "bool", "string", "text", "list", "file", "object", "template_list"}
|
|
assert set(DEFAULT_VALUE_MAP.keys()) == expected
|
|
|
|
def test_default_values_are_correct_types(self):
|
|
assert isinstance(DEFAULT_VALUE_MAP["int"], int)
|
|
assert isinstance(DEFAULT_VALUE_MAP["float"], float)
|
|
assert isinstance(DEFAULT_VALUE_MAP["bool"], bool)
|
|
assert isinstance(DEFAULT_VALUE_MAP["string"], str)
|
|
assert isinstance(DEFAULT_VALUE_MAP["text"], str)
|
|
assert isinstance(DEFAULT_VALUE_MAP["list"], list)
|
|
assert isinstance(DEFAULT_VALUE_MAP["file"], list)
|
|
assert isinstance(DEFAULT_VALUE_MAP["object"], dict)
|
|
assert isinstance(DEFAULT_VALUE_MAP["template_list"], list)
|
|
|
|
def test_specific_default_values(self):
|
|
assert DEFAULT_VALUE_MAP["int"] == 0
|
|
assert DEFAULT_VALUE_MAP["float"] == 0.0
|
|
assert DEFAULT_VALUE_MAP["bool"] is False
|
|
assert DEFAULT_VALUE_MAP["string"] == ""
|
|
assert DEFAULT_VALUE_MAP["list"] == []
|
|
assert DEFAULT_VALUE_MAP["object"] == {}
|
|
assert DEFAULT_VALUE_MAP["template_list"] == []
|
|
|
|
|
|
class TestWEBHOOK_SUPPORTED_PLATFORMS:
|
|
"""Tests for the webhook platforms list."""
|
|
|
|
def test_is_list_of_strings(self):
|
|
assert isinstance(WEBHOOK_SUPPORTED_PLATFORMS, list)
|
|
for p in WEBHOOK_SUPPORTED_PLATFORMS:
|
|
assert isinstance(p, str)
|
|
|
|
def test_contains_key_platforms(self):
|
|
assert "qq_official_webhook" in WEBHOOK_SUPPORTED_PLATFORMS
|
|
assert "weixin_official_account" in WEBHOOK_SUPPORTED_PLATFORMS
|
|
assert "slack" in WEBHOOK_SUPPORTED_PLATFORMS
|
|
assert "lark" in WEBHOOK_SUPPORTED_PLATFORMS
|
|
|
|
|
|
class TestDEFAULT_CONFIGStructure:
|
|
"""Tests for the top-level keys and nested structure of DEFAULT_CONFIG."""
|
|
|
|
def test_top_level_keys(self):
|
|
expected_keys = {
|
|
"config_version",
|
|
"platform_settings",
|
|
"provider_sources",
|
|
"provider",
|
|
"provider_settings",
|
|
"subagent_orchestrator",
|
|
"provider_stt_settings",
|
|
"provider_tts_settings",
|
|
"provider_ltm_settings",
|
|
"content_safety",
|
|
"admins_id",
|
|
"t2i",
|
|
"http_proxy",
|
|
"no_proxy",
|
|
"dashboard",
|
|
"platform",
|
|
"platform_specific",
|
|
"wake_prefix",
|
|
"log_level",
|
|
"persona",
|
|
"timezone",
|
|
"callback_api_base",
|
|
"default_kb_collection",
|
|
"plugin_set",
|
|
"kb_names",
|
|
"kb_fusion_top_k",
|
|
"kb_final_top_k",
|
|
"kb_agentic_mode",
|
|
"disable_builtin_commands",
|
|
"t2i_word_threshold",
|
|
"t2i_strategy",
|
|
"t2i_endpoint",
|
|
"t2i_use_file_service",
|
|
"t2i_active_template",
|
|
"log_file_enable",
|
|
"log_file_path",
|
|
"log_file_max_mb",
|
|
"temp_dir_max_size",
|
|
"trace_enable",
|
|
"trace_log_enable",
|
|
"trace_log_path",
|
|
"trace_log_max_mb",
|
|
"pip_install_arg",
|
|
"pypi_index_url",
|
|
}
|
|
for key in expected_keys:
|
|
assert key in DEFAULT_CONFIG, f"Missing top-level key: {key}"
|
|
|
|
def test_config_version_is_two(self):
|
|
assert DEFAULT_CONFIG["config_version"] == 2
|
|
|
|
def test_platform_settings_structure(self):
|
|
ps = DEFAULT_CONFIG["platform_settings"]
|
|
assert "unique_session" in ps
|
|
assert "rate_limit" in ps
|
|
assert "reply_prefix" in ps
|
|
assert isinstance(ps["rate_limit"], dict)
|
|
assert ps["rate_limit"]["strategy"] in ("stall", "discard")
|
|
|
|
def test_provider_ltm_settings_structure(self):
|
|
ltm = DEFAULT_CONFIG["provider_ltm_settings"]
|
|
assert "group_icl_enable" in ltm
|
|
assert "group_message_max_cnt" in ltm
|
|
assert "image_caption" in ltm
|
|
assert "active_reply" in ltm
|
|
assert isinstance(ltm["active_reply"], dict)
|
|
assert ltm["active_reply"]["method"] == "possibility_reply"
|
|
|
|
def test_dashboard_settings(self):
|
|
dash = DEFAULT_CONFIG["dashboard"]
|
|
assert dash["enable"] is True
|
|
assert dash["username"] == "astrbot"
|
|
assert dash["port"] == 6185
|
|
assert "ssl" in dash
|
|
|
|
def test_provider_settings_defaults(self):
|
|
ps = DEFAULT_CONFIG["provider_settings"]
|
|
assert ps["enable"] is True
|
|
assert ps["default_provider_id"] == ""
|
|
assert ps["agent_runner_type"] == "local"
|
|
assert ps["llm_safety_mode"] is True
|
|
|
|
def test_admins_id_default(self):
|
|
assert DEFAULT_CONFIG["admins_id"] == ["astrbot"]
|
|
|
|
def test_wake_prefix_default(self):
|
|
assert DEFAULT_CONFIG["wake_prefix"] == ["/"]
|
|
|
|
def test_log_level_default(self):
|
|
assert DEFAULT_CONFIG["log_level"] == "INFO"
|
|
|
|
def test_platform_specific_contains_lark_telegram_discord(self):
|
|
ps = DEFAULT_CONFIG["platform_specific"]
|
|
assert "lark" in ps
|
|
assert "telegram" in ps
|
|
assert "discord" in ps
|
|
|
|
def test_no_proxy_contains_localhost(self):
|
|
assert "localhost" in DEFAULT_CONFIG["no_proxy"]
|
|
|
|
def test_subagent_orchestrator_defaults(self):
|
|
sa = DEFAULT_CONFIG["subagent_orchestrator"]
|
|
assert sa["main_enable"] is False
|
|
assert isinstance(sa["agents"], list)
|
|
|
|
def test_quoted_message_parser_defaults(self):
|
|
qmp = DEFAULT_CONFIG["provider_settings"]["quoted_message_parser"]
|
|
assert qmp["max_component_chain_depth"] == 4
|
|
assert qmp["max_forward_node_depth"] == 6
|
|
assert qmp["max_forward_fetch"] == 32
|
|
|
|
def test_sandbox_defaults(self):
|
|
sb = DEFAULT_CONFIG["provider_settings"]["sandbox"]
|
|
assert sb["booter"] == "shipyard_neo"
|
|
assert sb["shipyard_neo_ttl"] == 3600
|