285 lines
9.9 KiB
Python
285 lines
9.9 KiB
Python
"""核心组件查询层与插件运行时聚合测试。"""
|
||
|
||
from types import SimpleNamespace
|
||
from typing import Any
|
||
|
||
import pytest
|
||
|
||
import src.plugin_runtime.integration as integration_module
|
||
|
||
from src.core.types import ActionInfo, ToolInfo
|
||
from src.plugin_runtime.component_query import component_query_service
|
||
from src.plugin_runtime.host.supervisor import PluginSupervisor
|
||
|
||
|
||
class _FakeRuntimeManager:
|
||
"""测试用插件运行时管理器。"""
|
||
|
||
def __init__(self, supervisor: PluginSupervisor, plugin_id: str, plugin_config: dict[str, Any]) -> None:
|
||
"""初始化测试用运行时管理器。
|
||
|
||
Args:
|
||
supervisor: 持有测试组件的监督器。
|
||
plugin_id: 目标插件 ID。
|
||
plugin_config: 需要返回的插件配置。
|
||
"""
|
||
|
||
self.supervisors = [supervisor]
|
||
self._plugin_id = plugin_id
|
||
self._plugin_config = plugin_config
|
||
|
||
def _get_supervisor_for_plugin(self, plugin_id: str) -> PluginSupervisor | None:
|
||
"""按插件 ID 返回对应监督器。
|
||
|
||
Args:
|
||
plugin_id: 目标插件 ID。
|
||
|
||
Returns:
|
||
PluginSupervisor | None: 命中时返回监督器。
|
||
"""
|
||
|
||
return self.supervisors[0] if plugin_id == self._plugin_id else None
|
||
|
||
def _load_plugin_config_for_supervisor(self, supervisor: Any, plugin_id: str) -> dict[str, Any]:
|
||
"""返回测试配置。
|
||
|
||
Args:
|
||
supervisor: 监督器实例。
|
||
plugin_id: 目标插件 ID。
|
||
|
||
Returns:
|
||
dict[str, Any]: 测试配置内容。
|
||
"""
|
||
|
||
del supervisor
|
||
if plugin_id != self._plugin_id:
|
||
return {}
|
||
return dict(self._plugin_config)
|
||
|
||
|
||
def _install_runtime_manager(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
supervisor: PluginSupervisor,
|
||
plugin_id: str,
|
||
plugin_config: dict[str, Any] | None = None,
|
||
) -> None:
|
||
"""为测试安装假的运行时管理器。
|
||
|
||
Args:
|
||
monkeypatch: pytest monkeypatch 对象。
|
||
supervisor: 持有测试组件的监督器。
|
||
plugin_id: 测试插件 ID。
|
||
plugin_config: 可选的测试配置内容。
|
||
"""
|
||
|
||
fake_manager = _FakeRuntimeManager(supervisor, plugin_id, plugin_config or {"enabled": True})
|
||
monkeypatch.setattr(integration_module, "get_plugin_runtime_manager", lambda: fake_manager)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_core_component_registry_reads_runtime_action_and_executor(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""核心查询层应直接读取运行时 Action，并返回 RPC 执行闭包。"""
|
||
|
||
plugin_id = "runtime_action_bridge_plugin"
|
||
action_name = "runtime_action_bridge_test"
|
||
supervisor = PluginSupervisor(plugin_dirs=[])
|
||
captured: dict[str, Any] = {}
|
||
|
||
supervisor.component_registry.register_component(
|
||
name=action_name,
|
||
component_type="ACTION",
|
||
plugin_id=plugin_id,
|
||
metadata={
|
||
"description": "发送一个测试回复",
|
||
"enabled": True,
|
||
"activation_type": "keyword",
|
||
"activation_probability": 0.25,
|
||
"activation_keywords": ["测试", "hello"],
|
||
"action_parameters": {"target": "目标对象"},
|
||
"action_require": ["需要发送回复时使用"],
|
||
"associated_types": ["text"],
|
||
"parallel_action": True,
|
||
},
|
||
)
|
||
_install_runtime_manager(monkeypatch, supervisor, plugin_id, {"enabled": True, "mode": "test"})
|
||
|
||
async def fake_invoke_plugin(
|
||
method: str,
|
||
plugin_id: str,
|
||
component_name: str,
|
||
args: dict[str, Any] | None = None,
|
||
timeout_ms: int = 30000,
|
||
) -> Any:
|
||
"""模拟动作 RPC 调用。"""
|
||
|
||
captured["method"] = method
|
||
captured["plugin_id"] = plugin_id
|
||
captured["component_name"] = component_name
|
||
captured["args"] = args or {}
|
||
captured["timeout_ms"] = timeout_ms
|
||
return SimpleNamespace(payload={"success": True, "result": (True, "runtime action executed")})
|
||
|
||
monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
|
||
|
||
action_info = component_query_service.get_action_info(action_name)
|
||
assert isinstance(action_info, ActionInfo)
|
||
assert action_info.plugin_name == plugin_id
|
||
assert action_info.description == "发送一个测试回复"
|
||
assert action_info.activation_keywords == ["测试", "hello"]
|
||
assert action_info.random_activation_probability == 0.25
|
||
assert action_info.parallel_action is True
|
||
assert action_name in component_query_service.get_default_actions()
|
||
assert component_query_service.get_plugin_config(plugin_id) == {"enabled": True, "mode": "test"}
|
||
|
||
executor = component_query_service.get_action_executor(action_name)
|
||
assert executor is not None
|
||
|
||
success, reason = await executor(
|
||
action_data={"target": "MaiBot"},
|
||
action_reasoning="当前适合使用这个动作",
|
||
cycle_timers={"planner": 0.1},
|
||
thinking_id="tid-1",
|
||
chat_stream=SimpleNamespace(session_id="stream-1"),
|
||
log_prefix="[test]",
|
||
shutting_down=False,
|
||
plugin_config={"enabled": True},
|
||
)
|
||
|
||
assert success is True
|
||
assert reason == "runtime action executed"
|
||
assert captured["method"] == "plugin.invoke_action"
|
||
assert captured["plugin_id"] == plugin_id
|
||
assert captured["component_name"] == action_name
|
||
assert captured["args"]["stream_id"] == "stream-1"
|
||
assert captured["args"]["chat_id"] == "stream-1"
|
||
assert captured["args"]["reasoning"] == "当前适合使用这个动作"
|
||
assert captured["args"]["target"] == "MaiBot"
|
||
assert captured["args"]["action_data"] == {"target": "MaiBot"}
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_core_component_registry_reads_runtime_command_and_executor(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""核心查询层应直接使用运行时命令匹配与执行闭包。"""
|
||
|
||
plugin_id = "runtime_command_bridge_plugin"
|
||
command_name = "runtime_command_bridge_test"
|
||
supervisor = PluginSupervisor(plugin_dirs=[])
|
||
captured: dict[str, Any] = {}
|
||
|
||
supervisor.component_registry.register_component(
|
||
name=command_name,
|
||
component_type="COMMAND",
|
||
plugin_id=plugin_id,
|
||
metadata={
|
||
"description": "测试命令",
|
||
"enabled": True,
|
||
"command_pattern": r"^/test(?:\s+.+)?$",
|
||
"aliases": ["/hello"],
|
||
"intercept_message_level": 1,
|
||
},
|
||
)
|
||
_install_runtime_manager(monkeypatch, supervisor, plugin_id, {"mode": "command"})
|
||
|
||
async def fake_invoke_plugin(
|
||
method: str,
|
||
plugin_id: str,
|
||
component_name: str,
|
||
args: dict[str, Any] | None = None,
|
||
timeout_ms: int = 30000,
|
||
) -> Any:
|
||
"""模拟命令 RPC 调用。"""
|
||
|
||
captured["method"] = method
|
||
captured["plugin_id"] = plugin_id
|
||
captured["component_name"] = component_name
|
||
captured["args"] = args or {}
|
||
captured["timeout_ms"] = timeout_ms
|
||
return SimpleNamespace(payload={"success": True, "result": (True, "command ok", True)})
|
||
|
||
monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
|
||
|
||
matched = component_query_service.find_command_by_text("/test hello")
|
||
assert matched is not None
|
||
command_executor, matched_groups, command_info = matched
|
||
|
||
assert matched_groups == {}
|
||
assert command_info.plugin_name == plugin_id
|
||
assert command_info.command_pattern == r"^/test(?:\s+.+)?$"
|
||
|
||
success, response_text, intercept = await command_executor(
|
||
message=SimpleNamespace(processed_plain_text="/test hello", session_id="stream-2"),
|
||
plugin_config={"mode": "command"},
|
||
matched_groups=matched_groups,
|
||
)
|
||
|
||
assert success is True
|
||
assert response_text == "command ok"
|
||
assert intercept is True
|
||
assert captured["method"] == "plugin.invoke_command"
|
||
assert captured["plugin_id"] == plugin_id
|
||
assert captured["component_name"] == command_name
|
||
assert captured["args"]["text"] == "/test hello"
|
||
assert captured["args"]["stream_id"] == "stream-2"
|
||
assert captured["args"]["plugin_config"] == {"mode": "command"}
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_core_component_registry_reads_runtime_tools_and_executor(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""核心查询层应直接读取运行时 Tool，并返回 RPC 执行闭包。"""
|
||
|
||
plugin_id = "runtime_tool_bridge_plugin"
|
||
tool_name = "runtime_tool_bridge_test"
|
||
supervisor = PluginSupervisor(plugin_dirs=[])
|
||
|
||
supervisor.component_registry.register_component(
|
||
name=tool_name,
|
||
component_type="TOOL",
|
||
plugin_id=plugin_id,
|
||
metadata={
|
||
"description": "测试工具",
|
||
"enabled": True,
|
||
"parameters": [
|
||
{
|
||
"name": "query",
|
||
"param_type": "string",
|
||
"description": "查询词",
|
||
"required": True,
|
||
}
|
||
],
|
||
},
|
||
)
|
||
_install_runtime_manager(monkeypatch, supervisor, plugin_id)
|
||
|
||
async def fake_invoke_plugin(
|
||
method: str,
|
||
plugin_id: str,
|
||
component_name: str,
|
||
args: dict[str, Any] | None = None,
|
||
timeout_ms: int = 30000,
|
||
) -> Any:
|
||
"""模拟工具 RPC 调用。"""
|
||
|
||
del timeout_ms
|
||
assert method == "plugin.invoke_tool"
|
||
assert plugin_id == "runtime_tool_bridge_plugin"
|
||
assert component_name == "runtime_tool_bridge_test"
|
||
assert args == {"query": "MaiBot"}
|
||
return SimpleNamespace(payload={"success": True, "result": {"content": "tool ok"}})
|
||
|
||
monkeypatch.setattr(supervisor, "invoke_plugin", fake_invoke_plugin)
|
||
|
||
tool_info = component_query_service.get_tool_info(tool_name)
|
||
assert isinstance(tool_info, ToolInfo)
|
||
assert tool_info.tool_description == "测试工具"
|
||
assert tool_name in component_query_service.get_llm_available_tools()
|
||
|
||
executor = component_query_service.get_tool_executor(tool_name)
|
||
assert executor is not None
|
||
assert await executor({"query": "MaiBot"}) == {"content": "tool ok"}
|