def test_post_model_get_archived(self): user = User('a', '*****@*****.**', '12312') db.session.add(user) db.session.commit() date = datetime.today() - timedelta(10) post = Post('Title', 'Body', user_id=user.id, publish_time=date, difficulty=3) db.session.add(post) db.session.commit() self.assertEqual(Post.get_archived_posts(), []) submission = Submission('a/b/c', 'asdf', post_id=post.id, user_id=user.id) submission.won = True db.session.add(submission) db.session.commit() self.assertEqual(Post.get_archived_posts(), [post])
def test_post_model_is_archived(self): user = User('dan', '*****@*****.**', '12345') db.session.add(user) db.session.commit() post = Post('Title', 'Body', user.id, difficulty=5) db.session.add(post) db.session.commit() self.assertFalse(post.is_archived()) submission = Submission('a/b/c', 'abcdef', user_id=user.id, post_id=post.id) submission.won = True db.session.add(submission) db.session.commit() self.assertTrue(post.is_archived())