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 post(self):
"""Handles the POST call to the CreateUserAPI."""
data = self.parser.parse_args()
username = data.get('username')
password = data.get('password')
if User.user_exist(username):
abort(400, 'SignUpFailed: A User with the specified username '
'already exist')
new_user = User(username=username, password=password)
new_user.save()
token = new_user.generate_auth_token()
return {'token': token.decode('utf-8')}, 201
def post(self):
"""Handles the POST call to the CreateUserAPI."""
data = self.parser.parse_args()
username = data.get('username')
password = data.get('password')
if User.user_exist(username):
abort(
400, 'SignUpFailed: A User with the specified username '
'already exist')
new_user = User(username=username, password=password)
new_user.save()
token = new_user.generate_auth_token()
return {'token': token.decode('utf-8')}, 201