def test_load_grant_returns_existed(self): # Setup. grant = oauth_models.Grant(client_id=constants.CLIENT_ID, code='test_grant') grant.put() # Exercise. loaded_grant = oauth_api.load_grant(constants.CLIENT_ID, 'test_grant') # Verify self.assertEqual(grant.key, loaded_grant.key)
def test_save_grant_saves_to_db(self): # Setup. request = mock.MagicMock() request.scopes = 'email' code_dict = {'code': 'test_grant'} # Exercise. saved_grant = oauth_api.save_grant(constants.CLIENT_ID, code_dict, request) # Verify found_grant = oauth_api.load_grant(constants.CLIENT_ID, 'test_grant') self.assertEqual(saved_grant.key, found_grant.key)