def test_expired_confirmation_token(self):
user = User('test')
user.password = '******'
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token(1)
time.sleep(2)
self.assertFalse(user.confirm(token))
def test_valid_confirmation_token(self):
user = User('test')
user.password = '******'
db.session.add(user)
# 只有commit了才能拿到id，以便生成token
db.session.commit()
token = user.generate_confirmation_token()
self.assertTrue(user.confirm(token))
def test_invalid_confirmation_token(self):
user1 = User('test1')
user2 = User('test2')
user1.password = '******'
user2.password = '******'
db.session.add(user1)
db.session.add(user2)
db.session.commit()
token = user1.generate_confirmation_token()
self.assertFalse(user2.confirm(token))
