""" 验证本地前端安装时的版本选择和兼容处理。 """ from __future__ import annotations import importlib.util import tempfile import uuid 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_frontend_{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 def test_repo_frontend_version_reads_version_file(): """ 前端固定版本取自当前源码声明。 """ module = load_local_setup_module() with tempfile.TemporaryDirectory() as temp_dir: root = Path(temp_dir) (root / "version.py").write_text( "APP_VERSION = 'v0.0.1'\nFRONTEND_VERSION = 'v9.9.9'\n", encoding="utf-8", ) with patch.object(module, "ROOT", root): assert module._repo_frontend_version() == "v9.9.9" def test_resolve_frontend_release_uses_repo_frontend_version_by_default(): """ 普通模式请求源码声明的前端 Release。 """ module = load_local_setup_module() release = { "tag_name": "v9.9.9", "assets": [ { "name": "dist.zip", "browser_download_url": "https://example.com/dist.zip", } ], } with patch.object(module, "_repo_frontend_version", return_value="v9.9.9"), patch.object( module, "fetch_json", return_value=release ) as fetch_mock: tag_name, download_url = module._resolve_frontend_release(None) fetch_mock.assert_called_once_with( module.FRONTEND_TAG_API.format(tag="v9.9.9") ) assert tag_name == "v9.9.9" assert download_url == "https://example.com/dist.zip" def test_parser_leaves_frontend_version_empty_until_runtime_resolution(): """ 未显式指定版本时延迟到运行阶段读取最新源码。 """ module = load_local_setup_module() parser = module.build_parser() install_args = parser.parse_args(["install-frontend"]) setup_args = parser.parse_args(["setup"]) update_args = parser.parse_args(["update", "frontend"]) assert install_args.version is None assert setup_args.frontend_version is None assert update_args.frontend_version is None @pytest.mark.parametrize("enabled", ["true", "1", "yes", "false", "0"]) @pytest.mark.parametrize("target", ["backend", "frontend", "all"]) def test_update_uses_configured_dev_mode(monkeypatch, enabled, target): """ DEV 配置作用于所有手动更新目标,前端从最新 Release 解析。 """ module = load_local_setup_module() monkeypatch.delenv("MOVIEPILOT_UPDATE_DEV", raising=False) monkeypatch.setattr(module, "read_env_value", lambda _key: enabled) monkeypatch.setattr(module, "_git_output", lambda *_args: "v3") args = module.build_parser().parse_args(["update", target]) assert module._resolve_update_versions(args) == ( "latest", "latest" if enabled in {"true", "1", "yes"} else None ) @pytest.mark.parametrize("options, expected", [ (["--dev"], ("v3", "latest")), (["--no-dev"], ("latest", None)), (["--dev", "--ref", "v3.0.0", "--frontend-version", "v3.0.0"], ("v3.0.0", "v3.0.0")), (["--offline-backend", "--ref", "v3.0.0", "--frontend-version", "v3.0.0"], ("v3.0.0", "v3.0.0")), ]) def test_explicit_update_options_and_detached_head(monkeypatch, options, expected): """ 显式选项覆盖 DEV 默认值,离线安装保持下载清单的版本。 """ module = load_local_setup_module() monkeypatch.setenv("MOVIEPILOT_UPDATE_DEV", "true") monkeypatch.setattr(module, "_git_output", lambda *_args: "HEAD") args = module.build_parser().parse_args(["update", *options]) assert args.target == "all" assert module._resolve_update_versions(args) == expected def test_environment_overrides_saved_dev_config(monkeypatch): """ 进程环境配置优先于 app.env 中保存的 DEV 偏好。 """ module = load_local_setup_module() monkeypatch.setenv("MOVIEPILOT_UPDATE_DEV", "false") monkeypatch.setattr(module, "read_env_value", lambda _key: "true") assert module._resolve_update_versions(module.build_parser().parse_args(["update"])) == ("latest", None) def test_dev_frontend_downloads_latest_release(): """ latest 必须访问最新 Release API,不读取后端固定前端版本。 """ module = load_local_setup_module() release = {"tag_name": "v3.0.11", "assets": [{"name": "dist.zip", "browser_download_url": "https://example.com/dist.zip"}]} with patch.object(module, "fetch_json", return_value=release) as fetch, patch.object( module, "_repo_frontend_version", side_effect=AssertionError("不应读取固定版本") ): assert module._resolve_frontend_release("latest")[0] == "v3.0.11" fetch.assert_called_once_with(module.FRONTEND_LATEST_API) @pytest.mark.parametrize("target", ["backend", "frontend", "all"]) def test_update_command_passes_dev_selection_to_installers(monkeypatch, tmp_path, target): """ 命令入口把 DEV 选择传给实际安装边界,并保持后端先于前端执行。 """ module = load_local_setup_module() monkeypatch.setenv("MOVIEPILOT_UPDATE_DEV", "true") monkeypatch.setattr(module.sys, "argv", ["local_setup.py", "update", target, "--skip-resources"]) monkeypatch.setattr(module, "configure_config_dir", lambda **_: tmp_path) monkeypatch.setattr(module, "_git_output", lambda *_: "v3") calls = [] with patch.object(module, "ensure_services_stopped"), patch.object( module, "update_backend", side_effect=lambda **kwargs: calls.append(("backend", kwargs["ref"])) ), patch.object(module, "install_frontend", side_effect=lambda **kwargs: calls.append( ("frontend", kwargs["frontend_version"]) ) or {"version": "v3.0.11"}): assert module.main() == 0 assert calls == [(name, "latest") for name in ("backend", "frontend") if target in {name, "all"}] @pytest.mark.parametrize("fetch, branch", [(True, "v3"), (False, "v3"), (True, "HEAD")]) def test_switching_to_dev_branch_fetches_latest_tip(monkeypatch, tmp_path, fetch, branch): """ 切回开发分支后快进远端提交;离线安装和标签检出不拉取分支。 """ module = load_local_setup_module() (tmp_path / ".git").touch() monkeypatch.setattr(module, "ROOT", tmp_path) monkeypatch.setattr(module, "_ensure_git_clean", lambda: None) monkeypatch.setattr(module, "_git_output", lambda *_: branch) with patch.object(module, "run") as run: assert module._update_backend_ref("v3", fetch=fetch) == "v3" pull_calls = [call.args[0] for call in run.call_args_list if call.args[0][:2] == ["git", "pull"]] assert pull_calls == ([["git", "pull", "--ff-only", "origin", "v3"]] if fetch and branch == "v3" else []) @pytest.mark.parametrize("target", ["frontend", "all"]) @pytest.mark.parametrize("version_args", [[], ["--frontend-version", "v3.0.11"]]) @pytest.mark.parametrize("saved_dev, mode_args, refresh", [ ("true", [], True), ("false", ["--dev"], True), ("true", ["--no-dev"], False), ("false", [], False), ]) def test_update_reinstalls_repacked_frontend_in_dev_mode( monkeypatch, tmp_path, target, version_args, saved_dev, mode_args, refresh ): """ 连续更新同版本不同内容的 ZIP:DEV 真正替换文件,普通模式仍复用已安装版本。 """ import zipfile module = load_local_setup_module() public = tmp_path / "public" (public / "node_modules" / "express").mkdir(parents=True) (public / "version.txt").write_text("v3.0.11", encoding="utf-8") (public / "service.js").touch() (public / "index.html").write_text("original", encoding="utf-8") npm = tmp_path / "npm" npm.touch() monkeypatch.setattr(module, "PUBLIC_DIR", public) monkeypatch.delenv("MOVIEPILOT_UPDATE_DEV", raising=False) monkeypatch.setattr(module, "read_env_value", lambda _key: saved_dev) monkeypatch.setattr(module, "configure_config_dir", lambda **_: tmp_path) monkeypatch.setattr(module, "ensure_services_stopped", lambda: None) monkeypatch.setattr(module, "update_backend", lambda **_: None) monkeypatch.setattr(module, "_git_output", lambda *_: "v3") monkeypatch.setattr(module, "_repo_frontend_version", lambda: "v3.0.11") monkeypatch.setattr(module, "install_node_runtime", lambda _version: tmp_path / "node") monkeypatch.setattr(module, "get_npm_bin", lambda: npm) monkeypatch.setattr(module, "fetch_json", lambda _url: { "tag_name": "v3.0.11", "assets": [{"name": "dist.zip", "browser_download_url": "https://example.com/dist.zip"}], }) monkeypatch.setattr(module.sys, "argv", [ "local_setup.py", "update", *version_args, target, *mode_args, "--skip-resources" ]) downloads = [] def download_repacked_release(url, path): """ 模拟同一个 Release URL 每次提供重新构建的内容。 """ downloads.append(url) with zipfile.ZipFile(path, "w") as archive: archive.writestr("dist/version.txt", "v3.0.11") archive.writestr("dist/index.html", f"build-{len(downloads)}") def install_runtime_deps(*_args, **kwargs): """ 恢复已安装依赖标记,保证第二次确实面对完整的同版本前端。 """ (kwargs["cwd"] / "node_modules" / "express").mkdir(parents=True, exist_ok=True) monkeypatch.setattr(module, "download_file", download_repacked_release) monkeypatch.setattr(module, "run", install_runtime_deps) for attempt in (1, 2): assert module._frontend_runtime_ready("v3.0.11") assert module.main() == 0 assert (public / "index.html").read_text(encoding="utf-8") == ( f"build-{attempt}" if refresh else "original" ) assert downloads == (["https://example.com/dist.zip"] * 2 if refresh else [])