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