def test_mediator_move_exception(self): ready_queue = Queue() def mycallback(value): raise KeyError("foo") m = Mediator(ready_queue, mycallback) ready_queue.put(MockTask("Elaine M. Benes")) m.move()
def test_mediator_move(self): ready_queue = Queue() got = {} def mycallback(value): got["value"] = value.value m = Mediator(ready_queue, mycallback) ready_queue.put(MockTask("George Costanza")) m.move() self.assertEqual(got["value"], "George Costanza")
def test_mediator_on_iteration(self): bucket_queue = Queue() got = {} def mycallback(value): got["value"] = value m = Mediator(bucket_queue, mycallback) bucket_queue.put("George Constanza") m.on_iteration() self.assertEquals(got["value"], "George Constanza")
def test_run(self): ready_queue = Queue() condition = [None] def mycallback(value): condition[0].set() m = Mediator(ready_queue, mycallback) condition[0] = m._shutdown ready_queue.put(MockTask("Elaine M. Benes")) m.run() self.assertTrue(m._shutdown.isSet()) self.assertTrue(m._stopped.isSet())
def test_mediator_move_revoked(self): ready_queue = Queue() got = {} def mycallback(value): got["value"] = value.value m = Mediator(ready_queue, mycallback) t = MockTask("Jerry Seinfeld") t.task_id = gen_unique_id() revoked_tasks.add(t.task_id) ready_queue.put(t) m.move() self.assertNotIn("value", got) self.assertTrue(t.acked)
def test_mediator_start__stop(self): ready_queue = Queue() m = Mediator(ready_queue, lambda t: t) m.start() self.assertFalse(m._shutdown.isSet()) self.assertFalse(m._stopped.isSet()) m.stop() m.join() self.assertTrue(m._shutdown.isSet()) self.assertTrue(m._stopped.isSet())
def test_mediator_on_iteration_revoked(self): ready_queue = Queue() got = {} def mycallback(value): got["value"] = value.value m = Mediator(ready_queue, mycallback) t = MockTask("Jerry Seinfeld") t.task_id = gen_unique_id() from celery.worker.revoke import revoked revoked.add(t.task_id) ready_queue.put(t) m.on_iteration() self.assertNotIn("value", got) self.assertTrue(t.acked)