MoviePilot/tests/test_storage_strict_query_coverage.py

import importlib
from pathlib import Path
from types import SimpleNamespace

import pytest

from app.modules.filemanager.storages import StorageBase
from app.schemas.exception import StorageQueryError

# 在树存储模块，全部必须实现严格查询
BUILTIN_STORAGE_MODULES = ["alipan", "alist", "local", "rclone", "smb", "u115"]


def test_base_get_item_strict_fails_conservatively():
    """未覆写严格查询的存储必须保守失败，而不是回退到宽松查询。"""
    stub = SimpleNamespace(schema="dummy")

    with pytest.raises(StorageQueryError):
        StorageBase.get_item_strict(stub, Path("/media/示例.mkv"))


def test_base_list_strict_fails_conservatively():
    """未适配严格目录查询的存储不能把宽松查询的空结果当作完整快照。"""
    stub = SimpleNamespace(schema="dummy")

    with pytest.raises(StorageQueryError):
        StorageBase.list_strict(stub, SimpleNamespace(path="/media"))


@pytest.mark.parametrize("module_name", BUILTIN_STORAGE_MODULES)
def test_builtin_storage_overrides_strict_query(module_name):
    """每个在树存储都必须覆写严格查询，否则整理会被基类保守拒绝。"""
    module = importlib.import_module(f"app.modules.filemanager.storages.{module_name}")
    storage_classes = [
        obj for obj in vars(module).values()
        if isinstance(obj, type)
        and issubclass(obj, StorageBase)
        and obj is not StorageBase
        and obj.__module__ == module.__name__
    ]

    assert storage_classes, f"{module_name} 未定义存储类"
    for storage_class in storage_classes:
        assert "get_item_strict" in vars(storage_class), \
            f"{storage_class.__name__} 未实现 get_item_strict"