refactor default artifacts workspace and path compatibility

This commit is contained in:
windyboy
2026-02-16 13:23:01 +08:00
parent 2dd329cba9
commit caa6881fd2
8 changed files with 314 additions and 79 deletions
+114
View File
@@ -0,0 +1,114 @@
"""Tests for CLI artifact default paths and legacy compatibility warnings."""
import json
from pathlib import Path
from unittest.mock import patch
from click.testing import CliRunner
from vlm.cli import main
def _write_config(path: Path, library_root: Path, workspace_dir: str | None = None) -> None:
workspace_line = f"\nworkspace_dir: {workspace_dir}" if workspace_dir else ""
path.write_text(
(
f"library_root: {library_root}\n"
"video_extensions:\n"
" - .mp4\n"
" - .mkv\n"
"categories:\n"
" movie: [movie, movies]\n"
" series: [series, tv, shows]\n"
" anime: [anime]\n"
f"{workspace_line}\n"
),
encoding="utf-8",
)
def _write_inventory_csv(path: Path) -> None:
path.parent.mkdir(parents=True, exist_ok=True)
path.write_text(
"# vlm inventory\n"
"path,filename,size_bytes,modified_timestamp,category,resolution,codec,duration_seconds,bitrate_kbps\n"
"/library/movie/Matrix (1999).mkv,Matrix (1999).mkv,1000000,2024-01-01T00:00:00,movie,,,\n",
encoding="utf-8",
)
def test_scan_default_output_uses_artifacts_dir(tmp_path):
"""Scan should write to artifacts/inventory.csv by default."""
library_root = tmp_path / "library"
library_root.mkdir(parents=True)
config_file = tmp_path / "config.yaml"
_write_config(config_file, library_root)
runner = CliRunner()
with patch("vlm.commands.scan.scan_library", return_value=[]), patch(
"vlm.commands.scan.save_inventory_csv"
) as mock_save:
result = runner.invoke(main, ["--config", str(config_file), "scan"])
assert result.exit_code == 0
assert mock_save.call_count == 1
assert mock_save.call_args.args[1] == Path("artifacts/inventory.csv")
def test_scan_default_output_uses_workspace_dir_from_config(tmp_path):
"""Scan should honor configured workspace_dir for default output."""
library_root = tmp_path / "library"
library_root.mkdir(parents=True)
config_file = tmp_path / "config.yaml"
_write_config(config_file, library_root, workspace_dir="work/cache")
runner = CliRunner()
with patch("vlm.commands.scan.scan_library", return_value=[]), patch(
"vlm.commands.scan.save_inventory_csv"
) as mock_save:
result = runner.invoke(main, ["--config", str(config_file), "scan"])
assert result.exit_code == 0
assert mock_save.call_count == 1
assert mock_save.call_args.args[1] == Path("work/cache/inventory.csv")
def test_parse_default_input_falls_back_to_legacy_root_file_with_warning(tmp_path):
"""Parse should warn and fallback to legacy root inventory.csv during stage-A migration."""
runner = CliRunner()
with runner.isolated_filesystem(temp_dir=str(tmp_path)):
library_root = Path("library")
library_root.mkdir(parents=True)
config_file = Path("config.yaml")
_write_config(config_file, library_root)
_write_inventory_csv(Path("inventory.csv"))
result = runner.invoke(main, ["--config", str(config_file), "parse"])
assert result.exit_code == 0
assert "Warning: detected legacy default input at" in result.output
assert "example: --input" in result.output
assert Path("artifacts/identities.json").exists()
data = json.loads(Path("artifacts/identities.json").read_text(encoding="utf-8"))
assert data["metadata"]["source_inventory"] == "inventory.csv"
def test_parse_default_input_no_warning_when_artifacts_input_exists(tmp_path):
"""When artifacts input exists, parse should use it without migration warning."""
runner = CliRunner()
with runner.isolated_filesystem(temp_dir=str(tmp_path)):
library_root = Path("library")
library_root.mkdir(parents=True)
config_file = Path("config.yaml")
_write_config(config_file, library_root)
_write_inventory_csv(Path("artifacts/inventory.csv"))
_write_inventory_csv(Path("inventory.csv"))
result = runner.invoke(main, ["--config", str(config_file), "parse"])
assert result.exit_code == 0
assert "Warning: detected legacy default input at" not in result.output