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_mediator_on_iteration(self): ready_queue = Queue() got = {} def mycallback(value): got["value"] = value.value m = Mediator(ready_queue, mycallback) ready_queue.put(MockTask("George Costanza")) m.on_iteration() self.assertEqual(got["value"], "George Costanza")
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() revoked_tasks.add(t.task_id) ready_queue.put(t) m.on_iteration() self.assertNotIn("value", got) self.assertTrue(t.acked)
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.assertTrue("value" not in got) self.assertTrue(t.acked)