def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(gen_unique_id(), 42) tb.save_taskset(gen_unique_id(), {"foo": "bar"}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.result_expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.result_expires * 2 s.commit() s.close() tb.cleanup() s2 = tb.ResultSession() self.assertEqual(s2.query(Task).count(), 0) self.assertEqual(s2.query(TaskSet).count(), 0)
def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(uuid(), 42) tb.save_group(uuid(), {'foo': 'bar'}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.expires * 2 s.commit() s.close() tb.cleanup()