MoviePilot/tests/test_transfer_pending_replay.py

"""
整理队列持久化与重启回放测试。

整理队列是纯内存的 queue.Queue：挂载挂死后的人工重启、版本升级、OOM、宿主
重启都会让队列连同「这些文件还没整理」这个事实一起蒸发，而已稳定落地的文件
不会再产生任何监控事件，也不会有新的补偿扫描起点——结果就是永久漏件。

这些测试固定三项不变量：入队即落盘登记、终态即注销、重启能回放。
"""
import threading
from pathlib import Path
from unittest.mock import MagicMock

from app.application.transfer import TransferAdmission, TransferTask
from app.chain.transfer import TransferChain
from app.schemas.file import FileItem


def _build_chain(admissions) -> TransferChain:
    """
    构造绕过单例初始化的 TransferChain 骨架。
    :param admissions: durable admission 仓储替身
    :return: TransferChain 骨架
    """
    chain = object.__new__(TransferChain)
    chain._transfer_admissions = admissions
    return chain


def _admission(path: str, task_id: str = "task-1") -> TransferAdmission:
    """构造一条可脱离数据库会话使用的准入快照。"""
    return TransferAdmission(
        task_id=task_id,
        storage="local",
        src_path=path,
        state="accepted",
        created_at="2026-08-27 10:00:00",
        updated_at="2026-08-27 10:00:00",
    )


def _task(path: str, storage: str = "local") -> TransferTask:
    """
    构造测试用整理任务。
    :param path: 源文件路径
    :param storage: 存储
    :return: 整理任务
    """
    file_path = Path(path)
    return TransferTask(fileitem=FileItem(
        storage=storage,
        path=path,
        type="file",
        name=file_path.name,
        basename=file_path.stem,
        extension=file_path.suffix[1:],
    ))


def test_admit_transfer_records_storage_and_path():
    """
    入队时必须落盘登记「存储 + 源路径」这一最小事实。
    """
    admissions = MagicMock()
    admissions.admit.return_value = _admission(
        "/mnt/cd2/downloads/Movie.2024.mkv"
    )
    chain = _build_chain(admissions)

    result = chain._TransferChain__admit_transfer(
        _task("/mnt/cd2/downloads/Movie.2024.mkv")
    )

    admissions.admit.assert_called_once_with(
        storage="local", src_path="/mnt/cd2/downloads/Movie.2024.mkv"
    )
    assert result.task_id == "task-1"


def test_discard_pending_on_terminal_state():
    """
    整理到达终态后必须注销登记，否则每次重启都会重复回放。
    """
    admissions = MagicMock()
    chain = _build_chain(admissions)
    task = _task("/mnt/cd2/downloads/Movie.2024.mkv")
    task.bind_admission_task_id("task-1")

    chain._TransferChain__discard_pending(task)

    admissions.discard_task.assert_called_once_with(task_id="task-1")


def test_replay_resends_pending_files_to_transfer(tmp_path, monkeypatch):
    """
    重启回放：登记过的文件要重新送入整理链，恢复被内存队列蒸发的任务。
    """
    media = tmp_path / "Movie.2024.mkv"
    media.write_bytes(b"x" * 10)

    admissions = MagicMock()
    admissions.list_accepted.return_value = [_admission(str(media))]
    chain = _build_chain(admissions)

    transferred = []
    monkeypatch.setattr(chain, "do_transfer", lambda **kw: transferred.append(kw["fileitem"]))

    chain._TransferChain__replay_pending()

    assert len(transferred) == 1
    item = transferred[0]
    assert item.path == media.as_posix()
    assert item.storage == "local"
    assert item.type == "file"
    # 回放时重新读取当前大小，不依赖登记时的陈旧信息
    assert item.size == 10


def test_replay_discards_vanished_files(tmp_path):
    """
    源文件已消失的登记要注销，否则每次启动都会重复回放一个不存在的文件。
    """
    admissions = MagicMock()
    missing = tmp_path / "gone.mkv"
    admissions.list_accepted.return_value = [_admission(str(missing))]
    chain = _build_chain(admissions)
    chain.do_transfer = MagicMock()

    chain._TransferChain__replay_pending()

    chain.do_transfer.assert_not_called()
    admissions.discard_task.assert_called_once_with(task_id="task-1")


def test_replay_keeps_registration_when_mount_unreadable(tmp_path, monkeypatch):
    """
    挂载未就绪时读取失败属于暂时性故障，登记必须保留，等下次启动或人工整理。

    这与「文件已消失」必须区别对待：把挂载抖动误判成文件消失就等于主动丢件。
    """
    media = tmp_path / "Movie.2024.mkv"
    media.write_bytes(b"x")

    admissions = MagicMock()
    admissions.list_accepted.return_value = [_admission(str(media))]
    chain = _build_chain(admissions)
    chain.do_transfer = MagicMock()

    def unreadable(self, *_args, **_kwargs):
        """
        模拟挂载未就绪时的 stat 失败。
        """
        raise OSError(107, "Transport endpoint is not connected")

    monkeypatch.setattr(Path, "stat", unreadable)

    chain._TransferChain__replay_pending()

    chain.do_transfer.assert_not_called()
    admissions.discard_task.assert_not_called()


def test_replay_restores_bluray_directory_type(tmp_path, monkeypatch):
    """
    蓝光原盘登记时保留尾部斜杠，回放必须还原成目录类型，否则会被当成单文件整理。
    """
    bluray = tmp_path / "Movie.2024.BluRay"
    bluray.mkdir()
    src_path = f"{bluray.as_posix()}/"

    admissions = MagicMock()
    admissions.list_accepted.return_value = [_admission(src_path)]
    chain = _build_chain(admissions)

    transferred = []
    monkeypatch.setattr(chain, "do_transfer", lambda **kw: transferred.append(kw["fileitem"]))

    chain._TransferChain__replay_pending()

    assert len(transferred) == 1
    assert transferred[0].type == "dir"
    assert transferred[0].path == src_path


def test_replay_is_noop_without_registrations():
    """
    没有登记时回放不应触碰整理链。
    """
    admissions = MagicMock()
    admissions.list_accepted.return_value = []
    chain = _build_chain(admissions)
    chain.do_transfer = MagicMock()

    chain._TransferChain__replay_pending()

    chain.do_transfer.assert_not_called()


def test_replay_survives_db_failure():
    """
    读取登记失败不能让启动流程报错。
    """
    admissions = MagicMock()
    admissions.list_accepted.side_effect = RuntimeError("db gone")
    chain = _build_chain(admissions)
    chain.do_transfer = MagicMock()

    chain._TransferChain__replay_pending()

    chain.do_transfer.assert_not_called()


def test_replay_continues_after_single_file_failure(tmp_path, monkeypatch):
    """
    单个文件回放失败不能中断整批回放，否则一个坏文件会拖住所有漏件的恢复。
    """
    first = tmp_path / "A.mkv"
    second = tmp_path / "B.mkv"
    for item in (first, second):
        item.write_bytes(b"x")

    admissions = MagicMock()
    admissions.list_accepted.return_value = [
        _admission(str(first), "task-1"),
        _admission(str(second), "task-2"),
    ]
    chain = _build_chain(admissions)

    handled = []

    def flaky(**kw):
        """
        第一个文件整理抛异常，第二个正常。
        """
        if kw["fileitem"].name == "A.mkv":
            raise RuntimeError("boom")
        handled.append(kw["fileitem"].name)

    monkeypatch.setattr(chain, "do_transfer", flaky)

    chain._TransferChain__replay_pending()

    assert handled == ["B.mkv"]


def test_replay_stop_keeps_unprocessed_registrations(tmp_path, monkeypatch):
    """
    宿主关闭后不得继续检查或注销下一条登记，未处理项留给下次启动回放。
    """
    first = tmp_path / "A.mkv"
    first.write_bytes(b"x")
    missing_second = tmp_path / "gone.mkv"
    admissions = MagicMock()
    admissions.list_accepted.return_value = [
        _admission(str(first), "task-1"),
        _admission(str(missing_second), "task-2"),
    ]
    chain = _build_chain(admissions)
    stop_event = threading.Event()
    transferred = []

    def transfer_first(**kwargs):
        """首条回放送入整理链后模拟宿主发出关闭信号。"""
        transferred.append(kwargs["fileitem"].path)
        stop_event.set()

    monkeypatch.setattr(chain, "do_transfer", transfer_first)

    chain._TransferChain__replay_pending(stop_event)

    assert transferred == [first.as_posix()]
    admissions.discard_task.assert_not_called()