def setUp(self):
def set_user(request):
request.user = mock.MagicMock()
request.client = mock.MagicMock()
request.client.client_id = 'mocked_client_id'
return True
self.mock_validator = mock.MagicMock()
self.mock_validator.authenticate_client.side_effect = set_user
self.addCleanup(setattr, self, 'mock_validator', mock.MagicMock())
auth_code = grant_types.AuthorizationCodeGrant(
request_validator=self.mock_validator)
password = grant_types.ResourceOwnerPasswordCredentialsGrant(
request_validator=self.mock_validator)
client = grant_types.ClientCredentialsGrant(
request_validator=self.mock_validator)
supported_types = {
'authorization_code': auth_code,
'password': password,
'client_credentials': client,
}
token = tokens.BearerToken(self.mock_validator)
self.endpoint = draft25.TokenEndpoint('authorization_code',
default_token_type=token,
grant_types=supported_types)
def setUp(self):
reload(grant_types) # grant_types indirectly import common
mock_validator = mock.MagicMock()
auth_code = grant_types.AuthorizationCodeGrant(
request_validator=mock_validator)
auth_code.save_authorization_code = mock.MagicMock()
implicit = grant_types.ImplicitGrant(request_validator=mock_validator)
implicit.save_token = mock.MagicMock()
response_types = {
'code': auth_code,
'token': implicit,
}
# TODO(ib-lundgren): Figure out where in the import order the monkey
# patching of oauthlib.common.generate_token breaks.
class MockToken(tokens.BearerToken):
def __call__(self, request, refresh_token=False):
token = super(MockToken,
self).__call__(request,
refresh_token=refresh_token)
token['access_token'] = 'abc'
return token
token = MockToken()
token.save_token = mock.MagicMock()
self.endpoint = draft25.AuthorizationEndpoint(
default_token=token, response_types=response_types)
def __init__(self, request_validator, *args, **kwargs):
auth_grant = grant_types.AuthorizationCodeGrant(request_validator)
bearer = tokens.BearerToken(request_validator)
AuthorizationEndpoint.__init__(self,
default_response_type='code',
response_types={'code': auth_grant})
TokenEndpoint.__init__(self,
default_grant_type='authorization_code',
grant_types={'authorization_code': auth_grant},
default_token_type=bearer)
ResourceEndpoint.__init__(self,
default_token='Bearer',
token_types={'Bearer': bearer})
def setUp(self):
reload(grant_types) # grant_types indirectly import common
mock_validator = mock.MagicMock()
auth_code = grant_types.AuthorizationCodeGrant(
request_validator=mock_validator)
password = grant_types.ResourceOwnerPasswordCredentialsGrant(
request_validator=mock_validator)
client = grant_types.ClientCredentialsGrant(
request_validator=mock_validator)
supported_types = {
'authorization_code': auth_code,
'password': password,
'client_credentials': client,
}
self.endpoint = draft25.TokenEndpoint(grant_types=supported_types)
def setUp(self):
self.mock_validator = mock.MagicMock()
self.addCleanup(setattr, self, 'mock_validator', mock.MagicMock())
auth_code = grant_types.AuthorizationCodeGrant(
request_validator=self.mock_validator)
auth_code.save_authorization_code = mock.MagicMock()
implicit = grant_types.ImplicitGrant(
request_validator=self.mock_validator)
implicit.save_token = mock.MagicMock()
response_types = {
'code': auth_code,
'token': implicit,
}
token = tokens.BearerToken(self.mock_validator)
self.endpoint = draft25.AuthorizationEndpoint(
default_response_type='code',
default_token_type=token,
response_types=response_types)