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))