84 lines
3.5 KiB
Python
84 lines
3.5 KiB
Python
import server.mf # noqa: F401
|
|
import musicfetch_core as mf
|
|
|
|
DISCOVERY_MBID = "48117b90-a16e-34ca-a514-19c702df1158"
|
|
|
|
DISCOVERY_ALBUM = {"title": "Discovery", "artist": {"artistName": "Daft Punk"},
|
|
"releaseDate": "2001-01-01", "foreignAlbumId": DISCOVERY_MBID}
|
|
|
|
|
|
def test_artist_track_uses_mbid_exact_lookup(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "testkey")
|
|
monkeypatch.setattr(mf, "musicbrainz_best_album",
|
|
lambda artist, track: {"album_title": "Discovery", "artist": "Daft Punk",
|
|
"year": "2001", "rg_mbid": DISCOVERY_MBID})
|
|
seen = {}
|
|
|
|
def fake_get(path, params=None, timeout=15):
|
|
seen["term"] = (params or {}).get("term")
|
|
if path == "/api/v1/album/lookup" and seen["term"] == f"mbid:{DISCOVERY_MBID}":
|
|
return [DISCOVERY_ALBUM]
|
|
return []
|
|
monkeypatch.setattr(mf, "lidarr_get", fake_get)
|
|
|
|
hits = mf.lidarr_search("Daft Punk - Harder Better Faster Stronger", 10)
|
|
assert seen["term"] == f"mbid:{DISCOVERY_MBID}"
|
|
assert hits[0].album == "Discovery"
|
|
assert hits[0].artist == "Daft Punk"
|
|
assert hits[0].payload["album"]["foreignAlbumId"] == DISCOVERY_MBID
|
|
|
|
|
|
def test_year_enriched_from_musicbrainz(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "testkey")
|
|
monkeypatch.setattr(mf, "musicbrainz_best_album",
|
|
lambda artist, track: {"album_title": "Discovery", "artist": "Daft Punk",
|
|
"year": "2001", "rg_mbid": DISCOVERY_MBID})
|
|
no_year = [{"title": "Discovery", "artist": {"artistName": "Daft Punk"},
|
|
"releaseDate": "", "foreignAlbumId": DISCOVERY_MBID}]
|
|
monkeypatch.setattr(mf, "lidarr_get",
|
|
lambda path, params=None, timeout=15: no_year if path == "/api/v1/album/lookup" else [])
|
|
hits = mf.lidarr_search("Daft Punk - Discovery", 10)
|
|
assert hits[0].year == "2001"
|
|
|
|
|
|
def test_no_api_key_returns_empty(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "")
|
|
assert mf.lidarr_search("Daft Punk - Discovery", 10) == []
|
|
|
|
|
|
def test_mb_miss_falls_back_to_lookup(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "testkey")
|
|
monkeypatch.setattr(mf, "musicbrainz_best_album", lambda artist, track: None)
|
|
monkeypatch.setattr(mf, "lidarr_get",
|
|
lambda path, params=None, timeout=15: [DISCOVERY_ALBUM] if path == "/api/v1/album/lookup" else [])
|
|
hits = mf.lidarr_search("Daft Punk - Discovery", 10)
|
|
assert hits[0].album == "Discovery"
|
|
|
|
|
|
def test_single_term_is_artist_first(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "testkey")
|
|
|
|
def fake_get(path, params=None, timeout=15):
|
|
if path == "/api/v1/artist/lookup":
|
|
return [{"artistName": "Daft Punk"}]
|
|
if path == "/api/v1/album/lookup":
|
|
return [DISCOVERY_ALBUM]
|
|
return []
|
|
monkeypatch.setattr(mf, "lidarr_get", fake_get)
|
|
hits = mf.lidarr_search("Daft Punk", 10)
|
|
assert hits[0].kind == "artist"
|
|
assert hits[0].artist == "Daft Punk"
|
|
|
|
|
|
def test_last_resort_universal_search(monkeypatch):
|
|
monkeypatch.setattr(mf, "API_KEY", "testkey")
|
|
monkeypatch.setattr(mf, "musicbrainz_best_album", lambda artist, track: None)
|
|
|
|
def fake_get(path, params=None, timeout=15):
|
|
if path == "/api/v1/search":
|
|
return [{"album": DISCOVERY_ALBUM}]
|
|
return []
|
|
monkeypatch.setattr(mf, "lidarr_get", fake_get)
|
|
hits = mf.lidarr_search("Daft Punk - Discovery", 10)
|
|
assert hits and hits[0].album == "Discovery"
|