MoviePilot/tests/test_transfer_overwrite_guard.py

from pathlib import Path
from threading import Lock
from unittest.mock import MagicMock

import pytest

from app.modules.filemanager.storages import StorageBase
from app.modules.filemanager.storages.alipan import AliPan
from app.modules.filemanager.storages.local import LocalStorage
from app.modules.filemanager.storages.rclone import Rclone
from app.modules.filemanager.storages.u115 import U115Pan
from app.schemas.exception import StorageQueryError


def _local() -> LocalStorage:
    """
    构造本地存储实例（跳过初始化）。
    """
    return object.__new__(LocalStorage)


def _u115() -> U115Pan:
    """
    构造 115 存储实例（跳过初始化）。
    """
    return object.__new__(U115Pan)


def _u115_with_api_payload(payload: dict) -> U115Pan:
    """
    构造返回固定业务响应的 115 存储实例。
    """
    storage = _u115()
    response = MagicMock(status_code=200)
    response.json.return_value = payload
    storage.session = MagicMock()
    storage.session.request.return_value = response
    storage._check_session = MagicMock()
    storage._download_limiter = MagicMock()
    storage._api_limiter = MagicMock()
    storage._rate_stats = MagicMock()
    storage._limit_lock = Lock()
    storage._limit_until = 0.0
    return storage


def _alipan(monkeypatch) -> AliPan:
    """
    构造阿里云盘存储实例（跳过初始化，_default_drive_id 为只读属性需在类级替换）。
    """
    monkeypatch.setattr(AliPan, "_default_drive_id", "drive-1", raising=False)
    return object.__new__(AliPan)


def test_local_strict_missing_file_returns_none(tmp_path):
    """
    目标文件确实不存在时应确认为不存在，允许正常整理。
    """
    assert _local().get_item_strict(tmp_path / "missing.mkv") is None


def test_local_strict_existing_file_returns_item(tmp_path):
    """
    目标文件存在时应返回文件项。
    """
    target = tmp_path / "movie.mkv"
    target.write_bytes(b"movie")

    item = _local().get_item_strict(target)

    assert item is not None
    assert item.path == target.as_posix()


def test_local_strict_broken_symlink_returns_none(tmp_path):
    """
    失效软链接视为目标不存在，不应阻断整理。
    """
    target = tmp_path / "movie.mkv"
    target.symlink_to(tmp_path / "gone.mkv")

    assert _local().get_item_strict(target) is None


def test_local_strict_raises_on_stat_error(tmp_path, monkeypatch):
    """
    FUSE 挂载抖动导致 stat 失败时应抛出 StorageQueryError，拒绝覆盖。
    """
    target = tmp_path / "movie.mkv"

    def raise_stat_error(*_args, **_kwargs):
        """
        模拟 CloudDrive FUSE 挂载返回 ENOTRECOVERABLE。
        """
        raise OSError(131, "State not recoverable")

    # 文件系统边界已下移到代理子进程，patch Path.stat 影响不到那里，
    # 必须在代理这一层注入故障
    monkeypatch.setattr(
        "app.modules.filemanager.storages.local.fsproxy.stat", raise_stat_error
    )

    with pytest.raises(StorageQueryError):
        _local().get_item_strict(target)


def test_u115_strict_transport_failure_raises():
    """
    115 请求失败（网络/限流重试用尽）时应抛出 StorageQueryError。
    """
    storage = _u115()
    storage._request_api = MagicMock(return_value=None)

    with pytest.raises(StorageQueryError):
        storage.get_item_strict(Path("/movie.mkv"))


def test_u115_get_item_keeps_swallowing_transport_failure():
    """
    宽松版 get_item 行为保持兼容：请求失败仍返回 None。
    """
    storage = _u115()
    storage._request_api = MagicMock(return_value=None)

    assert storage.get_item(Path("/movie.mkv")) is None


def test_u115_strict_confirmed_absent_returns_none():
    """
    115 业务码返回记录不存在（data 为空）时应确认为不存在。
    """
    storage = _u115()
    storage._request_api = MagicMock(return_value={"state": True, "code": 20004, "data": {}})

    assert storage.get_item_strict(Path("/movie.mkv")) is None


def test_u115_strict_path_not_found_returns_none():
    """
    115 查询路径返回 430004 时应确认为不存在，允许首次整理。
    """
    storage = _u115_with_api_payload(
        {"state": False, "code": 430004, "message": "路径不存在", "data": {}}
    )

    assert storage.get_item_strict(Path("/movie.mkv")) is None


def test_u115_strict_empty_list_returns_none():
    """
    115 查询不存在目标返回空列表时也应确认为不存在。
    """
    storage = _u115_with_api_payload(
        {"state": True, "code": 0, "message": "", "data": []}
    )

    assert storage.get_item_strict(Path("/movie.mkv")) is None


def test_u115_path_not_found_code_is_not_globally_accepted():
    """
    非路径查询接口返回 430004 时仍应视为业务错误。
    """
    storage = _u115_with_api_payload(
        {"state": False, "code": 430004, "message": "路径不存在", "data": {}}
    )

    result = storage._request_api("POST", "/open/folder/add", data={"file_name": "TV"})

    assert result is None


def test_u115_strict_returns_item():
    """
    115 返回有效文件数据时应构造文件项。
    """
    storage = _u115()
    storage._request_api = MagicMock(return_value={"state": True, "code": 0, "data": {
        "file_id": 123,
        "file_category": "1",
        "file_name": "movie.mkv",
        "pick_code": "abc",
        "size_byte": 1024,
        "utime": 100,
    }})

    item = storage.get_item_strict(Path("/movie.mkv"))

    assert item is not None
    assert item.fileid == "123"
    assert item.size == 1024


def test_alipan_strict_notfound_returns_none(monkeypatch):
    """
    阿里云盘 NotFound 系列错误码应确认为不存在。
    """
    storage = _alipan(monkeypatch)
    storage._request_api = MagicMock(return_value={"code": "NotFound.File", "message": "not found"})

    assert storage.get_item_strict(Path("/movie.mkv")) is None


def test_alipan_strict_other_error_raises(monkeypatch):
    """
    阿里云盘非 NotFound 的业务错误（如限流）应抛出 StorageQueryError。
    """
    storage = _alipan(monkeypatch)
    storage._request_api = MagicMock(return_value={"code": "TooManyRequests", "message": "limit"})

    with pytest.raises(StorageQueryError):
        storage.get_item_strict(Path("/movie.mkv"))


def test_alipan_strict_transport_failure_raises(monkeypatch):
    """
    阿里云盘请求失败时应抛出 StorageQueryError。
    """
    storage = _alipan(monkeypatch)
    storage._request_api = MagicMock(return_value=None)

    with pytest.raises(StorageQueryError):
        storage.get_item_strict(Path("/movie.mkv"))


def test_alipan_strict_returns_item(monkeypatch):
    """
    阿里云盘返回有效数据时应构造文件项。
    """
    storage = _alipan(monkeypatch)
    storage._request_api = MagicMock(return_value={"file_id": "f1", "name": "movie.mkv"})
    setattr(storage, "_AliPan__get_fileitem", MagicMock(return_value="ITEM"))

    assert storage.get_item_strict(Path("/movie.mkv")) == "ITEM"


def test_storage_base_strict_fails_conservatively_without_override():
    """
    未覆写严格查询的存储必须保守失败：沿用 get_item 会把「查询失败」当成
    「目标不存在」，让 overwrite_mode=size 的覆盖保护被绕过。
    """
    storage = object.__new__(Rclone)
    storage.get_item = MagicMock(return_value=None)

    with pytest.raises(StorageQueryError):
        StorageBase.get_item_strict(storage, Path("/movie.mkv"))
    storage.get_item.assert_not_called()