chendelian
aab0207aec
All checks were successful
Publish Python Package to Gitea / publish (push) Successful in 25s
Co-authored-by: Cursor <cursoragent@cursor.com>
234 lines
8.6 KiB
Python
234 lines
8.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""media_assets 共享媒体资源解析测试。"""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import platform
|
|
import zipfile
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from jiangchang_skill_core import media_assets as ma
|
|
|
|
|
|
def _make_complete_assets(root: Path, *, with_ffmpeg: bool = True, with_ffprobe: bool = True) -> None:
|
|
(root / "music" / "calm").mkdir(parents=True)
|
|
(root / "fonts").mkdir(parents=True)
|
|
(root / "watermark").mkdir(parents=True)
|
|
bin_key = ma._platform_bin_key()
|
|
bin_dir = root / "bin" / bin_key
|
|
bin_dir.mkdir(parents=True)
|
|
if with_ffmpeg:
|
|
(bin_dir / ma._tool_filename("ffmpeg")).write_bytes(b"ffmpeg")
|
|
if with_ffprobe:
|
|
(bin_dir / ma._tool_filename("ffprobe")).write_bytes(b"ffprobe")
|
|
|
|
|
|
def _sample_manifest() -> dict:
|
|
bin_key = ma._platform_bin_key()
|
|
return {
|
|
"name": "media-assets",
|
|
"version": "0.2.0",
|
|
"directories": {"music": "music", "fonts": "fonts", "watermark": "watermark"},
|
|
"tools": {
|
|
"ffmpeg": {
|
|
bin_key: {
|
|
"package": "ffmpeg-release-essentials",
|
|
"archive_type": "zip",
|
|
"urls": [{"name": "test", "url": "https://example.test/ffmpeg.zip"}],
|
|
"archive_paths": {
|
|
"ffmpeg": "*/bin/ffmpeg.exe",
|
|
"ffprobe": "*/bin/ffprobe.exe",
|
|
},
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
|
|
def _install_tools_from_manifest(root: Path, warnings: list[str]) -> bool:
|
|
bin_key = ma._platform_bin_key()
|
|
bin_dir = root / "bin" / bin_key
|
|
bin_dir.mkdir(parents=True, exist_ok=True)
|
|
(bin_dir / ma._tool_filename("ffmpeg")).write_bytes(b"ffmpeg")
|
|
(bin_dir / ma._tool_filename("ffprobe")).write_bytes(b"ffprobe")
|
|
return True
|
|
|
|
|
|
def test_media_assets_root_env_priority(tmp_path: Path) -> None:
|
|
custom = tmp_path / "custom-media"
|
|
custom.mkdir()
|
|
env = {
|
|
"MEDIA_ASSETS_ROOT": str(custom),
|
|
"JIANGCHANG_DATA_ROOT": str(tmp_path / "data"),
|
|
}
|
|
assert ma.resolve_media_assets_root(env=env) == custom
|
|
|
|
|
|
def test_default_media_assets_path(tmp_path: Path) -> None:
|
|
env = {"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")}
|
|
expected = tmp_path / "data" / "shared" / "media-assets"
|
|
assert ma.resolve_media_assets_root(env=env) == expected
|
|
|
|
|
|
def test_complete_assets_skips_download(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
root = tmp_path / "data" / "shared" / "media-assets"
|
|
_make_complete_assets(root)
|
|
|
|
def _fail_download(*args, **kwargs):
|
|
raise AssertionError("download should not be called")
|
|
|
|
monkeypatch.setattr(ma, "_try_download_via_zip", _fail_download)
|
|
monkeypatch.setattr(ma, "_try_download_via_git", _fail_download)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert status.ready is True
|
|
assert status.warnings == []
|
|
assert status.source == "local"
|
|
|
|
|
|
def test_missing_ffmpeg_warning(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
root = tmp_path / "data" / "shared" / "media-assets"
|
|
_make_complete_assets(root, with_ffmpeg=False)
|
|
|
|
monkeypatch.setattr(ma, "_try_download_via_zip", lambda *a, **k: False)
|
|
monkeypatch.setattr(ma, "_try_download_via_git", lambda *a, **k: False)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert status.ready is False
|
|
assert "ffmpeg_missing" in status.warnings
|
|
|
|
|
|
def test_missing_ffprobe_warning(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
root = tmp_path / "data" / "shared" / "media-assets"
|
|
_make_complete_assets(root, with_ffprobe=False)
|
|
|
|
monkeypatch.setattr(ma, "_try_download_via_zip", lambda *a, **k: False)
|
|
monkeypatch.setattr(ma, "_try_download_via_git", lambda *a, **k: False)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert status.ready is False
|
|
assert "ffprobe_missing" in status.warnings
|
|
|
|
|
|
def test_pick_background_music_stable_first(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
root = tmp_path / "data" / "shared" / "media-assets"
|
|
_make_complete_assets(root)
|
|
music = root / "music"
|
|
(music / "b-upbeat.mp3").write_bytes(b"mp3")
|
|
(music / "calm" / "a-calm.mp3").write_bytes(b"mp3")
|
|
(music / "z-last.wav").write_bytes(b"wav")
|
|
|
|
monkeypatch.setattr(
|
|
ma,
|
|
"ensure_media_assets",
|
|
lambda *a, **k: ma._inspect_media_assets(root, source="local"),
|
|
)
|
|
|
|
picked = ma.pick_background_music(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert picked == music / "b-upbeat.mp3"
|
|
|
|
|
|
def _build_archive_zip(dest: Path, inner_name: str) -> None:
|
|
manifest = _sample_manifest()
|
|
with zipfile.ZipFile(dest, "w") as zf:
|
|
zf.writestr(f"{inner_name}/README.md", "# media-assets\n")
|
|
zf.writestr(f"{inner_name}/manifest.json", json.dumps(manifest, ensure_ascii=False))
|
|
zf.writestr(f"{inner_name}/music/.keep", "")
|
|
zf.writestr(f"{inner_name}/fonts/.keep", "")
|
|
zf.writestr(f"{inner_name}/watermark/.keep", "")
|
|
|
|
|
|
def test_zip_wrapper_layout_detected(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
data_root = tmp_path / "data"
|
|
shared = data_root / "shared"
|
|
target = shared / "media-assets"
|
|
archive = tmp_path / "media-assets.zip"
|
|
|
|
_build_archive_zip(archive, "media-assets-main")
|
|
|
|
def _fake_download(url: str, dest: Path) -> None:
|
|
dest.write_bytes(archive.read_bytes())
|
|
|
|
monkeypatch.setattr(ma, "_download_zip", _fake_download)
|
|
monkeypatch.setattr(ma, "_find_git_executable", lambda: None)
|
|
monkeypatch.setattr(ma, "_try_download_ffmpeg_tools", _install_tools_from_manifest)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(data_root)})
|
|
assert target.is_dir()
|
|
assert status.ready is True
|
|
assert status.source == "zip"
|
|
assert "media_assets_zip_invalid_layout" not in status.warnings
|
|
|
|
|
|
def test_zip_download_failure_returns_warning(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
import urllib.error
|
|
|
|
def _raise_url_error(url: str, dest: Path) -> None:
|
|
raise urllib.error.URLError("network down")
|
|
|
|
monkeypatch.setattr(ma, "_download_zip", _raise_url_error)
|
|
monkeypatch.setattr(ma, "_find_git_executable", lambda: None)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert status.ready is False
|
|
assert "media_assets_zip_download_failed" in status.warnings
|
|
assert "git_not_available" in status.warnings
|
|
|
|
|
|
def test_git_not_available_when_zip_fails(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
import urllib.error
|
|
|
|
monkeypatch.setattr(
|
|
ma,
|
|
"_download_zip",
|
|
lambda url, dest: (_ for _ in ()).throw(urllib.error.URLError("fail")),
|
|
)
|
|
monkeypatch.setattr(ma, "_find_git_executable", lambda: None)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert "git_not_available" in status.warnings
|
|
|
|
|
|
def test_media_assets_root_override_no_download(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
custom = tmp_path / "override"
|
|
custom.mkdir()
|
|
|
|
def _fail_download(*args, **kwargs):
|
|
raise AssertionError("download should not be called when MEDIA_ASSETS_ROOT is set")
|
|
|
|
monkeypatch.setattr(ma, "_try_download_via_zip", _fail_download)
|
|
|
|
status = ma.ensure_media_assets(env={"MEDIA_ASSETS_ROOT": str(custom)})
|
|
assert status.root == custom
|
|
assert status.exists is True
|
|
|
|
|
|
@pytest.mark.skipif(platform.system().lower() != "windows", reason="Windows default path check")
|
|
def test_windows_default_data_root() -> None:
|
|
with patch.dict("os.environ", {}, clear=True):
|
|
root = ma.resolve_media_assets_root()
|
|
assert root == Path(r"D:\jiangchang-data") / "shared" / "media-assets"
|
|
|
|
|
|
def test_manifest_ffmpeg_download(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
root = tmp_path / "data" / "shared" / "media-assets"
|
|
(root / "music").mkdir(parents=True)
|
|
(root / "fonts").mkdir(parents=True)
|
|
(root / "watermark").mkdir(parents=True)
|
|
(root / "manifest.json").write_text(
|
|
json.dumps(_sample_manifest(), ensure_ascii=False),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
monkeypatch.setattr(ma, "_try_download_via_zip", lambda *a, **k: False)
|
|
monkeypatch.setattr(ma, "_try_download_via_git", lambda *a, **k: False)
|
|
monkeypatch.setattr(ma, "_try_download_ffmpeg_tools", _install_tools_from_manifest)
|
|
|
|
status = ma.ensure_media_assets(env={"JIANGCHANG_DATA_ROOT": str(tmp_path / "data")})
|
|
assert status.ready is True
|
|
assert status.ffmpeg_path is not None
|
|
assert status.ffprobe_path is not None
|