Philip Guzman
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from serato_doctor.crate_parser import clean_path, load_crate, parse_crate
|
|
|
|
|
|
@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/djsplice/OneDrive/Jukebox/House/First.mp漳牴k"
|
|
"metadata"
|
|
"Users/djsplice/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/djsplice/OneDrive/Jukebox/House/Incomplete.mp3".encode("utf-16-le")
|
|
)
|
|
|
|
assert parse_crate(crate_path) == []
|