def test_user_add_user(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyau')
usr.set_user_id('SomeUserid1234au')
ret = ps.add_user(usr)
self.assertTrue(isinstance(ret, ndb.Key))
def test_user_remove_user_by_key_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyruks')
usr.set_user_id('SomeUserid1234ruks')
ukey = ps.add_user(usr)
self.assertTrue(ps.remove_user_by_key(ukey))
def test_user_get_user_key_by_id(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyguki')
usr.set_user_id('SomeUserid1234guki')
key = ps.add_user(usr)
self.assertEqual(key,ps.get_user_key_by_id("SomeUserid1234guki"))
def test_user_get_user_by_key_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodyguks')
usr.set_user_id('SomeUserid1234guks')
key = ps.add_user(usr)
self.assertEqual(ps.get_user_by_key(key).email(), usr.email())
def test_user_modify_user_by_id_succeed(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodymids')
usr.set_user_id('SomeUserid1234mids')
ps.add_user(usr)
newusr = User('*****@*****.**')
newusr.set_nickname('SomebodyElsemids')
newusr.set_user_id('SomeUotherId3878mids')
key = ps.modify_user_by_id('SomeUserid1234mids',newusr)
self.assertEqual(ps.get_user_by_key(key).email(), newusr.email())
def get_user_by_id(user_id): users = Users.query(Users.unique_id == user_id) user = users.get() if user is None: ## user not found return None else: user_obj = User(user.email_address) ## return user object if details were retrieved from Google if user_obj.found: return user_obj else: ## set unknown details from records user_obj.set_user_id(user.user_id) user_obj.set_nickname(user.nickname) return user_obj
def test_user_modify_user_by_id_fail(self):
usr = User('*****@*****.**')
usr.set_nickname('Somebodymidf')
usr.set_user_id('SomeUserid1234midf')
self.assertFalse(ps.modify_user_by_id("AWSEDRFGYHUJUG23151",usr))
