def register(): """Handle registration; show form or process results.""" form = RegisterForm() openid_form = OpenIDForm() if openid_form.validate_on_submit(): return oid.try_login( openid_form.openid.data, ask_for=['nickname', 'email'], ask_for_optional=['fullname'], ) if form.validate_on_submit(): new_user = User(username=form.username.data) new_user.set_password(form.username.data) db.session.add(new_user) db.session.commit() flash( "Your user has been created, please login.", category="success" ) return redirect(url_for('.login')) openid_errors = oid.fetch_error() if openid_errors: flash(openid_errors, category='danger') return render_template('register.html', form=form, openid_form=openid_form)
def register(): if current_user.is_authenticated: return redirect(url_for('index')) form = RegistrationForm() if form.validate_on_submit(): user = User(username=form.username.data, email=form.email.data) user.set_password(form.password.data) db.session.add(user) db.session.commit() flash(_('Congratulations, you are now a registered user!')) return redirect(url_for('login')) return render_template('register.html', title=_('Register'), form=form)
def setUp(self): app.config['TESTING'] = True app.config['WTF_CSRF_ENABLED'] = False app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + os.path.join(basedir, 'test.db') self.app = app self.ctx = self.app.app_context() self.ctx.push() db.drop_all() db.create_all() u = User(username=self.default_username) u.set_password(self.default_password) db.session.add(u) db.session.commit() self.client = TestClient(self.app, u.generate_auth_token(), '')
def register(): if current_user.is_authenticated: return redirect(url_for('index')) form = RegistrationForm() if form.validate_on_submit(): user = User(username=form.username.data, email=form.email.data) user.set_password(form.password.data) db.session.add(user) db.session.commit() flash('You have registered successfully!') return redirect(url_for('login')) return render_template('register.html', title='Register', form=form)
def setup_users(): """Add users and roles.""" poster = Role(name="poster") admin = Role(name="admin") default = Role(name="default") db.session.add_all([poster, admin, default]) for username, roles in [("joel", [poster, admin]), ("george", [poster]), ("jane", [default])]: u = User(username=username, roles=roles) u.set_password("testtest") db.session.add(u) db.session.commit()
def test_login(self): """ Tests if the login form works correctly """ test_role = Role(name="default") db.session.add(test_role) db.session.commit() test_user = User(username="******") test_user.set_password("test") db.session.add(test_user) db.session.commit() result = self.client.post('/login', data=dict(username='******', password="******"), follow_redirects=True) self.assertEqual(result.status_code, 200) self.assertIn('You have been logged in', result.data)
def test_login(self): """ Tests if the login form works correctly """ test_role = Role(name="default") db.session.add(test_role) db.session.commit() test_user = User(username="******") test_user.set_password("test") db.session.add(test_user) db.session.commit() result = self.client.post('/login', data=dict( username='******', password="******" ), follow_redirects=True) self.assertEqual(result.status_code, 200) self.assertIn('You have been logged in', result.data)
def test_password_hashing(self): u = User(username='******') u.set_password('cat') self.assertFalse(u.check_password('dog')) self.assertTrue(u.check_password('cat'))
from blogapp import create_app from blogapp.models import db, User, Role app = create_app('blogapp.config.TestConfig') db.app = app db.create_all() default = Role(name="default") poster = Role(name="poster") db.session.add(default) db.session.add(poster) db.session.commit() test_user = User(username="******") test_user.set_password("test") test_user.roles.append(poster) db.session.add(test_user) db.session.commit() app.run(port=5001, use_reloader=False)