mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-05 07:27:15 +08:00
fix(setup): 精确同步当前站点资源 ABI (#6444)
This commit is contained in:
InfinityPacer
@@ -0,0 +1,254 @@
|
||||
import importlib.util
|
||||
import sys
|
||||
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_resources_{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
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("gil_disabled", "suffix"),
|
||||
[
|
||||
(0, ""),
|
||||
(1, "t"),
|
||||
],
|
||||
)
|
||||
def test_python_version_tag_distinguishes_free_threaded_abi(
|
||||
monkeypatch, gil_disabled, suffix
|
||||
):
|
||||
module = load_local_setup_module()
|
||||
monkeypatch.setattr(
|
||||
module.sysconfig,
|
||||
"get_config_var",
|
||||
lambda name: gil_disabled if name == "Py_GIL_DISABLED" else None,
|
||||
)
|
||||
|
||||
expected = f"cp{sys.version_info.major}{sys.version_info.minor}{suffix}"
|
||||
assert module._get_python_version_tag() == expected
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("platform_tag", "machine", "python_version", "expected"),
|
||||
[
|
||||
(
|
||||
"linux",
|
||||
"x86_64",
|
||||
"cp314",
|
||||
"sites.cpython-314-x86_64-linux-gnu.so",
|
||||
),
|
||||
(
|
||||
"linux",
|
||||
"x86_64",
|
||||
"cp314t",
|
||||
"sites.cpython-314t-x86_64-linux-gnu.so",
|
||||
),
|
||||
(
|
||||
"linux",
|
||||
"aarch64",
|
||||
"cp314",
|
||||
"sites.cpython-314-aarch64-linux-gnu.so",
|
||||
),
|
||||
(
|
||||
"linux",
|
||||
"aarch64",
|
||||
"cp314t",
|
||||
"sites.cpython-314t-aarch64-linux-gnu.so",
|
||||
),
|
||||
("darwin", "arm64", "cp314", "sites.cpython-314-darwin.so"),
|
||||
("darwin", "arm64", "cp314t", "sites.cpython-314t-darwin.so"),
|
||||
("windows", "amd64", "cp314", "sites.cp314-win_amd64.pyd"),
|
||||
("windows", "amd64", "cp314t", "sites.cp314t-win_amd64.pyd"),
|
||||
],
|
||||
)
|
||||
def test_filter_resources_files_selects_exact_runtime_artifact(
|
||||
tmp_path, platform_tag, machine, python_version, expected
|
||||
):
|
||||
module = load_local_setup_module()
|
||||
filenames = {
|
||||
"user.sites.v3.bin",
|
||||
"sites.pyi",
|
||||
"sites.cpython-314-x86_64-linux-gnu.so",
|
||||
"sites.cpython-314-aarch64-linux-gnu.so",
|
||||
"sites.cpython-314t-x86_64-linux-gnu.so",
|
||||
"sites.cpython-314t-aarch64-linux-gnu.so",
|
||||
"sites.cpython-314-darwin.so",
|
||||
"sites.cpython-314t-darwin.so",
|
||||
"sites.cp314-win_amd64.pyd",
|
||||
"sites.cp314t-win_amd64.pyd",
|
||||
}
|
||||
for filename in filenames:
|
||||
(tmp_path / filename).write_bytes(filename.encode())
|
||||
|
||||
matched = module._filter_resources_files(
|
||||
tmp_path,
|
||||
platform_tag,
|
||||
machine,
|
||||
python_version,
|
||||
)
|
||||
|
||||
assert [path.name for path in matched] == ["user.sites.v3.bin", expected]
|
||||
|
||||
|
||||
def test_copy_resource_files_keeps_only_current_runtime_artifact(
|
||||
monkeypatch, tmp_path
|
||||
):
|
||||
module = load_local_setup_module()
|
||||
source_dir = tmp_path / "resources.v3"
|
||||
target_dir = tmp_path / "site"
|
||||
source_dir.mkdir()
|
||||
target_dir.mkdir()
|
||||
for filename in (
|
||||
"user.sites.v3.bin",
|
||||
"sites.cpython-314-x86_64-linux-gnu.so",
|
||||
"sites.cpython-314-aarch64-linux-gnu.so",
|
||||
"sites.cpython-314t-x86_64-linux-gnu.so",
|
||||
):
|
||||
(source_dir / filename).write_text(f"source:{filename}", encoding="utf-8")
|
||||
(target_dir / "sites.pyi").write_text("typing", encoding="utf-8")
|
||||
(target_dir / "sites.cpython-313-x86_64-linux-gnu.so").write_text(
|
||||
"stale", encoding="utf-8"
|
||||
)
|
||||
(target_dir / "sites.cp314-win_amd64.pyd").write_text(
|
||||
"stale", encoding="utf-8"
|
||||
)
|
||||
(target_dir / "sites.cpython-314t-darwin.so").write_text(
|
||||
"stale", encoding="utf-8"
|
||||
)
|
||||
(target_dir / "sites.legacy.dylib").write_text("stale", encoding="utf-8")
|
||||
monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
|
||||
monkeypatch.setattr(module, "_get_platform_tag", lambda: ("linux", "x86_64"))
|
||||
monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
|
||||
|
||||
copied = module.copy_resource_files(source_dir)
|
||||
|
||||
assert copied == [
|
||||
"user.sites.v3.bin",
|
||||
"sites.cpython-314t-x86_64-linux-gnu.so",
|
||||
]
|
||||
assert sorted(path.name for path in target_dir.iterdir()) == [
|
||||
"sites.cpython-314t-x86_64-linux-gnu.so",
|
||||
"sites.pyi",
|
||||
"user.sites.v3.bin",
|
||||
]
|
||||
|
||||
|
||||
def test_copy_resource_files_rejects_partial_source_before_changing_target(
|
||||
monkeypatch, tmp_path
|
||||
):
|
||||
module = load_local_setup_module()
|
||||
source_dir = tmp_path / "resources.v3"
|
||||
target_dir = tmp_path / "site"
|
||||
source_dir.mkdir()
|
||||
target_dir.mkdir()
|
||||
(source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
|
||||
old_native = target_dir / "sites.cpython-314-darwin.so"
|
||||
old_data = target_dir / "user.sites.v3.bin"
|
||||
old_native.write_bytes(b"old-native")
|
||||
old_data.write_bytes(b"old-data")
|
||||
monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
|
||||
monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
|
||||
monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
|
||||
|
||||
with pytest.raises(RuntimeError, match="sites.cpython-314t-darwin.so"):
|
||||
module.copy_resource_files(source_dir)
|
||||
|
||||
assert old_native.read_bytes() == b"old-native"
|
||||
assert old_data.read_bytes() == b"old-data"
|
||||
|
||||
|
||||
def test_copy_resource_files_keeps_target_when_staging_fails(monkeypatch, tmp_path):
|
||||
module = load_local_setup_module()
|
||||
source_dir = tmp_path / "resources.v3"
|
||||
target_dir = tmp_path / "site"
|
||||
source_dir.mkdir()
|
||||
target_dir.mkdir()
|
||||
(source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
|
||||
new_native_name = "sites.cpython-314t-darwin.so"
|
||||
(source_dir / new_native_name).write_bytes(b"new-native")
|
||||
old_native = target_dir / "sites.cpython-314-darwin.so"
|
||||
old_data = target_dir / "user.sites.v3.bin"
|
||||
old_native.write_bytes(b"old-native")
|
||||
old_data.write_bytes(b"old-data")
|
||||
real_copy2 = module.shutil.copy2
|
||||
|
||||
def fail_native_staging(source, target):
|
||||
if Path(source).name == new_native_name:
|
||||
raise OSError("injected native staging failure")
|
||||
return real_copy2(source, target)
|
||||
|
||||
monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
|
||||
monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
|
||||
monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
|
||||
monkeypatch.setattr(module.shutil, "copy2", fail_native_staging)
|
||||
|
||||
with pytest.raises(OSError, match="injected native staging failure"):
|
||||
module.copy_resource_files(source_dir)
|
||||
|
||||
assert old_native.read_bytes() == b"old-native"
|
||||
assert old_data.read_bytes() == b"old-data"
|
||||
assert not (target_dir / new_native_name).exists()
|
||||
assert not list(target_dir.glob(".sites-sync-*"))
|
||||
|
||||
|
||||
def test_copy_resource_files_rolls_back_when_commit_fails(monkeypatch, tmp_path):
|
||||
module = load_local_setup_module()
|
||||
source_dir = tmp_path / "resources.v3"
|
||||
target_dir = tmp_path / "site"
|
||||
source_dir.mkdir()
|
||||
target_dir.mkdir()
|
||||
(source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
|
||||
new_native_name = "sites.cpython-314t-darwin.so"
|
||||
(source_dir / new_native_name).write_bytes(b"new-native")
|
||||
old_native = target_dir / "sites.cpython-314-darwin.so"
|
||||
old_data = target_dir / "user.sites.v3.bin"
|
||||
old_native.write_bytes(b"old-native")
|
||||
old_data.write_bytes(b"old-data")
|
||||
real_replace = module.os.replace
|
||||
|
||||
def fail_native_commit(source, target):
|
||||
if Path(source).parent.name == "staging" and Path(target).name == new_native_name:
|
||||
raise OSError("injected native commit failure")
|
||||
return real_replace(source, target)
|
||||
|
||||
monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
|
||||
monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
|
||||
monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
|
||||
monkeypatch.setattr(module.os, "replace", fail_native_commit)
|
||||
|
||||
with pytest.raises(OSError, match="injected native commit failure"):
|
||||
module.copy_resource_files(source_dir)
|
||||
|
||||
assert old_native.read_bytes() == b"old-native"
|
||||
assert old_data.read_bytes() == b"old-data"
|
||||
assert not (target_dir / new_native_name).exists()
|
||||
|
||||
|
||||
def test_local_resource_status_requires_current_native_artifact(
|
||||
monkeypatch, tmp_path
|
||||
):
|
||||
module = load_local_setup_module()
|
||||
monkeypatch.setattr(module, "SITE_RESOURCE_DIR", tmp_path)
|
||||
monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
|
||||
monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
|
||||
(tmp_path / "user.sites.v3.bin").write_bytes(b"data")
|
||||
(tmp_path / "sites.pyi").write_text("typing", encoding="utf-8")
|
||||
(tmp_path / "sites.cpython-314-darwin.so").write_bytes(b"standard")
|
||||
|
||||
assert module.local_resource_status() is False
|
||||
|
||||
(tmp_path / "sites.cpython-314t-darwin.so").write_bytes(b"free-threaded")
|
||||
|
||||
assert module.local_resource_status() is True
|
||||
Reference in New Issue
Block a user