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_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_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 get_user_by_key(user_key): if isinstance(user_key, ndb.Key): user = user_key.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_unque_id(user.user_id) user_obj.set_nickname(user.nickname) return user_obj else: return False
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))