def test_sched_clear(self): ready_queue = Queue() sched = Scheduler(ready_queue) sched.enter(MockItem("foo"), eta=datetime.now(), callback=None) self.assertFalse(sched.empty()) sched.clear() self.assertTrue(sched.empty())
def test_sched_revoked(self): ready_queue = Queue() sched = Scheduler(ready_queue) now = datetime.now() callback_called = [False] def callback(): callback_called[0] = True sched.enter(MockItem("foo", revoked=True), eta=now, callback=callback) iter(sched).next() self.assertFalse(callback_called[0]) self.assertRaises(Empty, ready_queue.get_nowait) self.assertFalse(sched.queue) sched.clear()