def test_expired_confirmation_token(self):
u = User(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_confirmation_token(1)
time.sleep(2)
self.assertFalse(u.confirm(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_confirmation_token()
self.assertFalse(u2.confirm(token))
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User(name=form.name.data,
email=form.email.data.lower(),
password=form.password.data)
db.session.add(user)
db.session.commit()
token = user.generate_confirmation_token()
send_mail(user.email, '激活账号', 'mail/confirm', user=user, token=token)
flash('一封确认邮件已经发送到您的邮箱，请及时激活账号！')
return redirect(url_for('main.hello_world'))
return render_template('auth/register.html', form=form)
def register():
form = RegisterForm()
if form.validate_on_submit():
user = User(username=form.username.data,
email=form.email.data.lower(),
password=form.password.data)
try:
db.session.add(user)
db.session.commit()
except:
db.session.rollback()
flash('注册失败,请重试')
return render_template('auth/register.html', form=form)
token = user.generate_confirmation_token()
send_mail(user.email,
'激活账号',
'mail/confirm',
user=user,
token=token,
email=user.email)
flash('一封确认邮件已经发送到您的邮箱，请及时激活账号！')
return redirect(url_for('main.index'))
return render_template('auth/register.html', form=form)
def test_valid_confirmation_token(self):
u = User(password='******')
db.session.add(u)
db.session.commit()
token = u.generate_confirmation_token()
self.assertTrue(u.confirm(token))
