def test_get_playlist_with_tracks_no_playlist(self):
"""
get_playlist_with_tracks() must be passed the ID of one of the
user's playlists. If it isnt', it should return None and not
change the _playlist variable.
"""
u = UserData(self.session)
playlists = u.playlists().copy()
data = u.get_playlist_with_tracks('0')
self.assertIsNone(data)
self.assertCountEqual(playlists, u.playlists())
def test_get_playlists_exists(self):
"""
If _playlists is not None, playlists() should return it.
"""
u = UserData(None)
test_data = ['test', 'hello']
u._playlists = test_data
self.assertEqual(u.playlists(), test_data)
def test_get_playlist_with_tracks(self):
"""
get_playlist_with_tracks() should request one playlist's tracks
from Spotify and save them in that playlist's entry in the
_playlist variable.
"""
u = UserData(self.session)
playlists = u.playlists()
data = u.get_playlist_with_tracks(playlists[0]['id'])
self.assertEqual(data['type'], 'playlist')
self.assertIsNotNone(data['tracks']['items'])
# Assures the data is saved in the class
self.assertIsNotNone(u.playlists()[0]['tracks']['items'])
def test_compile_playlists(self):
"""
_compile_playlists() should request compile data about the
user's playlists from Spotify and store it in the _playlists
variable.
"""
u = UserData(self.session)
data = u.playlists()
self.assertGreaterEqual(len(data), 5)
for p in data:
self.assertEqual(p['type'], 'playlist')
