class Testmodels(unittest.TestCase): def setUp(self): self.app = create_app(TestConfig) self.app.app_context().push() db.create_all() self.user = User(username="******", password="******") self.user.save() self.token = self.user.generate_auth_token() def test_user_model(self): self.assertGreater(self.user.id, 0) def test_verify_password(self): self.assertTrue(self.user.verify_password("wahab")) def test_get_user(self): user = User.get_user("tester", "wahab") self.assertIsNotNone(user) def test_verify_token(self): self.assertTrue(User.verify_token(self.token)) def test_verify_invalid_token(self): invalid_token = "this-is-no-way-a-token" self.assertFalse(User.verify_token(invalid_token)) def test_expired_token(self): token = self.user.generate_auth_token(expiration=1) time.sleep(2) self.assertFalse(self.user.verify_token(token)) def test_user_exist(self): self.assertTrue(User.user_exist("tester"))
def authenticate_token(token, password): """Autheticate User with the provideded token.""" user = User.verify_token(token) if user: g.user = user return True return False
def test_verify_invalid_token(self): invalid_token = "this-is-no-way-a-token" self.assertFalse(User.verify_token(invalid_token))
def test_verify_token(self): self.assertTrue(User.verify_token(self.token))