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