75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
import server.mf # noqa: F401
|
|
import musicfetch_core as mf
|
|
|
|
|
|
def test_pure_playlist_url_is_playlist():
|
|
assert mf.is_playlist_url("https://music.youtube.com/playlist?list=PLabc") is True
|
|
assert mf.is_playlist_url("https://www.youtube.com/playlist?list=PLabc") is True
|
|
|
|
|
|
def test_watch_with_list_is_not_playlist():
|
|
assert mf.is_playlist_url("https://www.youtube.com/watch?v=abc&list=PLx") is False
|
|
|
|
|
|
def test_plain_watch_is_not_playlist():
|
|
assert mf.is_playlist_url("https://www.youtube.com/watch?v=abc") is False
|
|
|
|
|
|
def test_non_url_is_not_playlist():
|
|
assert mf.is_playlist_url("Daft Punk - Discovery") is False
|
|
|
|
|
|
class _CP:
|
|
def __init__(self, stdout):
|
|
self.stdout = stdout
|
|
self.returncode = 0
|
|
|
|
|
|
def test_expand_playlist_ytdlp_fallback(monkeypatch):
|
|
import json as _json
|
|
monkeypatch.setattr(mf, "YTMusic", None)
|
|
payload = {"title": "My Mix", "entries": [
|
|
{"id": "v1", "title": "Song One", "uploader": "Artist A"},
|
|
{"id": "v2", "title": "Song Two", "channel": "Artist B"},
|
|
{"id": None, "title": "skip"},
|
|
]}
|
|
monkeypatch.setattr(mf.subprocess, "run", lambda *a, **k: _CP(_json.dumps(payload)))
|
|
title, hits = mf.expand_playlist("https://www.youtube.com/playlist?list=PLx")
|
|
assert title == "My Mix"
|
|
assert [h.payload["videoId"] for h in hits] == ["v1", "v2"]
|
|
assert hits[0].artist == "Artist A"
|
|
|
|
|
|
def test_download_playlist_counts_ok_and_total(monkeypatch):
|
|
h1 = mf.Hit(source="youtube", kind="track", title="A", artist="X", payload={"videoId": "1"})
|
|
h2 = mf.Hit(source="youtube", kind="track", title="B", artist="Y", payload={"videoId": "2"})
|
|
h3 = mf.Hit(source="youtube", kind="track", title="C", artist="Z", payload={"videoId": "3"})
|
|
monkeypatch.setattr(mf, "expand_playlist", lambda url: ("PL Title", [h1, h2, h3]))
|
|
monkeypatch.setattr(mf, "act_youtube", lambda hit, root, quality, dry_run: hit.title != "B")
|
|
ok, total, title = mf.download_playlist("u", "/tmp", "best", False)
|
|
assert (ok, total, title) == (2, 3, "PL Title")
|
|
|
|
|
|
def test_download_playlist_track_exception_counts_as_failure(monkeypatch):
|
|
h1 = mf.Hit(source="youtube", kind="track", title="A", artist="X", payload={"videoId": "1"})
|
|
h2 = mf.Hit(source="youtube", kind="track", title="B", artist="Y", payload={"videoId": "2"})
|
|
monkeypatch.setattr(mf, "expand_playlist", lambda url: ("T", [h1, h2]))
|
|
|
|
def fake_act(hit, root, quality, dry_run):
|
|
if hit.title == "B":
|
|
raise RuntimeError("boom")
|
|
return True
|
|
monkeypatch.setattr(mf, "act_youtube", fake_act)
|
|
ok, total, _ = mf.download_playlist("u", "/tmp", "best", False)
|
|
assert (ok, total) == (1, 2)
|
|
|
|
|
|
def test_yt_download_returns_true_on_zero_exit(monkeypatch):
|
|
monkeypatch.setattr(mf.os, "makedirs", lambda *a, **k: None)
|
|
monkeypatch.setattr(mf.subprocess, "run", lambda *a, **k: _CP(""))
|
|
assert mf.yt_download("u", "/tmp/x", "best", False) is True
|
|
|
|
|
|
def test_yt_download_dry_run_returns_true(monkeypatch):
|
|
assert mf.yt_download("u", "/tmp/x", "best", True) is True
|