Philip Guzman
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from serato_doctor.repair import (
|
|
BACKUP_FOLDER,
|
|
apply_duplicate_repair,
|
|
plan_duplicate_repair,
|
|
restore_backup,
|
|
rotate_backups,
|
|
)
|
|
|
|
|
|
def library(tmp_path):
|
|
serato = tmp_path / "_Serato_"
|
|
crate = serato / "Subcrates" / "House.crate"
|
|
crate.parent.mkdir(parents=True)
|
|
crate.write_bytes(b"crate-data")
|
|
(serato / "database V2").write_bytes(b"database-data")
|
|
music = tmp_path / "Music"
|
|
keeper = music / "Main" / "Track.mp3"
|
|
duplicate = music / "Old" / "Track.mp3"
|
|
keeper.parent.mkdir(parents=True)
|
|
duplicate.parent.mkdir(parents=True)
|
|
keeper.write_bytes(b"keeper")
|
|
duplicate.write_bytes(b"duplicate")
|
|
return serato, keeper, duplicate
|
|
|
|
|
|
def test_duplicate_repair_backs_up_then_preserves_old_path_as_link(tmp_path):
|
|
serato, keeper, duplicate = library(tmp_path)
|
|
plan = plan_duplicate_repair(keeper, (keeper, duplicate), serato)
|
|
|
|
receipt = apply_duplicate_repair(plan, serato, backup_limit=3)
|
|
|
|
assert duplicate.is_symlink()
|
|
assert duplicate.resolve() == keeper.resolve()
|
|
assert (receipt.backup / "manifest.json").is_file()
|
|
assert any((receipt.backup / "serato-metadata").rglob("House.crate"))
|
|
assert any((receipt.backup / "serato-metadata").rglob("database V2"))
|
|
|
|
restored = restore_backup(receipt.backup)
|
|
assert restored == (duplicate.resolve(),)
|
|
assert not duplicate.is_symlink()
|
|
assert duplicate.read_bytes() == b"duplicate"
|
|
|
|
|
|
def test_plan_rejects_keeper_outside_duplicate_group(tmp_path):
|
|
serato, keeper, duplicate = library(tmp_path)
|
|
outsider = tmp_path / "outsider.mp3"
|
|
outsider.write_bytes(b"other")
|
|
|
|
with pytest.raises(ValueError, match="must belong"):
|
|
plan_duplicate_repair(outsider, (keeper, duplicate), serato)
|
|
|
|
|
|
def test_backup_rotation_can_be_limited_or_unlimited(tmp_path):
|
|
root = tmp_path / BACKUP_FOLDER
|
|
for name in ("001", "002", "003"):
|
|
backup = root / name
|
|
backup.mkdir(parents=True)
|
|
(backup / "manifest.json").write_text("{}", encoding="utf-8")
|
|
|
|
rotate_backups(root, None)
|
|
assert len(list(root.iterdir())) == 3
|
|
rotate_backups(root, 2)
|
|
assert {path.name for path in root.iterdir()} == {"002", "003"}
|