def register():
"""Register new user."""
if current_user.is_authenticated:
return redirect(url_for("public.home"))
form = RegisterForm(request.form)
if form.validate_on_submit():
User.create(
username=form.username.data,
email=form.email.data,
password=form.password.data,
active=True,
)
if form.code.data:
group = StudentGroups.query.filter_by(code=form.code.data).first()
user = User.query.filter_by(username=form.username.data).first()
gid = group.get_id()
uid = user.get_id()
GroupUsers.create(user_id=uid, group_id=gid)
else:
group = StudentGroups.query.filter_by(name="ALL").first()
gid = group.get_id()
user = User.query.filter_by(username=form.username.data).first()
uid = user.get_id()
GroupUsers.create(user_id=uid, group_id=gid)
flash("Thank you for registering. You can now log in.", "success")
return redirect(url_for("public.home"))
else:
flash_errors(form)
return render_template("public/register.html", form=form)
def test_validate_email_already_registered(self, user):
"""Enter email that is already registered."""
form = RegisterForm(
username="******", email=user.email, password="******", confirm="example",
)
assert form.validate() is False
assert "Email already registered" in form.email.errors
def test_validate_success(self, db):
"""Register with success."""
form = RegisterForm(username="******",
email="*****@*****.**",
password="******",
confirm="example",
code="12345678")
assert form.validate() is True
def test_validate_user_already_registered(self, user):
"""Enter username that is already registered."""
form = RegisterForm(username=user.username,
email="*****@*****.**",
password="******",
confirm="example",
code="12345678")
assert form.validate() is False
assert "Username already registered" in form.username.errors
def register():
"""Register new user."""
form = RegisterForm(request.form)
if form.validate_on_submit():
User.create(
username=form.username.data,
email=form.email.data,
password=form.password.data,
active=True,
)
flash("Thank you for registering. You can now log in.", "success")
return redirect(url_for("public.home"))
else:
flash_errors(form)
return render_template("public/register.html", form=form)