MoviePilot/tests/test_event_runtime_components.py

"""事件注册、绑定、调度和错误策略组件的独立测试。"""

import sys
import threading
import types
from unittest.mock import Mock

import pytest

from app.foundation.singleton import Singleton
from app.runtime.event.binding import (
    EventBindingResolver,
    EventHandlerBinding,
)
from app.runtime.event.dispatch import EventDispatcher
from app.runtime.event.errors import EventErrorPolicy
from app.runtime.event.registry import EventRegistry
from app.runtime.events import Event
from app.runtime.extensions.plugin import manager as plugin_manager_module
from app.runtime.extensions.plugin.manager import PluginManager
from app.schemas.types import EventType
from app.startup.initializers.modules import (
    get_config_reload_handler_providers,
    get_host_event_handler_factories,
)


class _UnmanagedHandler:
    """记录构造次数，用于证明绑定未命中时不会被总线实例化。"""

    constructed = 0

    def __init__(self) -> None:
        """记录任何非预期的隐式构造。"""
        type(self).constructed += 1

    def handle(self, _event: Event) -> None:
        """提供可解析的实例方法声明。"""


class _PluginHandler:
    """代表插件运行时登记的事件 owner。"""

    @staticmethod
    def get_name() -> str:
        """返回测试插件名称。"""
        return "测试插件"


def _free_function_handler(_event: Event) -> None:
    """模块级自由函数处理器，用于验证直调路径保持不变。"""


def test_binding_miss_does_not_construct_handler_owner() -> None:
    """resolver 未命中时只记录诊断，不能调用 owner_class()。"""
    resolvers = {}
    binding = EventBindingResolver(
        lock=threading.Lock(),
        resolvers=lambda: resolvers,
    )
    _UnmanagedHandler.constructed = 0

    assert binding.resolve(_UnmanagedHandler.handle) is None
    assert _UnmanagedHandler.constructed == 0
    assert binding.unresolved_handlers() == (
        f"{__name__}._UnmanagedHandler.handle",
    )


def test_unloaded_module_class_handler_is_skipped() -> None:
    """模块缓存被清除后，残留的类方法声明必须跳过而非直调原始函数。"""
    fake_name = "tests._fake_unloaded_plugin"
    fake_module = types.ModuleType(fake_name)
    sys.modules[fake_name] = fake_module
    try:
        # 在伪模块命名空间内构造类，使处理器 __module__ 指向该模块
        exec(
            "class _ResidualPlugin:\n"
            "    def reload(self, event):\n"
            "        raise AssertionError('residual handler must not run')\n",
            fake_module.__dict__,
        )
        residual_handler = fake_module._ResidualPlugin.reload
        # 模拟插件重载 stop 阶段清除模块缓存后的残留注册
        del sys.modules[fake_name]

        binding = EventBindingResolver(
            lock=threading.Lock(),
            resolvers=lambda: {},
        )
        assert binding.resolve(residual_handler) is None
        assert binding.unresolved_handlers() == (
            f"{fake_name}._ResidualPlugin.reload",
        )
    finally:
        sys.modules.pop(fake_name, None)


def test_quiesced_handler_stays_disabled_after_module_unload() -> None:
    """事件快照中的旧 handler 在模块卸载后仍应按已禁用状态跳过。"""
    fake_name = "tests._fake_quiesced_plugin"
    fake_module = types.ModuleType(fake_name)
    sys.modules[fake_name] = fake_module
    try:
        exec(
            "class _QuiescedPlugin:\n"
            "    def reload(self, event):\n"
            "        raise AssertionError('quiesced handler must not run')\n",
            fake_module.__dict__,
        )
        residual_handler = fake_module._QuiescedPlugin.reload
        broadcast_subscribers: dict = {}
        disabled_handlers: set[str] = set()
        disabled_classes: set[str] = set()
        registry = EventRegistry(
            lock=threading.Lock(),
            broadcast_subscribers=lambda: broadcast_subscribers,
            chain_subscribers=lambda: {},
            disabled_handlers=lambda: disabled_handlers,
            disabled_classes=lambda: disabled_classes,
        )
        registry.add(EventType.PluginReload, residual_handler, 0)
        registry.disable(fake_module._QuiescedPlugin)
        del sys.modules[fake_name]

        assert EventRegistry.handler_identifier(residual_handler) == (
            f"{fake_name}._QuiescedPlugin.reload"
        )
        assert EventRegistry.handler_class_identifier(residual_handler) == (
            f"{fake_name}._QuiescedPlugin"
        )
        assert not registry.is_handler_enabled(residual_handler)

        binding = EventBindingResolver(
            lock=threading.Lock(),
            resolvers=lambda: {},
        )
        dispatcher = EventDispatcher(
            registry=registry,
            binding_resolver=binding,
            event_factory=Event,
            error_handler=lambda **_kwargs: None,
            async_handle_sink=lambda _handle: True,
            sync_handle_sink=lambda _callback, _args: True,
        )
        dispatcher.dispatch_broadcast_strict(
            Event(EventType.PluginReload, {"plugin_id": "_QuiescedPlugin"}),
            lambda _awaitable: None,
        )
        assert binding.unresolved_handlers() == ()
    finally:
        sys.modules.pop(fake_name, None)


def test_unloaded_module_decorator_wrapped_method_is_skipped() -> None:
    """装饰器包装的类方法限定名含 <locals>，模块卸载后也必须跳过而非直调。"""
    fake_name = "tests._fake_unloaded_decorated_plugin"
    fake_module = types.ModuleType(fake_name)
    sys.modules[fake_name] = fake_module
    try:
        exec(
            "def _deco(f):\n"
            "    def wrapper(self, event):\n"
            "        return f(self, event)\n"
            "    return wrapper\n"
            "class _DecoratedPlugin:\n"
            "    @_deco\n"
            "    def send_msg(self, event):\n"
            "        raise AssertionError('residual handler must not run')\n",
            fake_module.__dict__,
        )
        residual_handler = fake_module._DecoratedPlugin.send_msg
        # 装饰器包装后限定名含 <locals>，不能因此被误判为自由函数
        assert "<locals>" in residual_handler.__qualname__
        del sys.modules[fake_name]

        binding = EventBindingResolver(
            lock=threading.Lock(),
            resolvers=lambda: {},
        )
        assert binding.resolve(residual_handler) is None
        assert binding.unresolved_handlers() == (
            f"{fake_name}._deco.<locals>.wrapper",
        )
    finally:
        sys.modules.pop(fake_name, None)


def test_free_function_handler_still_invoked_directly() -> None:
    """自由函数处理器不属于类声明，保持直调路径不被新跳过逻辑影响。"""
    binding = EventBindingResolver(
        lock=threading.Lock(),
        resolvers=lambda: {},
    )

    resolved = binding.resolve(_free_function_handler)

    assert resolved is not None
    method, handler_binding, class_name, method_name = resolved
    assert method is _free_function_handler
    assert handler_binding.run_sync_in_threadpool is True
    assert class_name == ""
    assert method_name == "_free_function_handler"


def test_binding_uses_explicit_resolver_instance() -> None:
    """显式 resolver 应返回当前托管实例上的绑定方法。"""
    instance = object.__new__(_UnmanagedHandler)
    resolvers = {
        "test": lambda owner: EventHandlerBinding(
            instance=instance,
            owner_name="托管处理器",
        )
        if owner is _UnmanagedHandler
        else None
    }
    binding = EventBindingResolver(
        lock=threading.Lock(),
        resolvers=lambda: resolvers,
    )

    method, resolved, class_name, method_name = binding.resolve(
        _UnmanagedHandler.handle
    )

    assert method.__self__ is instance
    assert resolved.owner_name == "托管处理器"
    assert class_name == "_UnmanagedHandler"
    assert method_name == "handle"


def test_plugin_resolver_requires_exact_registered_class() -> None:
    """同名宿主类不能被 plugins resolver 误识别为插件 owner。"""
    manager = object.__new__(PluginManager)
    plugin = _PluginHandler()
    manager._plugins = {_PluginHandler.__name__: _PluginHandler}
    manager._running_plugins = {_PluginHandler.__name__: plugin}
    impostor = type(_PluginHandler.__name__, (), {})

    assert manager.resolve_event_handler_instance(
        _PluginHandler
    ) == EventHandlerBinding(
        instance=plugin,
        owner_name="测试插件",
        run_sync_in_threadpool=True,
    )
    assert manager.resolve_event_handler_instance(impostor) is None


def test_plugin_resolver_does_not_materialize_manager(monkeypatch) -> None:
    """残留 resolver 在管理器不存在时必须跳过，不能构造插件 Runtime。"""
    singleton_key = (PluginManager, (), frozenset())
    monkeypatch.delitem(Singleton._instances, singleton_key, raising=False)
    runtime_factory = Mock(side_effect=AssertionError("不得构造 PluginManager"))
    monkeypatch.setattr(
        plugin_manager_module,
        "_plugin_runtime_factory",
        runtime_factory,
    )

    assert plugin_manager_module._resolve_plugin_handler_instance(
        _PluginHandler
    ) is None
    runtime_factory.assert_not_called()


def test_system_error_failure_does_not_rebroadcast() -> None:
    """SystemError 处理器自身失败时只能通知和日志降级，不能再次发送事件。"""
    notifier = Mock()
    emit = Mock()
    policy = EventErrorPolicy(
        notifier=lambda: notifier,
        emit_system_error=emit,
    )

    policy.handle(
        event=Event(EventType.SystemError, {}),
        module_name="测试模块",
        class_name="BrokenHandler",
        method_name="handle",
        error=RuntimeError("broken"),
    )

    notifier.assert_called_once()
    emit.assert_not_called()


def test_regular_event_failure_emits_one_system_error() -> None:
    """普通事件失败应生成一次结构稳定的 SystemError 载荷。"""
    emit = Mock()
    policy = EventErrorPolicy(
        notifier=lambda: None,
        emit_system_error=emit,
    )

    policy.handle(
        event=Event(EventType.ConfigChanged, {}),
        module_name="测试模块",
        class_name="BrokenHandler",
        method_name="handle",
        error=RuntimeError("broken"),
    )

    payload = emit.call_args.args[0]
    assert payload["type"] == "event"
    assert payload["event_type"] is EventType.ConfigChanged
    assert payload["event_handle"] == "BrokenHandler.handle"
    assert payload["error"] == "broken"


def test_all_decorated_host_handler_classes_have_explicit_factories() -> None:
    """宿主中使用事件装饰器的类必须全部由组合根 resolver 白名单接管。"""
    factories = get_host_event_handler_factories()

    assert {owner.__name__ for owner in factories} == {
        "Command",
        "DownloadChain",
        "Scheduler",
        "ScrapingChain",
        "SearchChain",
        "SiteChain",
        "SubscribeChain",
        "WorkflowChain",
    }


@pytest.mark.parametrize(
    ("backend_type", "redis_owners"),
    [
        ("redis", {"AsyncRedisHelper", "RedisHelper"}),
        ("cachetools", set()),
    ],
)
def test_all_unmanaged_config_reload_classes_have_explicit_providers(
    monkeypatch,
    compose_cache_backend,
    backend_type,
    redis_owners,
) -> None:
    """专属 resolver 未覆盖的配置 owner 必须全部声明生命周期 Provider。

    Redis 连接 owner 只在启动时选用 Redis 缓存后才有 Provider。
    """
    compose_cache_backend(backend_type)
    manager = object.__new__(PluginManager)
    plugin_singleton_key = (PluginManager, (), frozenset())
    monkeypatch.setitem(Singleton._instances, plugin_singleton_key, manager)
    doh_helper = object.__new__(
        __import__("app.adapters.network.doh", fromlist=["DohHelper"]).DohHelper
    )
    doh_singleton_key = (type(doh_helper), (), frozenset())
    monkeypatch.setitem(Singleton._instances, doh_singleton_key, doh_helper)
    providers = get_config_reload_handler_providers()

    assert {owner.__name__ for owner in providers} == {
        "DohHelper",
        "Monitor",
        "PluginManager",
        "SystemHelper",
        "TransferChain",
    } | redis_owners


def test_error_alert_dedup_preserves_distinct_events_handlers_and_errors():
    """只合并同一持久事件同一处理器的相同错误，不压制新故障或普通事件。"""
    notify, emit = Mock(), Mock()
    policy = EventErrorPolicy(notifier=lambda: notify, emit_system_error=emit)
    for event_key, handler, error in [
        ("one", "handle", "broken"), ("one", "handle", "broken"),
        ("two", "handle", "broken"), ("one", "other", "broken"),
        ("one", "handle", "new error"), (None, "handle", "broken"),
        (None, "handle", "broken"),
    ]:
        policy.handle(event=Event(EventType.TransferFailed, {"idempotency_key": event_key}),
                      module_name="plugin", class_name="Handler", method_name=handler,
                      error=RuntimeError(error))
    assert notify.call_count == 6
    assert emit.call_count == 6


def test_durable_error_alert_cache_is_bounded(monkeypatch):
    """长时间运行只保留最近事件的提示记录，淘汰后允许重新提示。"""
    from app.runtime.event import errors

    monkeypatch.setattr(errors, "_MAX_REPORTED_ERRORS", 2)
    notify = Mock()
    policy = EventErrorPolicy(notifier=lambda: notify, emit_system_error=Mock())
    for key in ("one", "two", "one", "three", "two"):
        policy.handle(event=Event(EventType.TransferFailed, {"idempotency_key": key}),
                      module_name="plugin", class_name="Handler", method_name="handle",
                      error=RuntimeError("broken"))
    assert notify.call_count == 4
    assert len(policy._reported_errors) == 2