MoviePilot/tests/test_delete_transfer_history_tool.py

import asyncio
import threading
from types import SimpleNamespace

from app.agent.prompt.transfer import build_manual_redo_template_context
from app.agent.tools.impl.delete_transfer_history import DeleteTransferHistoryTool


def _tool(history_repository, execution_repository=None):
    """构造显式注入整理仓储的工具。"""
    return DeleteTransferHistoryTool(
        session_id="redo-session",
        user_id="10001",
        data=SimpleNamespace(
            transfer_history=history_repository,
            transfer_execution=execution_repository or SimpleNamespace(),
        ),
    )


def test_delete_transfer_history_tool_removes_old_dest_file_before_history(monkeypatch):
    """AI 重新整理删除整理记录前，应按历史目标文件清理旧媒体库文件。"""
    calls = []
    history = SimpleNamespace(
        id=7,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="link",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            calls.append(("get", history_id))
            return history

        async def async_delete(self, history_id):
            calls.append(("delete_history", history_id))

    class FakeStorageChain:
        def exists(self, fileitem):
            calls.append(("exists_dest", fileitem.path))
            return True

        def delete_media_file(self, fileitem):
            calls.append(("delete_dest", fileitem.path))
            return True

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    tool = _tool(FakeTransferHistoryOper())
    result = asyncio.run(tool.run(history_id=7))

    assert "已删除整理历史记录" in result
    assert calls == [
        ("get", 7),
        ("exists_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_history", 7),
    ]


def test_delete_transfer_history_tool_keeps_history_when_old_dest_delete_fails(monkeypatch):
    """旧媒体库文件删除失败时不得删除整理记录，避免重整链路丢失回滚依据。"""
    calls = []
    history = SimpleNamespace(
        id=8,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="copy",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            calls.append(("get", history_id))
            return history

        async def async_delete(self, history_id):
            calls.append(("delete_history", history_id))

    class FakeStorageChain:
        def exists(self, fileitem):
            calls.append(("exists_dest", fileitem.path))
            return True

        def delete_media_file(self, fileitem):
            calls.append(("delete_dest", fileitem.path))
            return False

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    tool = _tool(FakeTransferHistoryOper())
    result = asyncio.run(tool.run(history_id=8))

    assert "旧媒体库文件删除失败" in result
    assert calls == [
        ("get", 8),
        ("exists_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
    ]


def test_delete_transfer_history_tool_deletes_history_when_old_dest_is_missing(monkeypatch):
    """旧媒体库文件已不存在时应视为已清理，继续删除整理记录。"""
    calls = []
    history = SimpleNamespace(
        id=13,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="link",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            calls.append(("get", history_id))
            return history

        async def async_delete(self, history_id):
            calls.append(("delete_history", history_id))

    class FakeStorageChain:
        def exists(self, fileitem):
            calls.append(("exists_dest", fileitem.path))
            return False

        def delete_media_file(self, fileitem):
            calls.append(("delete_dest", fileitem.path))
            return False

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    tool = _tool(FakeTransferHistoryOper())
    result = asyncio.run(tool.run(history_id=13))

    assert "已删除整理历史记录" in result
    assert "已删除旧媒体库文件" not in result
    assert calls == [
        ("get", 13),
        ("exists_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_history", 13),
    ]


def test_delete_transfer_history_tool_keeps_successful_move_dest_as_reorganize_source(monkeypatch):
    """成功 move 记录的目标文件是重新整理输入，不应在删除历史时先删除。"""
    calls = []
    history = SimpleNamespace(
        id=9,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="move",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            calls.append(("get", history_id))
            return history

        async def async_delete(self, history_id):
            calls.append(("delete_history", history_id))

    class FakeStorageChain:
        def exists(self, fileitem):
            calls.append(("exists_dest", fileitem.path))
            return True

        def delete_media_file(self, fileitem):
            calls.append(("delete_dest", fileitem.path))
            return True

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    tool = _tool(FakeTransferHistoryOper())
    result = asyncio.run(tool.run(history_id=9))

    assert "已删除整理历史记录" in result
    assert calls == [
        ("get", 9),
        ("delete_history", 9),
    ]


def test_delete_transfer_history_tool_only_treats_exact_move_as_reorganize_source(monkeypatch):
    """整理方式必须精确等于 move，其他模式仍应清理旧目标文件。"""
    calls = []
    history = SimpleNamespace(
        id=11,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="not-move",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            calls.append(("get", history_id))
            return history

        async def async_delete(self, history_id):
            calls.append(("delete_history", history_id))

    class FakeStorageChain:
        def exists(self, fileitem):
            calls.append(("exists_dest", fileitem.path))
            return True

        def delete_media_file(self, fileitem):
            calls.append(("delete_dest", fileitem.path))
            return True

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    tool = _tool(FakeTransferHistoryOper())
    result = asyncio.run(tool.run(history_id=11))

    assert "已删除旧媒体库文件" in result
    assert calls == [
        ("get", 11),
        ("exists_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_dest", "/library/奔跑吧 (2014)/Keep.Running.mkv"),
        ("delete_history", 11),
    ]


def test_delete_transfer_history_storage_work_runs_outside_event_loop(monkeypatch):
    """整理历史的本地存储操作应在 storage worker 中执行。"""
    caller_thread = threading.get_ident()
    storage_threads = []
    history = SimpleNamespace(
        id=15,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="copy",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            return history

        async def async_delete(self, history_id):
            return None

    class FakeStorageChain:
        def exists(self, fileitem):
            storage_threads.append(threading.get_ident())
            return True

        def delete_media_file(self, fileitem):
            storage_threads.append(threading.get_ident())
            return True

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    result = asyncio.run(
        _tool(FakeTransferHistoryOper()).run(history_id=15)
    )

    assert "已删除整理历史记录" in result
    assert storage_threads
    assert all(thread_id != caller_thread for thread_id in storage_threads)


def test_delete_transfer_history_cancellation_keeps_history_record(monkeypatch):
    """取消等待存储清理时不得继续提交整理历史删除。"""
    started = threading.Event()
    release = threading.Event()
    finished = threading.Event()
    history = SimpleNamespace(
        id=16,
        title="奔跑吧",
        src="/downloads/Keep.Running.mkv",
        status=True,
        mode="copy",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
    )
    delete_history_calls = []

    class FakeTransferHistoryOper:
        async def async_get(self, history_id):
            return history

        async def async_delete(self, history_id):
            delete_history_calls.append(history_id)

    class FakeStorageChain:
        def exists(self, fileitem):
            started.set()
            release.wait(timeout=1)
            return True

        def delete_media_file(self, fileitem):
            finished.set()
            return True

    monkeypatch.setattr(
        "app.agent.tools.impl.delete_transfer_history.StorageChain",
        FakeStorageChain,
    )

    async def scenario():
        task = asyncio.create_task(
            _tool(FakeTransferHistoryOper()).run(history_id=16)
        )
        assert await asyncio.to_thread(started.wait, 1)
        task.cancel()
        try:
            await task
        except asyncio.CancelledError:
            pass

    try:
        asyncio.run(scenario())
    finally:
        release.set()

    assert delete_history_calls == []
    assert finished.wait(timeout=1)


def test_manual_redo_context_uses_dest_path_for_successful_move_record():
    """成功 move 记录重新整理时，旧目标文件才是可继续整理的输入路径。"""
    history = SimpleNamespace(
        id=10,
        status=True,
        title="奔跑吧",
        type="电视剧",
        category="综艺",
        year="2014",
        seasons="S01",
        episodes="E01",
        src="/downloads/Keep.Running.mkv",
        src_storage="local",
        src_fileitem={
            "storage": "local",
            "path": "/downloads/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
        dest="/library/奔跑吧 (2014)/Keep.Running.mkv",
        dest_storage="local",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
        mode="move",
        tmdbid=100,
        doubanid=None,
        bangumiid=None,
        anilistid=None,
        media_source="themoviedb",
        media_id="100",
        errmsg=None,
    )

    context = build_manual_redo_template_context(history)

    assert context["source_path"] == "/library/奔跑吧 (2014)/Keep.Running.mkv"
    assert context["source_storage"] == "local"


def test_manual_redo_context_only_treats_exact_move_as_dest_source():
    """非 move 整理方式即使名称包含 move，也应继续使用原始来源。"""
    history = SimpleNamespace(
        id=12,
        status=True,
        title="奔跑吧",
        type="电视剧",
        category="综艺",
        year="2014",
        seasons="S01",
        episodes="E01",
        src="/downloads/Keep.Running.mkv",
        src_storage="local",
        src_fileitem={
            "storage": "local",
            "path": "/downloads/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
        dest="/library/奔跑吧 (2014)/Keep.Running.mkv",
        dest_storage="local",
        dest_fileitem={
            "storage": "local",
            "path": "/library/奔跑吧 (2014)/Keep.Running.mkv",
            "name": "Keep.Running.mkv",
            "type": "file",
        },
        mode="not-move",
        tmdbid=100,
        doubanid=None,
        bangumiid=None,
        anilistid=None,
        media_source="themoviedb",
        media_id="100",
        errmsg=None,
    )

    context = build_manual_redo_template_context(history)

    assert context["source_path"] == "/downloads/Keep.Running.mkv"
    assert context["source_storage"] == "local"