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)