357 lines
10 KiB
Python
Executable file
357 lines
10 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""
|
|
Integration (not unit) tests for pylast.py
|
|
"""
|
|
import time
|
|
import unittest
|
|
|
|
import pylast
|
|
|
|
from .test_pylast import PyLastTestCase
|
|
|
|
|
|
class TestPyLastNetwork(PyLastTestCase):
|
|
|
|
def test_scrobble(self):
|
|
# Arrange
|
|
artist = "Test Artist"
|
|
title = "test title"
|
|
timestamp = self.unix_timestamp()
|
|
lastfm_user = self.network.get_user(self.username)
|
|
|
|
# Act
|
|
self.network.scrobble(artist=artist, title=title, timestamp=timestamp)
|
|
|
|
# Assert
|
|
# limit=2 to ignore now-playing:
|
|
last_scrobble = lastfm_user.get_recent_tracks(limit=2)[0]
|
|
self.assertEqual(str(last_scrobble.track.artist), str(artist))
|
|
self.assertEqual(str(last_scrobble.track.title), str(title))
|
|
self.assertEqual(str(last_scrobble.timestamp), str(timestamp))
|
|
|
|
def test_update_now_playing(self):
|
|
# Arrange
|
|
artist = "Test Artist"
|
|
title = "test title"
|
|
album = "Test Album"
|
|
track_number = 1
|
|
lastfm_user = self.network.get_user(self.username)
|
|
|
|
# Act
|
|
self.network.update_now_playing(
|
|
artist=artist, title=title, album=album, track_number=track_number)
|
|
|
|
# Assert
|
|
current_track = lastfm_user.get_now_playing()
|
|
self.assertIsNotNone(current_track)
|
|
self.assertEqual(str(current_track.title), "test title")
|
|
self.assertEqual(str(current_track.artist), "Test Artist")
|
|
|
|
def test_enable_rate_limiting(self):
|
|
# Arrange
|
|
self.assertFalse(self.network.is_rate_limited())
|
|
|
|
# Act
|
|
self.network.enable_rate_limit()
|
|
then = time.time()
|
|
# Make some network call, limit not applied first time
|
|
self.network.get_user(self.username)
|
|
# Make a second network call, limiting should be applied
|
|
self.network.get_top_artists()
|
|
now = time.time()
|
|
|
|
# Assert
|
|
self.assertTrue(self.network.is_rate_limited())
|
|
self.assertGreaterEqual(now - then, 0.2)
|
|
|
|
def test_disable_rate_limiting(self):
|
|
# Arrange
|
|
self.network.enable_rate_limit()
|
|
self.assertTrue(self.network.is_rate_limited())
|
|
|
|
# Act
|
|
self.network.disable_rate_limit()
|
|
# Make some network call, limit not applied first time
|
|
self.network.get_user(self.username)
|
|
# Make a second network call, limiting should be applied
|
|
self.network.get_top_artists()
|
|
|
|
# Assert
|
|
self.assertFalse(self.network.is_rate_limited())
|
|
|
|
def test_lastfm_network_name(self):
|
|
# Act
|
|
name = str(self.network)
|
|
|
|
# Assert
|
|
self.assertEqual(name, "Last.fm Network")
|
|
|
|
def test_geo_get_top_artists(self):
|
|
# Arrange
|
|
# Act
|
|
artists = self.network.get_geo_top_artists(
|
|
country="United Kingdom", limit=1)
|
|
|
|
# Assert
|
|
self.assertEqual(len(artists), 1)
|
|
self.assertIsInstance(artists[0], pylast.TopItem)
|
|
self.assertIsInstance(artists[0].item, pylast.Artist)
|
|
|
|
def test_geo_get_top_tracks(self):
|
|
# Arrange
|
|
# Act
|
|
tracks = self.network.get_geo_top_tracks(
|
|
country="United Kingdom", location="Manchester", limit=1)
|
|
|
|
# Assert
|
|
self.assertEqual(len(tracks), 1)
|
|
self.assertIsInstance(tracks[0], pylast.TopItem)
|
|
self.assertIsInstance(tracks[0].item, pylast.Track)
|
|
|
|
def test_network_get_top_artists_with_limit(self):
|
|
# Arrange
|
|
# Act
|
|
artists = self.network.get_top_artists(limit=1)
|
|
|
|
# Assert
|
|
self.helper_only_one_thing_in_top_list(artists, pylast.Artist)
|
|
|
|
def test_network_get_top_tags_with_limit(self):
|
|
# Arrange
|
|
# Act
|
|
tags = self.network.get_top_tags(limit=1)
|
|
|
|
# Assert
|
|
self.helper_only_one_thing_in_top_list(tags, pylast.Tag)
|
|
|
|
def test_network_get_top_tags_with_no_limit(self):
|
|
# Arrange
|
|
# Act
|
|
tags = self.network.get_top_tags()
|
|
|
|
# Assert
|
|
self.helper_at_least_one_thing_in_top_list(tags, pylast.Tag)
|
|
|
|
def test_network_get_top_tracks_with_limit(self):
|
|
# Arrange
|
|
# Act
|
|
tracks = self.network.get_top_tracks(limit=1)
|
|
|
|
# Assert
|
|
self.helper_only_one_thing_in_top_list(tracks, pylast.Track)
|
|
|
|
def test_country_top_tracks(self):
|
|
# Arrange
|
|
country = self.network.get_country("Croatia")
|
|
|
|
# Act
|
|
things = country.get_top_tracks(limit=2)
|
|
|
|
# Assert
|
|
self.helper_two_different_things_in_top_list(things, pylast.Track)
|
|
|
|
def test_country_network_top_tracks(self):
|
|
# Arrange
|
|
# Act
|
|
things = self.network.get_geo_top_tracks("Croatia", limit=2)
|
|
|
|
# Assert
|
|
self.helper_two_different_things_in_top_list(things, pylast.Track)
|
|
|
|
def test_tag_top_tracks(self):
|
|
# Arrange
|
|
tag = self.network.get_tag("blues")
|
|
|
|
# Act
|
|
things = tag.get_top_tracks(limit=2)
|
|
|
|
# Assert
|
|
self.helper_two_different_things_in_top_list(things, pylast.Track)
|
|
|
|
def test_album_data(self):
|
|
# Arrange
|
|
thing = self.network.get_album("Test Artist", "Test Album")
|
|
|
|
# Act
|
|
stringed = str(thing)
|
|
repr = thing.__repr__()
|
|
title = thing.get_title()
|
|
name = thing.get_name()
|
|
playcount = thing.get_playcount()
|
|
url = thing.get_url()
|
|
|
|
# Assert
|
|
self.assertEqual(stringed, "Test Artist - Test Album")
|
|
self.assertIn("pylast.Album('Test Artist', 'Test Album',", repr)
|
|
self.assertEqual(title, name)
|
|
self.assertIsInstance(playcount, int)
|
|
self.assertGreater(playcount, 1)
|
|
self.assertEqual(
|
|
"https://www.last.fm/music/test%2bartist/test%2balbum", url)
|
|
|
|
def test_track_data(self):
|
|
# Arrange
|
|
thing = self.network.get_track("Test Artist", "test title")
|
|
|
|
# Act
|
|
stringed = str(thing)
|
|
repr = thing.__repr__()
|
|
title = thing.get_title()
|
|
name = thing.get_name()
|
|
playcount = thing.get_playcount()
|
|
url = thing.get_url(pylast.DOMAIN_FRENCH)
|
|
|
|
# Assert
|
|
self.assertEqual(stringed, "Test Artist - test title")
|
|
self.assertIn("pylast.Track('Test Artist', 'test title',", repr)
|
|
self.assertEqual(title, "test title")
|
|
self.assertEqual(title, name)
|
|
self.assertIsInstance(playcount, int)
|
|
self.assertGreater(playcount, 1)
|
|
self.assertEqual(
|
|
"https://www.last.fm/fr/music/test%2bartist/_/test%2btitle", url)
|
|
|
|
def test_country_top_artists(self):
|
|
# Arrange
|
|
country = self.network.get_country("Ukraine")
|
|
|
|
# Act
|
|
artists = country.get_top_artists(limit=1)
|
|
|
|
# Assert
|
|
self.helper_only_one_thing_in_top_list(artists, pylast.Artist)
|
|
|
|
def test_caching(self):
|
|
# Arrange
|
|
user = self.network.get_user("RJ")
|
|
|
|
# Act
|
|
self.network.enable_caching()
|
|
tags1 = user.get_top_tags(limit=1, cacheable=True)
|
|
tags2 = user.get_top_tags(limit=1, cacheable=True)
|
|
|
|
# Assert
|
|
self.assertTrue(self.network.is_caching_enabled())
|
|
self.assertEqual(tags1, tags2)
|
|
self.network.disable_caching()
|
|
self.assertFalse(self.network.is_caching_enabled())
|
|
|
|
def test_album_mbid(self):
|
|
# Arrange
|
|
mbid = "a6a265bf-9f81-4055-8224-f7ac0aa6b937"
|
|
|
|
# Act
|
|
album = self.network.get_album_by_mbid(mbid)
|
|
album_mbid = album.get_mbid()
|
|
|
|
# Assert
|
|
self.assertIsInstance(album, pylast.Album)
|
|
self.assertEqual(album.title.lower(), "test")
|
|
self.assertEqual(album_mbid, mbid)
|
|
|
|
def test_artist_mbid(self):
|
|
# Arrange
|
|
mbid = "7e84f845-ac16-41fe-9ff8-df12eb32af55"
|
|
|
|
# Act
|
|
artist = self.network.get_artist_by_mbid(mbid)
|
|
|
|
# Assert
|
|
self.assertIsInstance(artist, pylast.Artist)
|
|
self.assertEqual(artist.name, "MusicBrainz Test Artist")
|
|
|
|
def test_track_mbid(self):
|
|
# Arrange
|
|
mbid = "ebc037b1-cc9c-44f2-a21f-83c219f0e1e0"
|
|
|
|
# Act
|
|
track = self.network.get_track_by_mbid(mbid)
|
|
track_mbid = track.get_mbid()
|
|
|
|
# Assert
|
|
self.assertIsInstance(track, pylast.Track)
|
|
self.assertEqual(track.title, "first")
|
|
self.assertEqual(track_mbid, mbid)
|
|
|
|
def test_init_with_token(self):
|
|
# Arrange/Act
|
|
try:
|
|
pylast.LastFMNetwork(
|
|
api_key=self.__class__.secrets["api_key"],
|
|
api_secret=self.__class__.secrets["api_secret"],
|
|
token="invalid",
|
|
)
|
|
except pylast.WSError as exc:
|
|
msg = str(exc)
|
|
|
|
# Assert
|
|
self.assertEqual(msg,
|
|
"Unauthorized Token - This token has not been issued")
|
|
|
|
def test_proxy(self):
|
|
# Arrange
|
|
host = "https://example.com"
|
|
port = 1234
|
|
|
|
# Act / Assert
|
|
self.network.enable_proxy(host, port)
|
|
self.assertTrue(self.network.is_proxy_enabled())
|
|
self.assertEqual(self.network._get_proxy(),
|
|
["https://example.com", 1234])
|
|
|
|
self.network.disable_proxy()
|
|
self.assertFalse(self.network.is_proxy_enabled())
|
|
|
|
def test_album_search(self):
|
|
# Arrange
|
|
album = "Nevermind"
|
|
|
|
# Act
|
|
search = self.network.search_for_album(album)
|
|
results = search.get_next_page()
|
|
|
|
# Assert
|
|
self.assertIsInstance(results, list)
|
|
self.assertIsInstance(results[0], pylast.Album)
|
|
|
|
def test_artist_search(self):
|
|
# Arrange
|
|
artist = "Nirvana"
|
|
|
|
# Act
|
|
search = self.network.search_for_artist(artist)
|
|
results = search.get_next_page()
|
|
|
|
# Assert
|
|
self.assertIsInstance(results, list)
|
|
self.assertIsInstance(results[0], pylast.Artist)
|
|
|
|
def test_track_search(self):
|
|
# Arrange
|
|
artist = "Nirvana"
|
|
track = "Smells Like Teen Spirit"
|
|
|
|
# Act
|
|
search = self.network.search_for_track(artist, track)
|
|
results = search.get_next_page()
|
|
|
|
# Assert
|
|
self.assertIsInstance(results, list)
|
|
self.assertIsInstance(results[0], pylast.Track)
|
|
|
|
def test_search_get_total_result_count(self):
|
|
# Arrange
|
|
artist = "Nirvana"
|
|
track = "Smells Like Teen Spirit"
|
|
search = self.network.search_for_track(artist, track)
|
|
|
|
# Act
|
|
total = search.get_total_result_count()
|
|
|
|
# Assert
|
|
self.assertGreater(int(total), 10000)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(failfast=True)
|