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)
