def test_receieve_message_not_registered(self):
l = CarrotListener(self.ready_queue, self.eta_scheduler, self.logger,
send_events=False)
backend = MockBackend()
m = create_message(backend, task="x.X.31x", args=[2, 4, 8], kwargs={})
l.event_dispatcher = MockEventDispatcher()
self.assertFalse(l.receive_message(m.decode(), m))
self.assertRaises(Empty, self.ready_queue.get_nowait)
self.assertTrue(self.eta_scheduler.empty())
def test_receieve_message(self):
l = CarrotListener(self.ready_queue, self.eta_scheduler, self.logger,
send_events=False)
backend = MockBackend()
m = create_message(backend, task=foo_task.name,
args=[2, 4, 8], kwargs={})
l.event_dispatcher = MockEventDispatcher()
l.receive_message(m.decode(), m)
in_bucket = self.ready_queue.get_nowait()
self.assertTrue(isinstance(in_bucket, TaskWrapper))
self.assertEquals(in_bucket.task_name, foo_task.name)
self.assertEquals(in_bucket.execute(), 2 * 4 * 8)
self.assertTrue(self.eta_scheduler.empty())
def test_revoke(self):
ready_queue = Queue()
l = CarrotListener(ready_queue, self.eta_scheduler, self.logger,
send_events=False)
backend = MockBackend()
id = gen_unique_id()
c = create_message(backend, control={"command": "revoke", "task_id": id})
t = create_message(backend, task=foo_task.name, args=[2, 4, 8],
kwargs={}, id=id)
l.event_dispatcher = MockEventDispatcher()
l.receive_message(c.decode(), c)
from celery.worker.revoke import revoked
self.assertTrue(id in revoked)
l.receive_message(t.decode(), t)
self.assertTrue(ready_queue.empty())
