def test_create(self): # assert default post added self.assertEqual(1, len(Post.query.all())) # assert correct post values q = Post.query.get(self.post.id) self.assertEqual(self.title, q.title) self.assertEqual(self.body, q.body) slug = slugify(datetime.utcnow(), self.title) self.assertEqual(slug, q.slug) self.assertEqual(self.author_id, q.author_id) # assert unique for same date db.session.add(Post(self.title, self.body, self.author_id)) self.assertRaises(IntegrityError, db_commit) # assert not unique for diffrent date now = datetime.utcnow() latest_post = Post(self.title, self.body, self.author_id) latest_post.published = datetime(now.year - 1, now.month, now.day) latest_post.edit(self.title, self.body) db.session.add(latest_post) db.session.commit()
def filldb(): print('Filling database...'), from app.models import User, Post admin = User(u'admin', u'admin') db.session.add(admin) db.session.commit() post = Post( title=u'Hello, world!', body=open('fill/p1.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() post.published = datetime(2011, 06, 13) post.edit(post.title, post.body) db.session.commit() post = Post( title=u'Random Words 1', body=open('fill/p5.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() post.published = datetime(2012, 8, 15) post.edit(post.title, post.body) db.session.commit() post = Post( title=u'Random Words 2', body=open('fill/p2.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() post.published = datetime(2012, 12, 24) post.edit(post.title, post.body) db.session.commit() post = Post( title=u'Picture of a cat!', body=open('fill/p3.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() post = Post( title=u'Some code', body=open('fill/p4.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() post = Post( title=u'Random Words 3', body=open('fill/p5.txt', 'r').read(), author_id=admin.id ) db.session.add(post) db.session.commit() print('done!')