def test_get_playlist_with_tracks_exists(self):
"""
If the playlist with the given ID exists in _playlists and it
already has a list of tracks, get_playlist_with_tracks() should
return that playlist data.
"""
u = UserData(None)
test_data = [{'id': '0', 'tracks': {'items': []}}]
u._playlists = test_data
self.assertEqual(u.get_playlist_with_tracks('0'), test_data[0])
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_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'])
