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())
