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)