def test_poet_model(self):
"""Does poet model work?"""
poet = Poet.signup(username='******',password='******',email="*****@*****.**",image_url=None)
db.session.add(poet)
db.session.commit()
self.assertEqual(poet.username, 'test')
self.assertEqual(Poet.authenticate(username='******',password='******'),poet)
def test_signup(self):
"""Does the signup route work"""
with self.client as c:
resp = c.post("/signup",
data={
"username": "******",
"password": "******",
"email": '*****@*****.**'
})
self.assertEqual(resp.status_code, 302)
self.assertTrue(
Poet.authenticate(username='******', password='******') !=
False)
def login():
if g.poet:
flash("you are already logged in!")
return redirect('/')
form = LoginForm()
if form.validate_on_submit():
username = form.username.data
password = form.password.data
poet = Poet.authenticate(username, password)
if poet:
do_login(poet)
return redirect('/')
else:
form.username.errors.append('Invalid username or password')
return render_template('/user/login.html', form=form)