class PostsModelTest(unittest.TestCase):
"""
"""
def setUp(self):
self.user_James = User(username='******',
password='******',
email='*****@*****.**')
self.new_post = Posts(title="You're testing",
blog='This the test of the century',
category="my-thoughts",
user=self.user_James)
def tearDown(self):
Posts.query.delete()
User.query.delete()
def test_check_instance_variables(self):
self.assertEquals(self.new_post.title, "Youre testing")
self.assertEquals(self.new_post.blog,
"This is the test of the century")
self.assertEquals(self.new_post.category, "my-thoughts")
self.assertEquals(self.new_post.user, self.user_James)
def test_save_post(self):
self.new_post.save_post()
self.assertTrue(len(Posts.query.all()) > 0)
class PitchModelTest(unittest.TestCase):
def setUp(self):
self.new_post = Posts(title="title", post="content")
db.session.add(self.new_blog)
db.session.commit()
def tearDown(self):
Posts.query.delete()
db.session.commit()
def test_save_blogs(self):
self.new_post.save_post()
self.assertTrue(len(Blog.query.all()) > 0)
def test_check_instance_variables(self):
self.assertEquals(self.new_blog.title, 'title')
self.assertEquals(self.new_blog.post, 'content')
class PostsModelTest(unittest.TestCase):
def setUp(self):
self.new_post = Posts(title='Terabyte', body='This is a new post',category='Technology')
def test_instance(self):
'''
Test case to check if new_post is an instance of Posts class
'''
self.assertTrue( isinstance( self.new_post, Posts) )
def test_save_post(self):
'''
Test case to check if a post is saved to the database
'''
self.new_post.save_post()
self.assertTrue( len(Posts.query.all()) > 0 )
def create_post():
post_form = PostForm()
user = current_user
if user.user_type == 'Writer':
if post_form.validate_on_submit():
post = Posts(title=post_form.title.data,
category=post_form.category.data,
blog=post_form.post.data,
user=current_user)
post.save_post()
users = User.query.filter_by(user_type='User').all()
for user in users:
mail_message("New Post has arrived",
"email/new_post",
user.email,
user=user)
return redirect(url_for('main.writer'))
else:
return "This page is for only writers"
return render_template('createpost.html', post_form=post_form)
class PostModelTest(unittest.TestCase):
def setUp(self):
self.user_denno = User(user_name="denno", password="******")
self.new_post = Posts(title='a',
post='b',
category='d',
posts=self.user_denno)
def test_instance(self):
self.assertEqual(self.new_post.title, 'a')
self.assertEqual(self.new_post.post, 'b')
self.assertEqual(self.new_post.category, 'd')
def test_save_post(self):
self.new_post.save_post()
self.assertTrue(len(Posts.query.all()) > 0)
def test_get_post_by_id(self):
self.new_post.save_post()
got_post = Posts.get_post(1)
self.assertTrue(len(got_post) > 0)
