diff --git a/tests/test_pylast_artist.py b/tests/test_pylast_artist.py index bec278e..75c951d 100755 --- a/tests/test_pylast_artist.py +++ b/tests/test_pylast_artist.py @@ -268,6 +268,17 @@ class TestPyLastArtist(PyLastTestCase): # Assert self.assertEqual(corrected_artist_name, "Guns N' Roses") + def test_get_userplaycount(self): + # Arrange + artist = pylast.Artist("John Lennon", self.network, + username=self.username) + + # Act + playcount = artist.get_userplaycount() + + # Assert + self.assertGreaterEqual(playcount, 0) + if __name__ == '__main__': unittest.main(failfast=True) diff --git a/tests/test_pylast_track.py b/tests/test_pylast_track.py index 9f446da..61ef132 100755 --- a/tests/test_pylast_track.py +++ b/tests/test_pylast_track.py @@ -104,6 +104,62 @@ class TestPyLastTrack(PyLastTestCase): self.assertIsNotNone(wiki) self.assertGreaterEqual(len(wiki), 1) + def test_track_get_duration(self): + # Arrange + track = pylast.Track("Nirvana", "Lithium", self.network) + + # Act + duration = track.get_duration() + + # Assert + self.assertGreaterEqual(duration, 200000) + + def test_track_is_streamable(self): + # Arrange + track = pylast.Track("Nirvana", "Lithium", self.network) + + # Act + streamable = track.is_streamable() + + # Assert + self.assertFalse(streamable) + + def test_track_is_fulltrack_available(self): + # Arrange + track = pylast.Track("Nirvana", "Lithium", self.network) + + # Act + fulltrack_available = track.is_fulltrack_available() + + # Assert + self.assertFalse(fulltrack_available) + + def test_track_get_album(self): + # Arrange + track = pylast.Track("Nirvana", "Lithium", self.network) + + # Act + album = track.get_album() + print(album) + + # Assert + self.assertEqual(str(album), "Nirvana - Nevermind") + + def test_track_get_similar(self): + # Arrange + track = pylast.Track("Cher", "Believe", self.network) + + # Act + similar = track.get_similar() + + # Assert + found = False + for track in similar: + if str(track.item) == "Madonna - Vogue": + found = True + break + self.assertTrue(found) + if __name__ == '__main__': unittest.main(failfast=True) diff --git a/tests/test_pylast_user.py b/tests/test_pylast_user.py index 7613345..b64e0ee 100755 --- a/tests/test_pylast_user.py +++ b/tests/test_pylast_user.py @@ -101,6 +101,16 @@ class TestPyLastUser(PyLastTestCase): # Assert self.assertIsNone(country) + def test_user_get_country(self): + # Arrange + lastfm_user = self.network.get_user("RJ") + + # Act + country = lastfm_user.get_country() + + # Assert + self.assertEqual(str(country), "United Kingdom") + def test_user_equals_none(self): # Arrange lastfm_user = self.network.get_user(self.username)