diff --git a/test_pylast.py b/test_pylast.py new file mode 100644 index 0000000..0942d3b --- /dev/null +++ b/test_pylast.py @@ -0,0 +1,86 @@ +#!/usr/bin/env python +""" +Integration (not unit) tests for pylast.py +""" +import datetime +import time +import unittest + +import pylast + +class TestSequenceFunctions(unittest.TestCase): + + def setUp(self): + self.username = "TODO" + password_hash = "TODO" + + API_KEY = "TODO" + API_SECRET = "TODO" + + self.network = pylast.LastFMNetwork(api_key = API_KEY, api_secret = + API_SECRET, username = self.username, password_hash = password_hash) + + + def test_scrobble(self): + # Arrange + artist = "Test Artist" + title = "Test Title" + timestamp = int(time.mktime(datetime.datetime.now().timetuple())) + lastfm_user = self.network.get_user(self.username) + + # Act + self.network.scrobble(artist = artist, title = title, timestamp = timestamp) + + # Assert + last_scrobble = lastfm_user.get_recent_tracks(limit = 1)[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_unscrobble(self): + # Arrange + artist = "Test Artist 2" + title = "Test Title 2" + timestamp = int(time.mktime(datetime.datetime.now().timetuple())) + library = pylast.Library(user = self.username, network = self.network) + self.network.scrobble(artist = artist, title = title, timestamp = timestamp) + lastfm_user = self.network.get_user(self.username) + + # Act + library.remove_scrobble(artist = artist, title = title, timestamp = timestamp) + + # Assert + last_scrobble = lastfm_user.get_recent_tracks(limit = 1)[0] + self.assertNotEqual(str(last_scrobble.timestamp), str(timestamp)) + + + def test_add_album(self): + # Arrange + library = pylast.Library(user = self.username, network = self.network) + album = self.network.get_album("Test Artist", "Test Album") + + # Act + library.add_album(album) + + # Assert + # Nothing here, just that no exception occurred + + + def test_get_venue(self): + # Arrange + venue_name = "Last.fm Office" + country_name = "United Kingom" + + # Act + venue_search = self.network.search_for_venue(venue_name, country_name) + venue = venue_search.get_next_page()[0] + + # Assert + self.assertEqual(str(venue.id), "8778225") + + +if __name__ == '__main__': + unittest.main() + +# End of file