def test_valid_email_change_token(self):
u = User(email='*****@*****.**', password='******')
db.session.add(u)
db.session.commit()
token = u.generate_token(new_email='*****@*****.**')
self.assertTrue(u.confirm_new_email(token))
self.assertTrue(u.email == '*****@*****.**')
def test_valid_reset_token(self):
u = User(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_token()
self.assertTrue(u.confirm_reset(token, 'dog'))
self.assertTrue(u.verify_password('dog'))
def test_expired_confirmation_token(self):
u = User(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_token(expiration=1)
time.sleep(2)
self.assertFalse(u.confirm_registration(token))
def test_invalid_confirmation_token(self):
u1 = User(password='******')
u2 = User(password='******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
token = u1.generate_token()
self.assertFalse(u2.confirm_registration(token))
def test_invalid_email_change_token(self):
u1 = User(email='*****@*****.**', password='******')
u2 = User(email='*****@*****.**', password='******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
token = u1.generate_token(new_email='*****@*****.**')
self.assertFalse(u2.confirm_new_email(token))
self.assertTrue(u2.email == '*****@*****.**')
def test_invalid_reset_token(self):
u1 = User(password='******')
u2 = User(password='******')
db.session.add(u1)
db.session.add(u2)
db.session.commit()
token = u1.generate_token()
self.assertFalse(u2.confirm_reset(token, 'horse'))
self.assertTrue(u2.verify_password('dog'))
def test_valid_confirmation_token(self):
u = User(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_token()
self.assertTrue(u.confirm_registration(token))
