def test_password_is_nullable(self):
"""
Test null password.
"""
user = User(username="******", email="*****@*****.**")
user.save()
assert user.password is None
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 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)
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_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 register():
"""Register new user."""
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
user_id = (db.session.query(func.max(User.user_id)).scalar() or 0) + 1
User.create(user_id=user_id,
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
active=True)
flash("Thanks for registering! You're now logged in.", 'success')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
def register():
"""
Register new user.
"""
form = RegisterForm(request.form)
if form.validate_on_submit():
u = User(
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
)
u.enforce_admin()
u.save()
if u.active:
flash("Thanks for registering! You're now logged in.", "success")
login_user(u)
else:
flash(
"Thanks for registering! You will now need to wait for your admin to approve this account.",
"success",
)
return redirect(url_for("public.home"))
else:
flash_errors(form)
return render_template("public/register.html", form=form)
def admin_add_users(user, *args, **kwargs):
data = request.get_json()
try:
data["user_id"] = int(data["user_id"])
except:
abort(400)
new_user = User(**data)
new_user.enforce_admin()
password = new_user.reset_password()
new_user.active = True
new_user.save()
return jsonify({
"success": True,
"password": password,
"api_token": user.api_token
})
def register():
"""Register new user."""
form = RegisterForm(request.form)
if form.validate_on_submit():
user_cnt = db.session.query(User).count()
u = User.create(
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
active=True,
is_admin=True if user_cnt == 0 else False,
)
flash("Thanks for registering! You're now logged in.", 'success')
login_user(u)
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
def admin_add_users(user, *args, **kwargs):
data = request.get_json()
try:
data['user_id'] = int(data['user_id'])
except:
abort(400)
new_user = User(**data)
password = new_user.reset_password()
new_user.active = True
new_user.save()
return jsonify({
'success': True,
'password': password,
'api_token': user.api_token
})
