mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 09:50:16 +00:00
117 lines
4.5 KiB
Python
117 lines
4.5 KiB
Python
from spotdl.lyrics import LyricBase
|
|
from spotdl.lyrics import exceptions
|
|
from spotdl.lyrics.providers import Genius
|
|
|
|
import urllib.request
|
|
import json
|
|
import pytest
|
|
|
|
class TestGenius:
|
|
def test_subclass(self):
|
|
assert issubclass(Genius, LyricBase)
|
|
|
|
@pytest.fixture(scope="module")
|
|
def expect_lyrics_count(self):
|
|
# This is the number of characters in lyrics found
|
|
# for the track in `lyric_url` fixture below
|
|
return 1845
|
|
|
|
@pytest.fixture(scope="module")
|
|
def genius(self):
|
|
return Genius()
|
|
|
|
def test_base_url(self, genius):
|
|
assert genius.base_url == "https://genius.com"
|
|
|
|
@pytest.fixture(scope="module")
|
|
def artist(self):
|
|
return "selena gomez"
|
|
|
|
@pytest.fixture(scope="module")
|
|
def track(self):
|
|
return "wolves"
|
|
|
|
@pytest.fixture(scope="module")
|
|
def query(self, artist, track):
|
|
return "{} {}".format(artist, track)
|
|
|
|
@pytest.fixture(scope="module")
|
|
def guess_url(self, query):
|
|
return "https://genius.com/selena-gomez-wolves-lyrics"
|
|
|
|
@pytest.fixture(scope="module")
|
|
def lyric_url(self):
|
|
return "https://genius.com/Selena-gomez-and-marshmello-wolves-lyrics"
|
|
|
|
def test_guess_lyric_url_from_artist_and_track(self, genius, artist, track, guess_url):
|
|
url = genius.guess_lyric_url_from_artist_and_track(artist, track)
|
|
assert url == guess_url
|
|
|
|
class MockHTTPResponse:
|
|
expect_lyrics = ""
|
|
|
|
def __init__(self, request, timeout=None):
|
|
search_results_url = "https://genius.com/api/search/multi?per_page=1&q=selena%2Bgomez%2Bwolves"
|
|
if request._full_url == search_results_url:
|
|
read_method = lambda: json.dumps({
|
|
"response": {"sections": [{"hits": [{"result": {
|
|
"path": "/Selena-gomez-and-marshmello-wolves-lyrics"
|
|
} }] }] }
|
|
})
|
|
else:
|
|
read_method = lambda: "<p>" + self.expect_lyrics + "</p>"
|
|
|
|
self.read = read_method
|
|
|
|
@pytest.mark.network
|
|
def test_best_matching_lyric_url_from_query(self, genius, query, lyric_url):
|
|
url = genius.best_matching_lyric_url_from_query(query)
|
|
assert url == lyric_url
|
|
|
|
def test_mock_best_matching_lyric_url_from_query(self, genius, query, lyric_url, monkeypatch):
|
|
monkeypatch.setattr("urllib.request.urlopen", self.MockHTTPResponse)
|
|
self.test_best_matching_lyric_url_from_query(genius, query, lyric_url)
|
|
|
|
@pytest.mark.network
|
|
def test_from_url(self, genius, lyric_url, expect_lyrics_count):
|
|
lyrics = genius.from_url(lyric_url)
|
|
assert len(lyrics) == expect_lyrics_count
|
|
|
|
def test_mock_from_url(self, genius, lyric_url, expect_lyrics_count, monkeypatch):
|
|
self.MockHTTPResponse.expect_lyrics = "a" * expect_lyrics_count
|
|
monkeypatch.setattr("urllib.request.urlopen", self.MockHTTPResponse)
|
|
self.test_from_url(genius, lyric_url, expect_lyrics_count)
|
|
|
|
@pytest.mark.network
|
|
def test_from_artist_and_track(self, genius, artist, track, expect_lyrics_count):
|
|
lyrics = genius.from_artist_and_track(artist, track)
|
|
assert len(lyrics) == expect_lyrics_count
|
|
|
|
def test_mock_from_artist_and_track(self, genius, artist, track, expect_lyrics_count, monkeypatch):
|
|
self.MockHTTPResponse.expect_lyrics = "a" * expect_lyrics_count
|
|
monkeypatch.setattr("urllib.request.urlopen", self.MockHTTPResponse)
|
|
self.test_from_artist_and_track(genius, artist, track, expect_lyrics_count)
|
|
|
|
@pytest.mark.network
|
|
def test_from_query(self, genius, query, expect_lyrics_count):
|
|
lyrics = genius.from_query(query)
|
|
assert len(lyrics) == expect_lyrics_count
|
|
|
|
def test_mock_from_query(self, genius, query, expect_lyrics_count, monkeypatch):
|
|
self.MockHTTPResponse.expect_lyrics = "a" * expect_lyrics_count
|
|
monkeypatch.setattr("urllib.request.urlopen", self.MockHTTPResponse)
|
|
self.test_from_query(genius, query, expect_lyrics_count)
|
|
|
|
@pytest.mark.network
|
|
def test_lyrics_not_found_error(self, genius):
|
|
with pytest.raises(exceptions.LyricsNotFoundError):
|
|
genius.from_artist_and_track(self, "nonexistent_artist", "nonexistent_track")
|
|
|
|
def test_mock_lyrics_not_found_error(self, genius, monkeypatch):
|
|
def mock_urlopen(url, timeout=None):
|
|
raise urllib.request.HTTPError("", "", "", "", "")
|
|
|
|
monkeypatch.setattr("urllib.request.urlopen", mock_urlopen)
|
|
self.test_lyrics_not_found_error(genius)
|
|
|