Added timeout test

This commit is contained in:
Paul Lamere 2016-12-31 09:52:03 -05:00
parent b9af5ad719
commit d18892a99e
2 changed files with 24 additions and 0 deletions

View File

@ -0,0 +1,14 @@
# shows artist info for a URN or URL
import spotipy
import sys
import pprint
if len(sys.argv) > 1:
search_str = sys.argv[1]
else:
search_str = 'Radiohead'
sp = spotipy.Spotify(requests_timeout=.1)
result = sp.search(search_str)
pprint.pprint(result)

View File

@ -2,6 +2,7 @@
import spotipy
import unittest
import pprint
import requests
from spotipy.client import SpotifyException
@ -105,6 +106,15 @@ class TestSpotipy(unittest.TestCase):
self.assertTrue(found)
def test_search_timeout(self):
sp = spotipy.Spotify(requests_timeout=.1)
try:
results = sp.search(q='my*', type='track')
self.assertTrue(False, 'unexpected search timeout')
except requests.ReadTimeout:
self.assertTrue(True, 'expected search timeout')
def test_album_search(self):
results = self.spotify.search(q='weezer pinkerton', type='album')
self.assertTrue('albums' in results)