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)
