from server import actions, mf


def make_yt_hit():
    return mf.Hit(source="youtube", kind="track", title="Together",
                  artist="Avril Lavigne", album="Under My Skin", year="2004",
                  payload={"videoId": "abc"})


def make_lidarr_album_hit():
    return mf.Hit(source="lidarr", kind="album", title="Under My Skin",
                  artist="Avril Lavigne", album="Under My Skin", year="2004",
                  payload={"album": {"id": 5, "title": "Under My Skin"}})


def test_started_message_mentions_source_and_title():
    msg = actions.started_message(make_yt_hit())
    assert "Under My Skin" in msg
    assert "Avril Lavigne" in msg
    assert "YouTube" in msg


def test_done_message_mentions_title():
    msg = actions.done_message(make_yt_hit())
    assert "Under My Skin" in msg
    assert "Avril Lavigne" in msg


def test_perform_youtube_calls_act_youtube(monkeypatch):
    calls = {}
    monkeypatch.setattr(mf, "act_youtube",
                        lambda hit, root, quality, dry_run: calls.update(hit=hit, root=root, quality=quality))
    hit = make_yt_hit()
    result = actions.perform_fetch(hit, [hit], quality="best", root="/media/music")
    assert calls["quality"] == "best"
    assert result["path"] == "/media/music/Avril Lavigne/youtube"
    assert result["lidarr_album_id"] is None


def test_perform_lidarr_album_handled(monkeypatch):
    monkeypatch.setattr(mf, "act_lidarr_album",
                        lambda hit, root, search_all, dry_run: True)
    hit = make_lidarr_album_hit()
    result = actions.perform_fetch(hit, [hit], quality="best", root="/media/music")
    assert result["lidarr_album_id"] == 5
    assert result["path"] is None


def test_perform_lidarr_album_fallsthrough_to_youtube(monkeypatch):
    monkeypatch.setattr(mf, "act_lidarr_album",
                        lambda hit, root, search_all, dry_run: False)
    yt_calls = {}
    monkeypatch.setattr(mf, "act_youtube",
                        lambda hit, root, quality, dry_run: yt_calls.update(hit=hit))
    lidarr_hit = make_lidarr_album_hit()
    yt_hit = make_yt_hit()
    result = actions.perform_fetch(lidarr_hit, [lidarr_hit, yt_hit],
                                   quality="best", root="/media/music")
    assert yt_calls["hit"] is yt_hit
    assert result["path"] == "/media/music/Avril Lavigne/youtube"