def check_and_return_user(): """ Returns a Users.user object, an ndb_user, and whether or not they exist in datastore """ user = users.get_current_user() ndb_user = None in_datastore = False if user: user_query = User.lookup_all_by_user_id(user.user_id()) if len(user_query): ndb_user = user_query[0] in_datastore = True return user, ndb_user, in_datastore
def test_user_getting_to_datastore_successfully(self): user = create_user(self.name, self.email, self.user_id) ndb_user = User.lookup_all_by_user_id(self.user_id) self.assertEqual(user.user_id, ndb_user[0].user_id)
def test_lookup_user_by_user_id(self): user = User.lookup_all_by_user_id(self.user_id) self.assertEqual(user[0].user_id, self.user_id) user2 = User.lookup_all_by_user_id(self.user_id2) self.assertEqual(user2[0].user_id, self.user_id2)
def test_looking_up_correct_user_by_id(self): user = User.lookup_all_by_user_id(self.user_id) self.assertNotEqual(user[0].user_id, self.user_id2)
def test_lookup_all_by_user_id_requires_user_id(self): with self.assertRaises(ValueError): User.lookup_all_by_user_id(None)