Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-22 00:23:33 +01:00
Code Issues Projects Releases Wiki Activity

try oauth token mock

Browse source
This commit is contained in:
Rahul Ahuja 2019-01-19 23:21:02 -08:00
parent e6c8f79a07
commit dc77943da2
2 changed files with 45 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
beetsplug/spotify.py
View file

@ -64,13 +64,16 @@ class SpotifyPlugin(BeetsPlugin):
self.access_token = token_data['access_token']
def authenticate(self):
b64_encoded = base64.b64encode(
':'.join(
self.config[k].as_str() for k in ('client_id', 'client_secret')
).encode()
).decode()
headers = {'Authorization': 'Basic {}'.format(b64_encoded)}
headers = {
'Authorization': 'Basic {}'.format(
base64.b64encode(
':'.join(
self.config[k].as_str()
for k in ('client_id', 'client_secret')
).encode()
).decode()
)
}
response = requests.post(
self.oauth_token_url,
data={'grant_type': 'client_credentials'},

49
test/test_spotify.py
View file

@ -29,10 +29,21 @@ def _params(url):
class SpotifyPluginTest(_common.TestCase, TestHelper):
@responses.activate
def setUp(self):
config.clear()
self.setup_beets()
responses.add(
responses.POST,
spotify.SpotifyPlugin.oauth_token_url,
status=200,
json={
'access_token': '3XyiC3raJySbIAV5LVYj1DaWbcocNi3LAJTNXRnYYGVUl6mbbqXNhW3YcZnQgYXNWHFkVGSMlc0tMuvq8CF',
'token_type': 'Bearer',
'expires_in': 3600,
'scope': '',
},
)
self.spotify = spotify.SpotifyPlugin()
opts = ArgumentsMock("list", False)
self.spotify.parse_opts(opts)
@ -51,20 +62,25 @@ class SpotifyPluginTest(_common.TestCase, TestHelper):
@responses.activate
def test_missing_request(self):
json_file = os.path.join(_common.RSRC, b'spotify',
b'missing_request.json')
json_file = os.path.join(
_common.RSRC, b'spotify', b'missing_request.json'
)
with open(json_file, 'rb') as f:
response_body = f.read()
responses.add(responses.GET, 'https://api.spotify.com/v1/search',
body=response_body, status=200,
content_type='application/json')
responses.add(
responses.GET,
spotify.SpotifyPlugin.base_url,
body=response_body,
status=200,
content_type='application/json',
)
item = Item(
mb_trackid=u'01234',
album=u'lkajsdflakjsd',
albumartist=u'ujydfsuihse',
title=u'duifhjslkef',
length=10
length=10,
)
item.add(self.lib)
self.assertEqual([], self.spotify.query_spotify(self.lib, u""))
@ -78,21 +94,25 @@ class SpotifyPluginTest(_common.TestCase, TestHelper):
@responses.activate
def test_track_request(self):
json_file = os.path.join(_common.RSRC, b'spotify',
b'track_request.json')
json_file = os.path.join(
_common.RSRC, b'spotify', b'track_request.json'
)
with open(json_file, 'rb') as f:
response_body = f.read()
responses.add(responses.GET, 'https://api.spotify.com/v1/search',
body=response_body, status=200,
content_type='application/json')
responses.add(
responses.GET,
'https://api.spotify.com/v1/search',
body=response_body,
status=200,
content_type='application/json',
)
item = Item(
mb_trackid=u'01234',
album=u'Despicable Me 2',
albumartist=u'Pharrell Williams',
title=u'Happy',
length=10
length=10,
)
item.add(self.lib)
results = self.spotify.query_spotify(self.lib, u"Happy")
@ -111,5 +131,6 @@ class SpotifyPluginTest(_common.TestCase, TestHelper):
def suite():
return unittest.TestLoader().loadTestsFromName(__name__)
if __name__ == '__main__':
unittest.main(defaultTest='suite')