diff --git a/test/unit/test_authorization_v1.py b/test/unit/test_authorization_v1.py new file mode 100644 index 000000000..bdc2d5f65 --- /dev/null +++ b/test/unit/test_authorization_v1.py @@ -0,0 +1,17 @@ +# coding: utf-8 +import responses +import watson_developer_cloud + + +@responses.activate +def test_request_token(): + url = 'https://stream.watsonplatform.net/authorization/api/v1/token?url=https://stream.watsonplatform.net/speech-to-text/api' + responses.add(responses.GET, + url=url, + body=b'mocked token', + status=200) + authorization = watson_developer_cloud.AuthorizationV1(username='xxx', + password='yyy') + authorization.get_token(url=watson_developer_cloud.SpeechToTextV1.default_url) + assert responses.calls[0].request.url == url + assert responses.calls[0].response.content.decode('utf-8') == 'mocked token' diff --git a/watson_developer_cloud/authorization_v1.py b/watson_developer_cloud/authorization_v1.py index f1e15aaae..75cc8f248 100644 --- a/watson_developer_cloud/authorization_v1.py +++ b/watson_developer_cloud/authorization_v1.py @@ -52,4 +52,4 @@ def get_token(self, url): self.url = urlparse.urlunsplit(parsed_url) response = self.request(method='GET', url='/v1/token?url=' + url) - return response.text + return response.result.text