def test_validate_invite_code_used_already(self, user):
department = Department.create(name="dept", short_name="DPD", load_defaults=False)
Invite_Code.create(department_id=department.id, code="code1", used=True)
form = RegisterForm(username='******', email='*****@*****.**', password='******', confirm='example', invite_code="code1")
assert form.validate() is False
assert 'Invite Code has already been used.' in form.invite_code.errors
def register():
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
invite_code = Invite_Code.query.filter_by(
code=form.invite_code.data).first()
invite_code.used = True
invite_code.save()
new_user = User.create(username=form.username.data,
email=form.email.data,
password=form.password.data,
active=True)
new_user.departments.append(
Department.get_by_id(invite_code.department_id))
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_user_already_registered(self, user):
# Enters username that is already registered
form = RegisterForm(username=user.username, email='*****@*****.**', password='******', confirm='example', invite_code="bad")
assert form.validate() is False
assert 'Username already registered' in form.username.errors
def test_validate_success(self, db):
department = Department.create(name="dept", short_name="DPD", load_defaults=False)
Invite_Code.create(department_id=department.id, code="code")
form = RegisterForm(username='******', email='*****@*****.**', password='******', confirm='example', invite_code="code")
assert form.validate() is True
def test_validate_invite_code_valid(self, user):
form = RegisterForm(username='******', email='*****@*****.**', password='******', confirm='example', invite_code="bad")
assert form.validate() is False
assert 'Invite Code not recognized.' in form.invite_code.errors
def test_validate_email_already_registered(self, user):
# enters email that is already registered
form = RegisterForm(username='******', email=user.email, password='******', confirm='example', invite_code="bad")
assert form.validate() is False
assert 'Email already registered' in form.email.errors
