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