mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-30 20:54:32 +08:00
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
"""Chain 运行上下文注入和无参兼容 provider 测试。"""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from app.application.chain import context as chain_context
|
|
from app.application.chain import data as chain_data
|
|
from app.application.chain.context import ChainRuntimeContext
|
|
from app.application.configuration import ChainRuntimeConfig
|
|
from app.chain import ChainBase
|
|
from app.runtime.extensions.module.dispatcher import ModuleInvocationDispatcher
|
|
|
|
|
|
def _context() -> ChainRuntimeContext:
|
|
"""构造不连接数据库、不启动线程的最小 Chain 上下文。"""
|
|
return ChainRuntimeContext(
|
|
module_manager=Mock(),
|
|
plugin_manager=Mock(),
|
|
event_manager=Mock(),
|
|
message_oper=Mock(),
|
|
message_helper=Mock(),
|
|
file_cache=Mock(),
|
|
async_file_cache=Mock(),
|
|
message_queue_factory=Mock(return_value=Mock()),
|
|
module_dispatcher_factory=ModuleInvocationDispatcher,
|
|
configuration=ChainRuntimeConfig(media_extensions=(".mkv",)),
|
|
durable_event_writer=Mock(),
|
|
)
|
|
|
|
|
|
def test_chain_accepts_explicit_runtime_context() -> None:
|
|
"""新代码应能显式注入最小运行时依赖而不创建真实管理器。"""
|
|
context = _context()
|
|
|
|
chain = ChainBase(context)
|
|
|
|
assert chain.modulemanager is context.module_manager
|
|
assert chain.pluginmanager is context.plugin_manager
|
|
assert chain.eventmanager is context.event_manager
|
|
assert chain.messagehelper is context.message_helper
|
|
assert chain.durable_event_writer is context.durable_event_writer
|
|
context.message_queue_factory.assert_called_once_with(chain.run_module)
|
|
|
|
|
|
def test_no_arg_chain_uses_compatibility_context_provider(monkeypatch) -> None:
|
|
"""V3 兼容期内无参 Chain() 应从组合根 provider 获取相同上下文。"""
|
|
context = _context()
|
|
provider = Mock(return_value=context)
|
|
monkeypatch.setattr(chain_context, "_context_provider", provider)
|
|
|
|
chain = ChainBase()
|
|
|
|
provider.assert_called_once_with()
|
|
assert chain.modulemanager is context.module_manager
|
|
assert chain.pluginmanager is context.plugin_manager
|
|
|
|
|
|
def test_chain_runtime_context_rejects_unconfigured_provider(monkeypatch) -> None:
|
|
"""未由组合根配置运行上下文时必须显式拒绝无参 Chain。"""
|
|
monkeypatch.setattr(
|
|
chain_context,
|
|
"_context_provider",
|
|
chain_context._unconfigured_chain_runtime_context,
|
|
)
|
|
|
|
with pytest.raises(RuntimeError, match="Chain 运行上下文尚未由启动组合根配置"):
|
|
chain_context.get_chain_runtime_context()
|
|
|
|
|
|
def test_chain_data_registry_rejects_unconfigured_and_returns_factories(
|
|
monkeypatch,
|
|
) -> None:
|
|
"""数据 registry 未配置时拒绝访问，配置后按字段返回工厂实例。"""
|
|
monkeypatch.setattr(chain_data, "_ports", None)
|
|
|
|
with pytest.raises(RuntimeError, match="Chain 数据端口尚未配置"):
|
|
chain_data.get_chain_data_ports()
|
|
|
|
media_server = Mock()
|
|
user = Mock()
|
|
chain_data.configure_chain_data_ports(
|
|
site=Mock,
|
|
subscribe=Mock,
|
|
download_history=Mock,
|
|
transfer_history=Mock,
|
|
transfer_pending=Mock,
|
|
transfer_execution=Mock,
|
|
media_server=lambda: media_server,
|
|
download_failure=Mock,
|
|
user=lambda: user,
|
|
)
|
|
|
|
assert chain_data.get_chain_media_server_port() is media_server
|
|
assert chain_data.get_chain_user_port() is user
|