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"