mirror of
https://github.com/c0de-archive/spotipy.git
synced 2024-11-04 23:17:48 +00:00
143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
# -*- coding: latin-1 -*-
|
|
|
|
import spotipy
|
|
from spotipy import util
|
|
import unittest
|
|
import pprint
|
|
import sys
|
|
|
|
'''
|
|
Since these tests require authentication they are maintained
|
|
separately from the other tests.
|
|
|
|
These tests try to be benign and leave your collection and
|
|
playlists in a relatively stable state.
|
|
'''
|
|
|
|
class AuthTestSpotipy(unittest.TestCase):
|
|
'''
|
|
These tests require user authentication
|
|
'''
|
|
|
|
|
|
playlist = "spotify:user:plamere:playlist:2oCEWyyAPbZp9xhVSxZavx"
|
|
four_tracks = ["spotify:track:6RtPijgfPKROxEzTHNRiDp",
|
|
"spotify:track:7IHOIqZUUInxjVkko181PB",
|
|
"4VrWlk8IQxevMvERoX08iC",
|
|
"http://open.spotify.com/track/3cySlItpiPiIAzU3NyHCJf"]
|
|
|
|
two_tracks = ["spotify:track:6RtPijgfPKROxEzTHNRiDp",
|
|
"spotify:track:7IHOIqZUUInxjVkko181PB"]
|
|
|
|
other_tracks=["spotify:track:2wySlB6vMzCbQrRnNGOYKa", "spotify:track:29xKs5BAHlmlX1u4gzQAbJ",
|
|
"spotify:track:1PB7gRWcvefzu7t3LJLUlf"]
|
|
|
|
bad_id = 'BAD_ID'
|
|
|
|
def test_track_bad_id(self):
|
|
try:
|
|
track = spotify.track(self.bad_id)
|
|
self.assertTrue(False)
|
|
except spotipy.SpotifyException:
|
|
self.assertTrue(True)
|
|
|
|
|
|
def test_basic_user_profile(self):
|
|
user = spotify.user(username)
|
|
self.assertTrue(user['id'] == username)
|
|
|
|
def test_current_user(self):
|
|
user = spotify.current_user()
|
|
self.assertTrue(user['id'] == username)
|
|
|
|
def test_me(self):
|
|
user = spotify.me()
|
|
self.assertTrue(user['id'] == username)
|
|
|
|
|
|
def test_user_playlists(self):
|
|
playlists = spotify.user_playlists(username, limit=5)
|
|
self.assertTrue('items' in playlists)
|
|
|
|
# known API issue currently causes this test to fail
|
|
self.assertTrue(len(playlists['items']) == 5)
|
|
|
|
def test_current_user_saved_tracks(self):
|
|
tracks = spotify.current_user_saved_tracks()
|
|
self.assertTrue(len(tracks['items']) > 0)
|
|
|
|
def test_current_user_save_and_unsave_tracks(self):
|
|
tracks = spotify.current_user_saved_tracks()
|
|
total = tracks['total']
|
|
|
|
spotify.current_user_saved_tracks_add(self.four_tracks)
|
|
|
|
tracks = spotify.current_user_saved_tracks()
|
|
new_total = tracks['total']
|
|
self.assertTrue(new_total - total == len(self.four_tracks))
|
|
|
|
tracks = spotify.current_user_saved_tracks_delete(self.four_tracks)
|
|
tracks = spotify.current_user_saved_tracks()
|
|
new_total = tracks['total']
|
|
self.assertTrue(new_total == total)
|
|
|
|
def test_user_playlist_ops(self):
|
|
# create empty playlist
|
|
if False:
|
|
# this creates a new playlist every time, so for now
|
|
# use a known id
|
|
playlist = spotify.user_playlist_create(username, 'spotipy-testing-playlist')
|
|
playlist_id = playlist['uri']
|
|
else:
|
|
playlist_id = "spotify:user:plamere:playlist:4csW04zSLNZYFsH7qk5X3t"
|
|
|
|
# remove all tracks from it
|
|
|
|
spotify.user_playlist_replace_tracks(username, playlist_id,[])
|
|
|
|
playlist = spotify.user_playlist(username, playlist_id)
|
|
self.assertTrue(playlist['tracks']['total'] == 0)
|
|
self.assertTrue(len(playlist['tracks']['items']) == 0)
|
|
|
|
# add tracks to it
|
|
|
|
spotify.user_playlist_add_tracks(username, playlist_id, self.four_tracks)
|
|
playlist = spotify.user_playlist(username, playlist_id)
|
|
self.assertTrue(playlist['tracks']['total'] == 4)
|
|
self.assertTrue(len(playlist['tracks']['items']) == 4)
|
|
|
|
# remove two tracks from it
|
|
|
|
spotify.user_playlist_remove_all_occurrences_of_tracks (username,
|
|
playlist_id, self.two_tracks)
|
|
|
|
playlist = spotify.user_playlist(username, playlist_id)
|
|
self.assertTrue(playlist['tracks']['total'] == 2)
|
|
self.assertTrue(len(playlist['tracks']['items']) == 2)
|
|
|
|
# replace with 3 other tracks
|
|
spotify.user_playlist_replace_tracks(username,
|
|
playlist_id, self.other_tracks)
|
|
|
|
playlist = spotify.user_playlist(username, playlist_id)
|
|
self.assertTrue(playlist['tracks']['total'] == 3)
|
|
self.assertTrue(len(playlist['tracks']['items']) == 3)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) > 0:
|
|
username = sys.argv[1]
|
|
del sys.argv[1]
|
|
|
|
scope = 'playlist-modify-public '
|
|
scope += 'user-library-read '
|
|
scope += 'user-library-modify '
|
|
scope += 'user-read-private'
|
|
|
|
token = util.prompt_for_user_token(username, scope)
|
|
spotify = spotipy.Spotify(auth=token)
|
|
spotify.trace = False
|
|
unittest.main()
|
|
else:
|
|
print "Usage: %s username" % (sys.argv[0],)
|