|
|
# -*- coding: utf-8 -*-# Copyright 2016 SYLEAM# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import jsonimport loggingfrom .common_test_controller import OAuthProviderControllerTransactionCasefrom .common_test_oauth_provider_controller import \ TestOAuthProviderRefreshTokenController, \ TestOAuthProviderAurhorizeController, \ TestOAuthProviderTokeninfoController, \ TestOAuthProviderUserinfoController, \ TestOAuthProviderOtherinfoController, \ TestOAuthProviderRevokeTokenController
_logger = logging.getLogger(__name__)
try: import oauthlibexcept ImportError: _logger.debug('Cannot `import oauthlib`.')
class TestOAuthProviderController( OAuthProviderControllerTransactionCase, TestOAuthProviderRefreshTokenController, TestOAuthProviderAurhorizeController, TestOAuthProviderTokeninfoController, TestOAuthProviderUserinfoController, TestOAuthProviderOtherinfoController, TestOAuthProviderRevokeTokenController): def setUp(self): super(TestOAuthProviderController, self).setUp('web application')
def new_code(self): # Configure the client to skip the authorization page self.client.skip_authorization = True
# Call the authorize method with good values state = 'Some custom state' self.login('demo', 'demo') response = self.get_request('/oauth2/authorize', data={ 'client_id': self.client.identifier, 'response_type': self.client.response_type, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, 'state': state, }) # A new authorization code should have been generated # We can safely pick the latest generated code here, because no other # code could have been generated during the test code = self.env['oauth.provider.authorization.code'].search([ ('client_id', '=', self.client.id), ], order='id DESC', limit=1) # The response should be a redirect to the redirect URI, with the # authorization_code added as GET parameter self.assertEqual(response.status_code, 302) query_string = oauthlib.common.urlencode( {'state': state, 'code': code.code}.items()) self.assertEqual( response.headers['Location'], '{uri_base}?{query_string}'.format( uri_base=self.redirect_uri_base, query_string=query_string)) self.assertEqual(code.user_id, self.user)
self.logout()
return code
def test_token_error_missing_session(self): """ Check /oauth2/token without any session set
Must return an invalid_client_id error """
response = self.post_request('/oauth2/token') self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_missing_arguments(self): """ Check /oauth2/token without any argument
Must return an invalid_client_id error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token') self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_wrong_grant_type(self): """ Check /oauth2/token with an invalid grant type
Must return an invalid_client_id error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': 'Wrong grant type', }) self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_missing_code(self): """ Check /oauth2/token without code
Must return an invalid_client_id error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, }) self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_missing_client_id(self): """ Check /oauth2/token without client
Must return an invalid_client_id error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'code': 'Wrong code', }) self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_wrong_client_identifier(self): """ Check /oauth2/token with a wrong client identifier
Must return an invalid_client_id error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'client_id': 'Wrong client identifier', 'code': 'Wrong code', }) self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_wrong_code(self): """ Check /oauth2/token with a wrong code
Must return an invalid_grant error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'client_id': self.client.identifier, 'code': 'Wrong code', }) self.assertEqual(response.status_code, 401) self.assertEqual(json.loads(response.data), {'error': 'invalid_grant'})
def test_token_error_missing_redirect_uri(self): """ Check /oauth2/token without redirect_uri
Must return an access_denied error """
# Generate an authorization code code = self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'client_id': self.client.identifier, 'code': code.code, }) self.assertEqual(response.status_code, 401) self.assertEqual(json.loads(response.data), {'error': 'access_denied'})
def test_token_error_wrong_redirect_uri(self): """ Check /oauth2/token with a wrong redirect_uri
Must return an access_denied error """
# Generate an authorization code code = self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'client_id': self.client.identifier, 'code': code.code, 'redirect_uri': 'Wrong redirect URI', }) self.assertEqual(response.status_code, 401) self.assertEqual(json.loads(response.data), {'error': 'access_denied'})
def test_token_error_wrong_client_id(self): """ Check /oauth2/token with a wrong client id
Must return an invalid_client_id error """
# Generate an authorization code code = self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': self.client.grant_type, 'client_id': 'Wrong client id', 'code': code.code, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, }) self.assertEqual(response.status_code, 401) self.assertEqual( json.loads(response.data), {'error': 'invalid_client_id'})
def test_token_error_missing_refresh_token(self): """ Check /oauth2/token in refresh token mode without refresh token
Must return an invalid_request error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': 'refresh_token', 'client_id': self.client.identifier, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, }) self.assertEqual(response.status_code, 400) self.assertEqual(json.loads(response.data), { 'error': 'invalid_request', 'error_description': 'Missing refresh token parameter.', })
def test_token_error_invalid_refresh_token(self): """ Check /oauth2/token in refresh token mode with an invalid refresh token
Must return an invalid_grant error """
# Generate an authorization code to set the session self.new_code()
response = self.post_request('/oauth2/token', data={ 'grant_type': 'refresh_token', 'client_id': self.client.identifier, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, 'refresh_token': 'Wrong refresh token', }) self.assertEqual(response.status_code, 401) self.assertEqual(json.loads(response.data), {'error': 'invalid_grant'})
def test_authorize_skip_authorization(self): """ Call /oauth2/authorize while skipping the authorization page """ # Configure the client to skip the authorization page self.client.skip_authorization = True
# Call the authorize method with good values state = 'Some custom state' self.login('demo', 'demo') response = self.get_request('/oauth2/authorize', data={ 'client_id': self.client.identifier, 'response_type': self.client.response_type, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, 'state': state, }) # A new authorization code should have been generated # We can safely pick the latest generated code here, because no other # code could have been generated during the test code = self.env['oauth.provider.authorization.code'].search([ ('client_id', '=', self.client.id), ], order='id DESC', limit=1) # The response should be a redirect to the redirect URI, with the # authorization_code added as GET parameter self.assertEqual(response.status_code, 302) query_string = oauthlib.common.urlencode({ 'state': state, 'code': code.code, }.items()) self.assertEqual( response.headers['Location'], '{uri_base}?{query_string}'.format( uri_base=self.redirect_uri_base, query_string=query_string)) self.assertEqual(code.user_id, self.user)
def test_successful_token_retrieval(self): """ Check the full process for a WebApplication
GET, then POST, token and informations retrieval """
# Call the authorize method with good values to fill the session scopes # and credentials variables state = 'Some custom state' self.login('demo', 'demo') response = self.get_request('/oauth2/authorize', data={ 'client_id': self.client.identifier, 'response_type': self.client.response_type, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, 'state': state, }) self.assertEqual(response.status_code, 200) self.assertTrue(self.client.name in response.data) self.assertTrue(self.client.scope_ids[0].name in response.data) self.assertTrue(self.client.scope_ids[0].description in response.data)
# Then, call the POST route to validate the authorization response = self.post_request('/oauth2/authorize') # A new authorization code should have been generated # We can safely pick the latest generated code here, because no other # code could have been generated during the test code = self.env['oauth.provider.authorization.code'].search([ ('client_id', '=', self.client.id), ], order='id DESC', limit=1) # The response should be a redirect to the redirect URI, with the # authorization_code added as GET parameter self.assertEqual(response.status_code, 302) query_string = oauthlib.common.urlencode({ 'state': state, 'code': code.code, }.items()) self.assertEqual( response.headers['Location'], '{uri_base}?{query_string}'.format( uri_base=self.redirect_uri_base, query_string=query_string)) self.assertEqual(code.user_id, self.user)
self.logout()
# Now that the user vaidated the authorization, we can ask for a token, # using the returned code response = self.post_request('/oauth2/token', data={ 'client_id': self.client.identifier, 'redirect_uri': self.redirect_uri_base, 'scope': self.client.scope_ids[0].code, 'code': code.code, 'grant_type': self.client.grant_type, }) response_data = json.loads(response.data) # A new token should have been generated # We can safely pick the latest generated token here, because no other # token could have been generated during the test token = self.env['oauth.provider.token'].search([ ('client_id', '=', self.client.id), ], order='id DESC', limit=1) self.assertEqual(response.status_code, 200) self.assertEqual(token.token, response_data['access_token']) self.assertEqual(token.token_type, response_data['token_type']) self.assertEqual(token.refresh_token, response_data['refresh_token']) self.assertEqual(token.scope_ids, code.scope_ids) self.assertEqual(token.user_id, self.user)
|
