# -*- coding: utf-8 -*-
"""订阅自定义识别词快照用例：下载时保存完整识别词，整理时快照优先、实时反查兜底。

回归场景：订阅做季+集组合偏移（如 S04E05→S01E71），下载阶段生效但整理阶段因实时反查订阅
返回空而静默回退全局识别词、丢失偏移。修复后由订阅链在发起下载时将完整识别词作为入参传入
下载模块并存档（避免下载模块反查订阅的同级循环依赖），整理时优先复用该快照。
"""
from types import SimpleNamespace

import app.chain.transfer as transfer_module
from app.chain.transfer import TransferChain


def _fake_history(custom_words=None, note=None):
    """构造仅含测试所需字段的下载历史替身。"""
    return SimpleNamespace(custom_words=custom_words, note=note)


def test_transfer_prefers_snapshot_over_live_lookup(monkeypatch):
    """整理时存在下载快照，应直接使用快照且不触发实时反查订阅。"""
    called = {"lookup": False}

    class _GuardSubscribeChain:
        def get_subscribe_by_source(self, source):
            # 一旦走到实时反查即视为失败：快照存在时不应触发
            called["lookup"] = True
            return SimpleNamespace(custom_words="不应使用\n实时反查")

    monkeypatch.setattr(transfer_module, "SubscribeChain", _GuardSubscribeChain)

    history = _fake_history(
        custom_words="S04 => S01\n第 <> 集 >> EP+66",
        note={"source": "Subscribe|{...}"},
    )
    result = TransferChain._get_subscribe_custom_words(history)

    assert result == ["S04 => S01", "第 <> 集 >> EP+66"]
    assert called["lookup"] is False


def test_transfer_falls_back_to_live_lookup_without_snapshot(monkeypatch):
    """整理时无快照（历史旧记录），应按下载来源实时反查订阅取识别词。"""

    class _FakeSubscribeChain:
        def get_subscribe_by_source(self, source):
            assert source == "Subscribe|{...}"
            return SimpleNamespace(custom_words="A => B")

    monkeypatch.setattr(transfer_module, "SubscribeChain", _FakeSubscribeChain)

    history = _fake_history(custom_words=None, note={"source": "Subscribe|{...}"})
    result = TransferChain._get_subscribe_custom_words(history)

    assert result == ["A => B"]


def test_transfer_returns_none_when_unavailable(monkeypatch):
    """无下载记录、note 非字典、或来源反查不到订阅时返回 None（回退全局识别词）。"""

    class _NoneSubscribeChain:
        def get_subscribe_by_source(self, source):
            return None

    monkeypatch.setattr(transfer_module, "SubscribeChain", _NoneSubscribeChain)

    # 无下载记录
    assert TransferChain._get_subscribe_custom_words(None) is None
    # 无快照且 note 非字典：不应触发实时反查
    assert TransferChain._get_subscribe_custom_words(_fake_history(note="不是字典")) is None
    # 无快照、来源可解析但反查不到订阅
    assert (
        TransferChain._get_subscribe_custom_words(
            _fake_history(note={"source": "Subscribe|{}"})
        )
        is None
    )