Hirad/pylast
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add test for adding artist, and allow artist as Artist or string

Browse source
This commit is contained in:
hugovk 2014-03-01 23:54:45 +02:00
parent 7eb640fa88
commit bdb429cc8a
2 changed files with 22 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pylast.py
View file

@ -2032,7 +2032,10 @@ class Library(_BaseObject):
"""Add an artist to this library.""" """Add an artist to this library."""
params = self._get_params() params = self._get_params()
params["artist"] = artist.get_name() if type(artist) == str:
params["artist"] = artist
else:
params["artist"] = artist.get_name()
self._request("library.addArtist", False, params) self._request("library.addArtist", False, params)

20
test_pylast.py
View file

@ -104,7 +104,23 @@ class TestPyLast(unittest.TestCase):
self.assertFalse(value) self.assertFalse(value)
def test_get_venue(self): def test_add_artist(self):
# Arrange
artist = "Test Artist 2"
library = pylast.Library(user = self.username, network = self.network)
# Act
library.add_artist(artist)
# Assert
artists = library.get_artists()
for artist in artists:
value = (str(artist[0]) == "Test Artist 2")
if value:
break
self.assertTrue(value)
# Arrange # Arrange
venue_name = "Last.fm Office" venue_name = "Last.fm Office"
country_name = "United Kingom" country_name = "United Kingom"
@ -288,7 +304,7 @@ class TestPyLast(unittest.TestCase):
if __name__ == '__main__': if __name__ == '__main__':
# suite = unittest.TestSuite() # suite = unittest.TestSuite()
# suite.addTest(TestPyLast('test_remove_album')) # suite.addTest(TestPyLast('test_add_artist'))
# unittest.TextTestRunner().run(suite) # unittest.TextTestRunner().run(suite)
unittest.main() unittest.main()