from unittest.mock import Mock, patch

from app.chain import ChainBase
from app.core.context import MediaInfo
from app.core.meta import MetaBase
from app.schemas.types import MediaType


def _chain_without_init() -> ChainBase:
    """构造不加载真实模块和外部服务的识别链实例。"""
    return object.__new__(ChainBase)


def test_generic_source_id_wins_over_legacy_ids() -> None:
    """显式source与media_id应优先于同一请求残留的兼容ID字段。"""
    resolved = ChainBase._resolve_media_source_params(
        source="anilist",
        mediaid="154587",
        tmdbid=999,
        doubanid="888",
    )

    assert resolved == ("anilist", None, None, None, 154587)


def test_custom_plugin_source_is_not_discarded() -> None:
    """插件自定义来源应保留来源名，并通过通用原生 ID 交给插件。"""
    resolved = ChainBase._resolve_media_source_params(
        source="plugin_source",
        mediaid="custom-1",
    )

    assert resolved == ("plugin_source", None, None, None, None)


def test_explicit_source_recognition_reaches_plugins_with_all_ids() -> None:
    """显式选择数据源时应保留通用参数并进入完整模块调度。"""
    chain = _chain_without_init()
    media = MediaInfo(
        anilist_info={
            "id": 154587,
            "title": {"english": "Frieren"},
            "format": "TV",
        }
    )
    chain.run_module = Mock(return_value=media)

    with patch(
        "app.chain.MoviePilotServerHelper.report_recognize_share",
        return_value=False,
    ):
        result = chain.recognize_media(
            source="anilist",
            mediaid="154587",
            tmdbid=999,
            mtype=MediaType.TV,
        )

    assert result is media
    call = chain.run_module.call_args
    assert call.kwargs["source"] == "anilist"
    assert call.kwargs["mediaid"] == "154587"
    assert call.kwargs["anilistid"] == 154587
    assert call.kwargs["tmdbid"] is None


def test_default_recognition_passes_empty_generic_identity() -> None:
    """默认识别也应向插件传递完整但为空的通用媒体身份参数。"""
    chain = _chain_without_init()
    media = MediaInfo(title="测试电影", type=MediaType.MOVIE, tmdb_id=1)
    chain.run_module = Mock(return_value=media)
    meta = MetaBase("测试电影")
    meta.cn_name = "测试电影"
    meta.type = MediaType.MOVIE

    with patch(
        "app.chain.MoviePilotServerHelper.report_recognize_share",
        return_value=False,
    ):
        result = chain.recognize_media(meta=meta)

    assert result is media
    call = chain.run_module.call_args
    assert call.kwargs["source"] is None
    assert call.kwargs["mediaid"] is None
    assert call.kwargs["anilistid"] is None


def test_module_dispatch_always_reaches_plugins() -> None:
    """模块调度必须始终先执行插件模块。"""
    chain = _chain_without_init()
    chain._ChainBase__execute_plugin_modules = Mock(return_value="plugin")
    chain._ChainBase__execute_system_modules = Mock(return_value="system")

    result = chain.run_module("search_medias", meta=MetaBase("test"))

    assert result == "plugin"
    chain._ChainBase__execute_plugin_modules.assert_called_once()
    chain._ChainBase__execute_system_modules.assert_not_called()


def test_explicit_search_source_reaches_plugins() -> None:
    """请求级搜索来源应进入包含插件的完整模块调度。"""
    chain = _chain_without_init()
    chain.run_module = Mock(return_value=[])
    meta = MetaBase("Frieren")

    result = chain.search_medias(meta, source="anilist")

    assert result == []
    chain.run_module.assert_called_once_with(
        "search_medias",
        meta=meta,
        source="anilist",
    )