def register():
if current_user.is_authenticated:
return redirect(url_for('home'))
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')
user = User(username=form.username.data, email=form.email.data, password=hashed_password)
db.session.add(user)
db.session.commit()
flash('Your account has been created! You are now able to log in', 'success')
return redirect(url_for('login'))
return render_template('register.html', title='Register', form=form)
class UserTest(unittest.TestCase):
def setUp(self):
self.new_user = User(password='******')
def test_init(self):
self.assertTrue(self.new_user.password_encrypt is not None)
def test_alert(self):
with self.assertRaises(AttributeError):
self.new_user.password
def test_save(self):
self.assertTrue(self.new_user.check_password('friday123'))
def register():
if current_user.is_authenticated:
return redirect(url_for("home"))
form = RegistrationForm()
if form.validate_on_submit():
hashed_password = bcrypt.generate_password_hash(
form.password.data).decode("utf-8")
user = User(username=form.username.data,
email=form.email.data,
password=hashed_password)
db.session.add(user)
db.session.commit()
flash("Your account has been created! You are now able to Log In",
"success")
return redirect(url_for("login"))
return render_template("register.html", title="Register", form=form)
def setUp(self):
self.new_user = User(password='******')