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 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)
def added_user():
user = User(first_name=request.form['First name'],
last_name=request.form['Last name'],
birth_date=request.form['birthday'])
db.session.add(user)
try:
db.session.commit()
except OperationalError as ex:
logger.warning('Произошла ошибка подключения к базе данных')
return jsonify({'error': "Bad connection. Try again later."})
except DataError as ex:
db.session.rollback()
logger.warning('Были введены неверные данные')
return jsonify({'error': "Data Format Error"})
except IntegrityError as ex:
db.session.rollback()
logger.warning('Была попытка добавить существующего пользователя')
return jsonify({'error': "User already exists"})
return 'USER {} {} added successfully'.format(user.first_name,
user.last_name)
def load_user(user_id):
"""Load user by 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)
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