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