def test_mediator_body_exception(self):
ready_queue = Queue()
def mycallback(value):
raise KeyError('foo')
m = Mediator(ready_queue, mycallback)
ready_queue.put(MockTask('Elaine M. Benes'))
m.body()
def test_mediator_body(self):
ready_queue = Queue()
got = {}
def mycallback(value):
got["value"] = value.value
m = Mediator(ready_queue, mycallback)
ready_queue.put(MockTask("George Costanza"))
m.body()
self.assertEqual(got["value"], "George Costanza")
def test_mediator_body(self):
ready_queue = Queue()
got = {}
def mycallback(value):
got["value"] = value.value
m = Mediator(ready_queue, mycallback)
ready_queue.put(MockTask("George Costanza"))
m.body()
self.assertEqual(got["value"], "George Costanza")
def test_mediator_body_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 = uuid()
revoked_tasks.add(t.task_id)
ready_queue.put(t)
m.body()
self.assertNotIn("value", got)
self.assertTrue(t.on_ack.call_count)
def test_mediator_body_revoked(self):
ready_queue = Queue()
got = {}
def mycallback(value):
got['value'] = value.value
m = Mediator(ready_queue, mycallback)
t = MockTask('Jerry Seinfeld')
t.id = uuid()
revoked_tasks.add(t.id)
ready_queue.put(t)
m.body()
self.assertNotIn('value', got)
self.assertTrue(t.on_ack.call_count)
def test_mediator_body(self):
ready_queue = Queue()
got = {}
def mycallback(value):
got['value'] = value.value
m = Mediator(ready_queue, mycallback)
ready_queue.put(MockTask('George Costanza'))
m.body()
self.assertEqual(got['value'], 'George Costanza')
ready_queue.put(MockTask('Jerry Seinfeld'))
m._does_debug = False
m.body()
self.assertEqual(got['value'], 'Jerry Seinfeld')
