def test_no_shared_tokens(self):
auth = CognitoAuthentication()
request = mock.MagicMock()
request.session = {
'access_token': 'test_access_token',
'refresh_token': 'test_refresh_token',
'id_token': 'test_id_token',
}
token = auth.get_token(request, 'access_token')
self.assertEqual(token, 'test_access_token')
def test_shared_tokens_no_session_token_present(self):
auth = CognitoAuthentication()
request = mock.MagicMock()
request.session = {}
request.COOKIES = {
'access_token': 'test_access_token_shared',
'refresh_token': 'test_refresh_token_shared',
'id_token': 'test_id_token_shared',
}
settings.SHARED_TOKENS = True
settings.SHARED_TOKENS_DOMAIN = '.test.com'
token = auth.get_token(request, 'access_token')
self.assertEqual(token, 'test_access_token_shared')