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)