mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-28 19:06:36 +08:00
244 lines
7.7 KiB
Python
244 lines
7.7 KiB
Python
"""
|
|
验证 MoviePilot 启动前即可使用的安装向导行为。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import builtins
|
|
import importlib.util
|
|
import sys
|
|
import types
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "local_setup.py"
|
|
|
|
|
|
def load_local_setup_module():
|
|
"""
|
|
加载独立的安装模块，隔离向导测试中的模块状态。
|
|
"""
|
|
module_name = f"moviepilot_local_setup_wizard_{uuid.uuid4().hex}"
|
|
spec = importlib.util.spec_from_file_location(module_name, MODULE_PATH)
|
|
assert spec is not None and spec.loader is not None
|
|
module = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
def test_llm_provider_module_loads_when_repository_root_is_not_on_import_path(
|
|
monkeypatch,
|
|
):
|
|
"""
|
|
独立安装器必须让模型提供方能够导入应用模块。
|
|
"""
|
|
module = load_local_setup_module()
|
|
repository_root = str(module.ROOT)
|
|
monkeypatch.setattr(
|
|
module.sys,
|
|
"path",
|
|
[entry for entry in sys.path if entry != repository_root],
|
|
)
|
|
|
|
provider_module = module._load_llm_provider_module()
|
|
|
|
assert module.sys.path[0] == repository_root
|
|
assert hasattr(provider_module, "LLMProviderManager")
|
|
monkeypatch.delitem(module.sys.modules, provider_module.__name__)
|
|
|
|
|
|
def test_startup_python_resolution_preserves_virtualenv_symlink(
|
|
monkeypatch, tmp_path: Path
|
|
):
|
|
"""
|
|
自动启动必须调用虚拟环境启动器，不能绕过环境使用基础解释器。
|
|
"""
|
|
module = load_local_setup_module()
|
|
venv_dir = tmp_path / "venv"
|
|
python_launcher = module.get_venv_python(venv_dir)
|
|
python_launcher.parent.mkdir(parents=True)
|
|
python_launcher.symlink_to(Path(sys.executable))
|
|
expected_path = python_launcher.absolute()
|
|
monkeypatch.setattr(
|
|
module,
|
|
"_can_run_moviepilot_cli",
|
|
lambda candidate: candidate == expected_path,
|
|
)
|
|
|
|
runtime_python = module._resolve_runtime_python_for_startup(
|
|
python_launcher, venv_dir
|
|
)
|
|
|
|
assert runtime_python == expected_path
|
|
assert runtime_python.is_symlink()
|
|
|
|
|
|
def test_startup_python_resolution_prefers_project_runtime_launcher(
|
|
monkeypatch, tmp_path: Path
|
|
):
|
|
"""
|
|
自动启动优先使用项目运行时启动器，避免选择外部解释器。
|
|
"""
|
|
module = load_local_setup_module()
|
|
venv_dir = tmp_path / "venv"
|
|
runtime_launcher = module.get_venv_bin_dir(venv_dir) / "moviepilot-python"
|
|
runtime_launcher.parent.mkdir(parents=True)
|
|
runtime_launcher.symlink_to(Path(sys.executable))
|
|
expected_path = runtime_launcher.absolute()
|
|
external_python = tmp_path / "external-python"
|
|
monkeypatch.setattr(
|
|
module,
|
|
"_can_run_moviepilot_cli",
|
|
lambda candidate: candidate == expected_path,
|
|
)
|
|
|
|
runtime_python = module._resolve_runtime_python_for_startup(
|
|
external_python, venv_dir
|
|
)
|
|
|
|
assert runtime_python == expected_path
|
|
|
|
|
|
def test_directory_config_keeps_download_path_as_downloader_path(monkeypatch):
|
|
"""
|
|
向导保留下载器路径，不将其作为宿主路径解析。
|
|
"""
|
|
module = load_local_setup_module()
|
|
text_prompts: list[tuple[str, str]] = []
|
|
path_prompts: list[str] = []
|
|
monkeypatch.setattr(module, "print_step", lambda _message: None)
|
|
|
|
def prompt_text(label: str, default: str) -> str:
|
|
"""
|
|
记录下载器路径提示，并返回容器内的路径。
|
|
"""
|
|
text_prompts.append((label, default))
|
|
return "/downloads"
|
|
|
|
def prompt_path(label: str, default: Path) -> str:
|
|
"""
|
|
提供宿主侧媒体库路径。
|
|
"""
|
|
assert isinstance(default, Path)
|
|
path_prompts.append(label)
|
|
return "/video"
|
|
|
|
monkeypatch.setattr(module, "_prompt_text", prompt_text)
|
|
monkeypatch.setattr(module, "_prompt_path", prompt_path)
|
|
monkeypatch.setattr(module, "_prompt_choice", lambda *_args, **_kwargs: "copy")
|
|
|
|
directory_config = module._collect_directory_config()
|
|
|
|
assert directory_config["download_path"] == "/downloads"
|
|
assert directory_config["library_path"] == "/video"
|
|
assert text_prompts[0][0] == "下载器中的下载目录根路径"
|
|
assert path_prompts == ["媒体库目录"]
|
|
|
|
|
|
def test_auth_site_definitions_load_without_database_transaction_runners(
|
|
monkeypatch,
|
|
):
|
|
"""
|
|
安装器初始化数据库之前必须能够加载静态认证字段。
|
|
"""
|
|
from app.db import uow
|
|
|
|
module = load_local_setup_module()
|
|
monkeypatch.setattr(uow, "_sync_transaction_runner", None)
|
|
monkeypatch.setattr(uow, "_async_transaction_runner", None)
|
|
|
|
class AuthSitesHelper:
|
|
"""
|
|
通过临时工作单元执行查询，并返回静态认证结构。
|
|
"""
|
|
|
|
def get_authsites(self) -> dict[str, object]:
|
|
"""
|
|
通过临时空结果执行器读取活动站点查询。
|
|
"""
|
|
assert uow.run_sync_transaction(lambda _session: "queried") == []
|
|
return {
|
|
"example": {
|
|
"name": "示例站点",
|
|
"params": {
|
|
"cookie": {
|
|
"name": "Cookie",
|
|
"type": "text",
|
|
"placeholder": "填写 Cookie",
|
|
"tooltip": "用于站点认证",
|
|
"convert": "upper",
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
sites_module = types.ModuleType("app.application.site.sites")
|
|
sites_module.SitesHelper = AuthSitesHelper
|
|
monkeypatch.setitem(sys.modules, "app.application.site.sites", sites_module)
|
|
|
|
definitions = module._load_auth_site_definitions_inner()
|
|
|
|
assert definitions == {
|
|
"example": {
|
|
"name": "示例站点",
|
|
"params": {
|
|
"cookie": {
|
|
"name": "Cookie",
|
|
"type": "text",
|
|
"placeholder": "填写 Cookie",
|
|
"tooltip": "用于站点认证",
|
|
"convert": "upper",
|
|
}
|
|
},
|
|
}
|
|
}
|
|
assert uow._sync_transaction_runner is None
|
|
assert uow._async_transaction_runner is None
|
|
|
|
|
|
def test_apply_config_does_not_create_downloader_internal_path(monkeypatch):
|
|
"""
|
|
应用安装配置时只创建宿主侧媒体目录。
|
|
"""
|
|
module = load_local_setup_module()
|
|
created_directories: list[str] = []
|
|
original_import = builtins.__import__
|
|
|
|
def fake_path(value: str):
|
|
"""
|
|
记录目录创建请求，不修改宿主文件系统。
|
|
"""
|
|
|
|
def mkdir(*, parents: bool, exist_ok: bool) -> None:
|
|
"""
|
|
记录请求创建的目录。
|
|
"""
|
|
assert parents and exist_ok
|
|
created_directories.append(value)
|
|
|
|
return types.SimpleNamespace(mkdir=mkdir)
|
|
|
|
def fail_before_runtime_initialization(name: str, *args, **kwargs):
|
|
"""
|
|
路径创建阶段结束后中止，避免进入数据库初始化。
|
|
"""
|
|
if name == "app.application.classification.reference":
|
|
raise ModuleNotFoundError("stop after path setup")
|
|
return original_import(name, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(module, "Path", fake_path)
|
|
monkeypatch.setattr(builtins, "__import__", fail_before_runtime_initialization)
|
|
|
|
with pytest.raises(RuntimeError, match="尚未安装 MoviePilot 运行依赖"):
|
|
module._apply_local_system_config_inner(
|
|
{
|
|
"directories": [
|
|
{"download_path": "/downloads", "library_path": "/video"}
|
|
]
|
|
}
|
|
)
|
|
|
|
assert created_directories == ["/video"]
|