More tests

This commit is contained in:
hugovk 2017-10-19 23:37:10 +03:00
parent 85bf9f9928
commit 8f3628a638
3 changed files with 77 additions and 0 deletions

View file

@ -268,6 +268,17 @@ class TestPyLastArtist(PyLastTestCase):
# Assert # Assert
self.assertEqual(corrected_artist_name, "Guns N' Roses") 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__': if __name__ == '__main__':
unittest.main(failfast=True) unittest.main(failfast=True)

View file

@ -104,6 +104,62 @@ class TestPyLastTrack(PyLastTestCase):
self.assertIsNotNone(wiki) self.assertIsNotNone(wiki)
self.assertGreaterEqual(len(wiki), 1) 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__': if __name__ == '__main__':
unittest.main(failfast=True) unittest.main(failfast=True)

View file

@ -101,6 +101,16 @@ class TestPyLastUser(PyLastTestCase):
# Assert # Assert
self.assertIsNone(country) 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): def test_user_equals_none(self):
# Arrange # Arrange
lastfm_user = self.network.get_user(self.username) lastfm_user = self.network.get_user(self.username)