def test_get_by_id(self):
"""Get user by ID."""
user = User("foo", "*****@*****.**")
user.save()
retrieved = User.get_by_id(user.id)
assert retrieved == user
def reviewer_register():
form = ReviewerRegisterForm(request.form)
if form.validate_on_submit():
created_user = User.create(username=form.username.data, email=form.email.data, password=form.password.data, user_group=2, active=True)
Reviewer.create(name=form.reviewer_name.data, user=created_user, user_id=created_user.id, department=form.reviewer_department.data)
flash('Reviewer created successfully', 'success')
return redirect(url_for('admin.reviewer_manage_panel'))
else:
flash_errors(form)
return render_template('admin/reviewer_register.html', form=form)
def register():
"""
Register new team. It creates a new User entry and a new Team entry, then links the two.
"""
# User.create(username='******', email='*****@*****.**', password='******', user_group=1, active=True)
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
created_user = User.create(username=form.username.data, email=form.email.data, password=form.password.data, user_group=3, active=True)
Team.create(name=form.team_name.data, user=created_user, user_id=created_user.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 load_user(user_id):
return User.get_by_id(int(user_id))
def test_check_password(self):
"""Check password."""
user = User.create(username="******", email="*****@*****.**", password="******")
assert user.check_password("foobarbaz123") is True
assert user.check_password("barfoobaz") is False
def test_password_is_nullable(self):
"""Test null password."""
user = User(username="******", email="*****@*****.**")
user.save()
assert user.password is None
def test_created_at_defaults_to_datetime(self):
"""Test creation date."""
user = User(username="******", email="*****@*****.**")
user.save()
assert bool(user.created_at)
assert isinstance(user.created_at, dt.datetime)
