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))