def test_auth_token_generation(self):
wrong_email_data = dict(email='*****@*****.**', password='******')
wrong_pass_data = dict(email='*****@*****.**', password='******')
correct_data = dict(email='*****@*****.**', password='******')
with self.assertRaises(NotFound):
User.generate_auth_token(**wrong_email_data)
with self.assertRaises(BadRequest):
User.generate_auth_token(**wrong_pass_data)
self.assertTrue(User.generate_auth_token(**correct_data))
def test_verify_auth_token(self):
correct_data = dict(email='*****@*****.**', password='******')
token = User.generate_auth_token(**correct_data)
self.assertEqual(self.user, User.verify_auth_token(token))
with self.assertRaises(BadSignature):
User.verify_auth_token(token[:-5])