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')