def test_process_hold_queue(self): bucket_queue = Queue() hold_queue = Queue() m = PeriodicWorkController(bucket_queue, hold_queue) m.process_hold_queue() scratchpad = {} def on_accept(): scratchpad["accepted"] = True hold_queue.put((MockTask("task1"), datetime.now() - timedelta(days=1), on_accept)) m.process_hold_queue() self.assertRaises(Empty, hold_queue.get_nowait) self.assertTrue(scratchpad.get("accepted")) self.assertEquals(bucket_queue.get_nowait().value, "task1") tomorrow = datetime.now() + timedelta(days=1) hold_queue.put((MockTask("task2"), tomorrow, on_accept)) m.process_hold_queue() self.assertRaises(Empty, bucket_queue.get_nowait) value, eta, on_accept = hold_queue.get_nowait() self.assertEquals(value.value, "task2") self.assertEquals(eta, tomorrow)
def test_process_hold_queue(self): bucket_queue = Queue() hold_queue = Queue() m = PeriodicWorkController(bucket_queue, hold_queue) m.process_hold_queue() hold_queue.put(("task1", datetime.now() - timedelta(days=1))) m.process_hold_queue() self.assertRaises(Empty, hold_queue.get_nowait) self.assertEquals(bucket_queue.get_nowait(), "task1") tomorrow = datetime.now() + timedelta(days=1) hold_queue.put(("task2", tomorrow)) m.process_hold_queue() self.assertRaises(Empty, bucket_queue.get_nowait) self.assertEquals(hold_queue.get_nowait(), ("task2", tomorrow))
def test_on_iteration(self): bucket_queue = Queue() hold_queue = Queue() m = PeriodicWorkController(bucket_queue, hold_queue) m.on_iteration()
def test_run_periodic_tasks(self): bucket_queue = Queue() hold_queue = Queue() m = PeriodicWorkController(bucket_queue, hold_queue) m.run_periodic_tasks()