def setUp(self):
Test.setUp(self)
self.post = Post(
title="Title",
description="Description"
)
self.post.add()
def add():
form = get_form(post_heads)
if request.method == "POST":
post = Post(title=form["title"], description=form["description"])
error = post.request.add()
if not error:
return redirect(url_for('home.main_page'))
flash(error)
return render_template('post/add.html', post_heads=post_heads, form=form)
def update(id):
post = Post.get(id)
if request.method == "POST":
update_obj_attrs(post, post_heads)
error = post.request.update()
if not error:
return redirect(url_for('home.main_page'))
flash(error)
return render_template('post/update.html',
post_heads=post_heads,
post=post)
def test_should_not_add_post_given_invalid_post_data_and_LUHP(self):
self.login_user(self.admin_user)
post_data = dict(
title="",
description="Valid Description"
)
with self.client as client:
client.post(
url_for('post.add'),
data=post_data
)
self.assertFalse(Post.search(""))
def test_should_add_post_given_valid_post_data_and_LUHP(self):
self.login_user(self.admin_user)
post_data = dict(
title="Valid Title",
description="Valid Description"
)
with self.client as client:
client.post(
url_for("post.add"),
data=post_data
)
self.assertTrue(Post.search("Valid Title"))
def main_page():
posts = Post.get_all()
return render_template('home/main-page.html', posts=posts)
def test_should_add_post_given_valid_post(self):
post = Post(title="Valid Title", description="Valid Description")
post.request.add()
self.assertIn(post, self.db.session)
def delete(id):
post = Post.get(id)
post.delete()
return redirect(url_for('home.main_page'))