mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
jxxghp
8d62f33155
- projection 只接受映射类型的 get_module 声明，非法值跳过并记日志 - dispatcher 同步/异步路径防御非映射方法表，单个坏插件被隔离 - media 端点改用 application 层插件运行时门面，消除分层违规依赖 - 图片能力测试打桩 models.dev 目录边界，适配仓库内离线目录留空 Closes #6346
237 lines
7.4 KiB
Python
237 lines
7.4 KiB
Python
"""模块调用调度器的同步、异步协议回归测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from app.runtime.extensions.module.dispatcher import ModuleInvocationDispatcher
|
|
|
|
|
|
class _PluginCatalog:
|
|
"""提供固定插件方法表的内存目录。"""
|
|
|
|
def __init__(self, modules: dict) -> None:
|
|
"""保存测试提供的插件模块快照。"""
|
|
self.modules = modules
|
|
|
|
def get_plugin_modules(self) -> dict:
|
|
"""返回当前插件模块快照。"""
|
|
return self.modules
|
|
|
|
|
|
class _ModuleCatalog:
|
|
"""提供固定宿主模块序列的内存目录。"""
|
|
|
|
def __init__(self, modules: list) -> None:
|
|
"""保存测试提供的宿主模块。"""
|
|
self.modules = modules
|
|
|
|
def get_running_modules(self, _method: str) -> list:
|
|
"""返回所有测试模块，由调度器负责优先级排序。"""
|
|
return list(self.modules)
|
|
|
|
|
|
class _Module:
|
|
"""实现可配置优先级和调用函数的测试宿主模块。"""
|
|
|
|
def __init__(self, name: str, priority: int, func: Callable) -> None:
|
|
"""保存展示名、优先级和测试调用函数。"""
|
|
self._name = name
|
|
self._priority = priority
|
|
self._func = func
|
|
|
|
def get_name(self) -> str:
|
|
"""返回测试模块展示名。"""
|
|
return self._name
|
|
|
|
def get_priority(self) -> int:
|
|
"""返回调度优先级。"""
|
|
return self._priority
|
|
|
|
def execute(self, *args, **kwargs):
|
|
"""把模块调用转发到测试函数。"""
|
|
return self._func(*args, **kwargs)
|
|
|
|
|
|
def _dispatcher(
|
|
*,
|
|
plugins: dict | None = None,
|
|
modules: list | None = None,
|
|
async_runner: Callable | None = None,
|
|
) -> tuple[ModuleInvocationDispatcher, Mock, Mock, Mock]:
|
|
"""构造完全内存化的调度器及错误策略替身。"""
|
|
plugin_error = Mock()
|
|
system_error = Mock()
|
|
rate_error = Mock()
|
|
|
|
async def default_runner(func, *args, **kwargs):
|
|
"""在测试事件循环中直接运行同步函数。"""
|
|
return func(*args, **kwargs)
|
|
|
|
dispatcher = ModuleInvocationDispatcher(
|
|
module_catalog=_ModuleCatalog(modules or []),
|
|
plugin_catalog=_PluginCatalog(plugins or {}),
|
|
plugin_error_handler=plugin_error,
|
|
system_error_handler=system_error,
|
|
rate_limit_handler=rate_error,
|
|
async_function_runner=async_runner or default_runner,
|
|
)
|
|
return dispatcher, plugin_error, system_error, rate_error
|
|
|
|
|
|
def test_plugin_scalar_short_circuits_system_modules() -> None:
|
|
"""插件返回非空标量时不得继续执行宿主模块。"""
|
|
system_call = Mock(return_value="system")
|
|
dispatcher, _, _, _ = _dispatcher(
|
|
plugins={("P1", "插件一"): {"execute": lambda: "plugin"}},
|
|
modules=[_Module("系统", 10, system_call)],
|
|
)
|
|
|
|
assert dispatcher.dispatch("execute") == "plugin"
|
|
system_call.assert_not_called()
|
|
|
|
|
|
def test_list_results_merge_in_plugin_then_priority_order() -> None:
|
|
"""列表结果应先按插件顺序合并，再按宿主优先级继续合并。"""
|
|
calls = []
|
|
|
|
def result(value: str) -> Callable:
|
|
"""生成记录调用顺序并返回单项列表的模块函数。"""
|
|
return lambda: calls.append(value) or [value]
|
|
|
|
dispatcher, _, _, _ = _dispatcher(
|
|
plugins={
|
|
("P1", "插件一"): {"execute": result("plugin-1")},
|
|
("P2", "插件二"): {"execute": result("plugin-2")},
|
|
},
|
|
modules=[
|
|
_Module("慢模块", 20, result("system-20")),
|
|
_Module("快模块", 10, result("system-10")),
|
|
],
|
|
)
|
|
|
|
assert dispatcher.dispatch("execute") == [
|
|
"plugin-1",
|
|
"plugin-2",
|
|
"system-10",
|
|
"system-20",
|
|
]
|
|
assert calls == ["plugin-1", "plugin-2", "system-10", "system-20"]
|
|
|
|
|
|
def test_system_signature_relay_passes_previous_result() -> None:
|
|
"""单参数宿主方法应接收上一模块的非列表结果。"""
|
|
class FirstModule:
|
|
"""产生首个字典结果的测试模块。"""
|
|
|
|
@staticmethod
|
|
def get_name() -> str:
|
|
"""返回测试模块名。"""
|
|
return "第一步"
|
|
|
|
@staticmethod
|
|
def get_priority() -> int:
|
|
"""返回第一优先级。"""
|
|
return 10
|
|
|
|
@staticmethod
|
|
def execute() -> dict:
|
|
"""产生首个模块结果。"""
|
|
return {"value": 1}
|
|
|
|
class SecondModule:
|
|
"""消费上一结果的测试模块。"""
|
|
|
|
@staticmethod
|
|
def get_name() -> str:
|
|
"""返回测试模块名。"""
|
|
return "第二步"
|
|
|
|
@staticmethod
|
|
def get_priority() -> int:
|
|
"""返回第二优先级。"""
|
|
return 20
|
|
|
|
@staticmethod
|
|
def execute(previous: dict) -> dict:
|
|
"""接收上一模块结果并生成下一结果。"""
|
|
return {"value": previous["value"] + 1}
|
|
|
|
dispatcher, _, _, _ = _dispatcher(
|
|
modules=[SecondModule(), FirstModule()]
|
|
)
|
|
|
|
assert dispatcher.dispatch("execute") == {"value": 2}
|
|
|
|
|
|
def test_module_exception_uses_error_policy_and_continues() -> None:
|
|
"""普通异常应交给错误策略，后续空结果模块仍可继续运行。"""
|
|
def broken():
|
|
"""模拟模块执行失败。"""
|
|
raise RuntimeError("broken")
|
|
|
|
dispatcher, _, system_error, _ = _dispatcher(
|
|
modules=[
|
|
_Module("失败模块", 10, broken),
|
|
_Module("后续模块", 20, lambda: "ok"),
|
|
],
|
|
)
|
|
|
|
assert dispatcher.dispatch("execute") == "ok"
|
|
system_error.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_dispatch_awaits_coroutines_and_offloads_sync_functions() -> None:
|
|
"""异步路径应直接等待协程，并通过注入执行器运行同步方法。"""
|
|
offloaded = []
|
|
|
|
async def async_runner(func, *args, **kwargs):
|
|
"""记录被移出事件循环的同步函数。"""
|
|
offloaded.append(func)
|
|
return func(*args, **kwargs)
|
|
|
|
async def plugin_call():
|
|
"""返回插件列表结果。"""
|
|
return ["plugin"]
|
|
|
|
sync_module = _Module("同步模块", 10, lambda: ["system"])
|
|
dispatcher, _, _, _ = _dispatcher(
|
|
plugins={("P1", "插件一"): {"execute": plugin_call}},
|
|
modules=[sync_module],
|
|
async_runner=async_runner,
|
|
)
|
|
|
|
assert await dispatcher.async_dispatch("execute") == ["plugin", "system"]
|
|
assert offloaded == [sync_module.execute]
|
|
|
|
|
|
def test_plugin_non_mapping_module_decl_is_reported_and_skipped() -> None:
|
|
"""插件把方法表声明成 list 时走错误策略，且不影响后续健康插件。"""
|
|
dispatcher, plugin_error, _, _ = _dispatcher(
|
|
plugins={
|
|
("Bad", "坏插件"): ["not-a-mapping"],
|
|
("Good", "好插件"): {"execute": lambda: "ok"},
|
|
},
|
|
)
|
|
|
|
assert dispatcher.dispatch("execute") == "ok"
|
|
plugin_error.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_plugin_non_mapping_module_decl_is_reported_and_skipped() -> None:
|
|
"""异步路径下坏插件同样被隔离，嵌套补丁场景不再冒泡击穿调度。"""
|
|
dispatcher, plugin_error, _, _ = _dispatcher(
|
|
plugins={
|
|
("Bad", "坏插件"): ["not-a-mapping"],
|
|
("Good", "好插件"): {"execute": lambda: "ok"},
|
|
},
|
|
)
|
|
|
|
assert await dispatcher.async_dispatch("execute") == "ok"
|
|
plugin_error.assert_called_once()
|