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'])