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