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_Quote_model(self):
"""Does the quote model work?"""
poet = Poet.signup(username='******',password='******',email="*****@*****.**",image_url=None)
db.session.add(poet)
db.session.commit()
quote = Quote(content='this is a test message',poet_id=poet.id)
db.session.add(quote)
db.session.commit()
self.assertEqual(quote.poet,poet)
self.assertEqual(quote.content,'this is a test message')
def setUp(self):
"""Create test client, add sample data."""
Poet.query.delete()
Quote.query.delete()
self.client = app.test_client()
self.poet = Poet.signup(username="******",
email="*****@*****.**",
password="******",
image_url=None)
db.session.commit()
def test_comment_model(self):
"""Does the comment functionality work?"""
poet = Poet.signup(username='******',password='******',email="*****@*****.**",image_url=None)
db.session.add(poet)
db.session.commit()
quote = Quote(content='this is a test message',poet_id=poet.id)
db.session.add(quote)
db.session.commit()
comment = Comment(content='test comment', poet_id=poet.id,quote_id=quote.id)
db.session.add(comment)
db.session.commit()
self.assertEqual(comment.poet,poet)
self.assertEqual(comment.content,'test comment')
self.assertIn(comment, quote.comments)
def signup():
form = SignupForm()
if form.validate_on_submit():
try:
poet = Poet.signup(username=form.username.data,
email=form.email.data,
password=form.password.data,
image_url=form.image_url.data
or Poet.image_url.default.arg)
db.session.commit()
do_login(poet)
return redirect('/')
except IntegrityError:
beans = IntegrityError
raise
return redirect('/beans')
return render_template('/user/signup.html', form=form)