MoviePilot/tests/test_local_setup_uninstall.py

"""
验证本地卸载的确认流程与配置数据保留边界。
"""

from __future__ import annotations

import importlib.util
import uuid
from contextlib import ExitStack
from pathlib import Path
from unittest.mock import patch

import pytest

MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "local_setup.py"


def load_local_setup_module():
    """
    按源码路径加载隔离的本地安装脚本模块。
    """
    module_name = f"moviepilot_local_setup_{uuid.uuid4().hex}"
    spec = importlib.util.spec_from_file_location(module_name, MODULE_PATH)
    module = importlib.util.module_from_spec(spec)
    assert spec and spec.loader
    spec.loader.exec_module(module)
    return module


@pytest.fixture
def prepare_install_tree(tmp_path: Path, request):
    """
    创建隔离的本地安装目录，并在测试结束时撤销脚本路径配置替换。

    :return: 接受 legacy_config 的构造函数，用于选择旧版或独立配置目录
    """

    def build(*, legacy_config: bool = False):
        """
        填充安装文件并返回脚本模块、根目录、配置目录、虚拟环境与环境文件。
        """
        module = load_local_setup_module()
        temp_path = tmp_path
        root_dir = temp_path / "MoviePilot"
        resource_dir = root_dir / "app" / "infrastructure"
        runtime_dir = root_dir / ".runtime"
        public_dir = root_dir / "public"
        venv_dir = root_dir / "venv"
        install_env_file = root_dir / ".moviepilot.env"
        config_dir = root_dir / "config" if legacy_config else temp_path / "moviepilot-config"
        temp_config_dir = config_dir / "temp"

        resource_dir.mkdir(parents=True)
        runtime_dir.mkdir(parents=True)
        public_dir.mkdir(parents=True)
        venv_dir.mkdir(parents=True)
        temp_config_dir.mkdir(parents=True)
        install_env_file.write_text("CONFIG_DIR=/tmp/moviepilot-config\n", encoding="utf-8")
        (root_dir / "moviepilot").write_text("#!/usr/bin/env bash\n", encoding="utf-8")
        (resource_dir / "sites.py").write_text("generated\n", encoding="utf-8")
        (resource_dir / "user.sites.v3.bin").write_bytes(b"binary")
        (temp_config_dir / "moviepilot.runtime.json").write_text("{}", encoding="utf-8")
        (temp_config_dir / "moviepilot.frontend.runtime.json").write_text("{}", encoding="utf-8")

        stack = ExitStack()
        request.addfinalizer(stack.close)
        stack.enter_context(patch.object(module, "ROOT", root_dir))
        stack.enter_context(patch.object(module, "SITE_RESOURCE_DIR", resource_dir))
        stack.enter_context(patch.object(module, "RUNTIME_DIR", runtime_dir))
        stack.enter_context(patch.object(module, "PUBLIC_DIR", public_dir))
        stack.enter_context(patch.object(module, "INSTALL_ENV_FILE", install_env_file))
        stack.enter_context(patch.object(module, "LEGACY_CONFIG_DIR", root_dir / "config"))
        stack.enter_context(patch.object(module, "CONFIG_DIR", config_dir))
        stack.enter_context(patch.object(module, "TEMP_DIR", temp_config_dir))

        return module, root_dir, config_dir, venv_dir, install_env_file

    return build


def test_remove_config_data_deletes_legacy_config_directory(prepare_install_tree) -> None:
    """
    清理配置数据时删除旧版项目内配置目录及其内容。
    """
    module, _, config_dir, _, _ = prepare_install_tree(legacy_config=True)
    category_file = config_dir / "category.yaml"
    category_file.write_text("seed\n", encoding="utf-8")
    (config_dir / "logs").mkdir(exist_ok=True)
    (config_dir / "user.db").write_text("db\n", encoding="utf-8")

    removed = module._remove_config_data(config_dir)

    assert not config_dir.exists()
    assert not category_file.exists()
    assert str(config_dir.resolve()) in {str(path.resolve()) for path in removed}


def test_uninstall_keeps_config_by_default(prepare_install_tree) -> None:
    """
    默认卸载仅删除程序文件并保留配置数据。
    """
    module, root_dir, config_dir, venv_dir, install_env_file = prepare_install_tree()
    cli_dir = root_dir.parent / "bin"
    cli_dir.mkdir()
    cli_link = cli_dir / "moviepilot"
    cli_link.symlink_to(root_dir / "moviepilot")

    yes_no_answers = iter([False, True])
    with (
        patch.object(module, "_is_interactive", return_value=True),
        patch.object(
            module,
            "_prompt_yes_no",
            side_effect=lambda label, default=False: next(yes_no_answers),
        ),
        patch.object(
            module,
            "_prompt_text",
            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
        ),
        patch.object(module, "_stop_managed_services", return_value=None),
    ):
        result = module.uninstall_local(
            venv_dir=venv_dir,
            config_dir=config_dir,
            launch_path=str(cli_link),
        )

    assert not result["cancelled"]
    assert config_dir.exists()
    assert install_env_file.exists()
    assert not venv_dir.exists()
    assert not (root_dir / ".runtime").exists()
    assert not (root_dir / "public").exists()
    assert not (root_dir / "app" / "infrastructure" / "sites.py").exists()
    assert not (root_dir / "app" / "infrastructure" / "user.sites.v3.bin").exists()
    assert not cli_link.exists()


def test_uninstall_deletes_external_config_when_requested(prepare_install_tree) -> None:
    """
    明确要求删除数据时卸载独立配置目录。
    """
    module, _, config_dir, venv_dir, install_env_file = prepare_install_tree()
    yes_no_answers = iter([True, True])

    with (
        patch.object(module, "_is_interactive", return_value=True),
        patch.object(
            module,
            "_prompt_yes_no",
            side_effect=lambda label, default=False: next(yes_no_answers),
        ),
        patch.object(
            module,
            "_prompt_text",
            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
        ),
        patch.object(module, "_stop_managed_services", return_value=None),
    ):
        result = module.uninstall_local(
            venv_dir=venv_dir,
            config_dir=config_dir,
        )

    assert not result["cancelled"]
    assert result["config_deleted"]
    assert not config_dir.exists()
    assert not install_env_file.exists()


def test_uninstall_deletes_legacy_config_when_requested(prepare_install_tree) -> None:
    """
    明确要求删除数据时卸载旧版项目内配置目录。
    """
    module, _, config_dir, venv_dir, install_env_file = prepare_install_tree(legacy_config=True)
    (config_dir / "category.yaml").write_text("seed\n", encoding="utf-8")
    yes_no_answers = iter([True, True])

    with (
        patch.object(module, "_is_interactive", return_value=True),
        patch.object(
            module,
            "_prompt_yes_no",
            side_effect=lambda label, default=False: next(yes_no_answers),
        ),
        patch.object(
            module,
            "_prompt_text",
            side_effect=lambda label, default=None, allow_empty=False, secret=False: module.UNINSTALL_CONFIRM_TEXT,
        ),
        patch.object(module, "_stop_managed_services", return_value=None),
    ):
        result = module.uninstall_local(
            venv_dir=venv_dir,
            config_dir=config_dir,
        )

    assert not result["cancelled"]
    assert result["config_deleted"]
    assert not config_dir.exists()
    assert not install_env_file.exists()