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