def test_fetch_first_or_none_returns_none_if_not_exists(self):
# Setup.
query = oauth_models.Token.query(oauth_models.Token.user_id == 123)
# Exercise.
found_token = persistence_utils.fetch_first_or_none(query)
# Verify.
self.assertFalse(found_token)
def test_fetch_first_or_none_returns_first_if_exists(self):
# Setup.
token1 = oauth_models.Token(user_id=123)
token1.put()
query = oauth_models.Token.query(oauth_models.Token.user_id == 123)
# Exercise.
found_token = persistence_utils.fetch_first_or_none(query)
# Verify.
self.assertTrue(found_token)
self.assertEqual(token1.key, found_token.key)
def find_by_refresh_code(cls, refresh_token):
query = Token.query(Token.refresh_token == refresh_token)
return persistence_utils.fetch_first_or_none(query)
def find_by_access_code(cls, access_token):
query = Token.query(Token.access_token == access_token)
return persistence_utils.fetch_first_or_none(query)
def find_by_client_id_and_code(cls, client_id, code):
query = Grant.query(Grant.client_id == client_id, Grant.code == code)
return persistence_utils.fetch_first_or_none(query)
def find_by_client_id(cls, client_id):
query = Client.query(Client.client_id == client_id)
return persistence_utils.fetch_first_or_none(query)
def find_by_email(cls, email):
query = User.query(User.email == email)
return persistence_utils.fetch_first_or_none(query)
def find_by_id(cls, key_id):
query = User.query(User.key == ndb.Key(User, key_id))
return persistence_utils.fetch_first_or_none(query)
