Example #1
0
    def test_purge(self):
        import datetime

        _, token1 = self._register()
        _, token2 = self._register()
        self.assertEqual(self.db.session.query(myownmocker.MockToken).count(), 2)
        self.assertEqual(myownmocker.purge_tokens(), 0)

        t = self.db.session.query(myownmocker.MockToken).filter(myownmocker.MockToken.token == token1).first()
        t.created_on -= datetime.timedelta(days=2)
        self.assertEqual(myownmocker.purge_tokens(), 1)
        self.assertEqual(self.db.session.query(myownmocker.MockToken).count(), 1)