Philip Guzman
87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from serato_doctor.crate_parser import (
|
|
clean_path,
|
|
load_crate,
|
|
parse_crate,
|
|
parse_crates,
|
|
path_markers,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("artifact", "expected"),
|
|
[
|
|
("song.mp漳", "song.mp3"),
|
|
("song.m4愠", "song.m4a"),
|
|
("song.wa瘠", "song.wav"),
|
|
("song.ai映", "song.aif"),
|
|
],
|
|
)
|
|
def test_clean_path_repairs_known_extension_artifacts(artifact, expected):
|
|
assert clean_path(artifact) == expected
|
|
|
|
|
|
def test_parse_crate_extracts_references(tmp_path):
|
|
crate_path = tmp_path / "House.crate"
|
|
crate_text = (
|
|
"header"
|
|
"Users/sample-user/OneDrive/Jukebox/House/First.mp漳牴k"
|
|
"metadata"
|
|
"Users/sample-user/OneDrive/Jukebox/House/Second.m4愠otrk"
|
|
)
|
|
crate_path.write_bytes(crate_text.encode("utf-16-le"))
|
|
|
|
crate = load_crate(crate_path)
|
|
|
|
assert crate.path == crate_path
|
|
assert [reference.filename for reference in crate.references] == [
|
|
"First.mp3",
|
|
"Second.m4a",
|
|
]
|
|
assert parse_crate(crate_path) == list(crate.references)
|
|
|
|
|
|
def test_parse_crate_ignores_record_without_stop_marker(tmp_path):
|
|
crate_path = Path(tmp_path) / "Incomplete.crate"
|
|
crate_path.write_bytes(
|
|
"Users/sample-user/OneDrive/Jukebox/House/Incomplete.mp3".encode(
|
|
"utf-16-le"
|
|
)
|
|
)
|
|
|
|
assert parse_crate(crate_path) == []
|
|
|
|
|
|
def test_parse_crate_uses_configured_reference_root(tmp_path):
|
|
crate_path = tmp_path / "Migrated.crate"
|
|
crate_path.write_bytes(
|
|
"Archive/Old Library/House/Track.mp3otrk".encode("utf-16-le")
|
|
)
|
|
|
|
references = parse_crate(crate_path, [Path("/Archive/Old Library")])
|
|
|
|
assert references[0].path == Path("/Archive/Old Library/House/Track.mp3")
|
|
|
|
|
|
def test_default_path_markers_do_not_contain_a_username():
|
|
assert path_markers([]) == ("Users/", "Volumes/")
|
|
|
|
|
|
def test_parse_crates_reuses_configured_roots_for_every_crate(tmp_path):
|
|
for name in ("First", "Second"):
|
|
(tmp_path / f"{name}.crate").write_bytes(
|
|
f"Archive/Jukebox/{name}.mp3otrk".encode("utf-16-le")
|
|
)
|
|
|
|
references = parse_crates(
|
|
tmp_path, (root for root in [Path("/Archive/Jukebox")])
|
|
)
|
|
|
|
assert {reference.filename for reference in references} == {
|
|
"First.mp3",
|
|
"Second.mp3",
|
|
}
|