def revoke(panel, task_id, task_name=None, **kwargs):
"""Revoke task by task id."""
revoked.add(task_id)
backend = default_backend
if task_name: # Use custom task backend (if any)
try:
backend = tasks[task_name].backend
except KeyError:
pass
backend.mark_as_revoked(task_id)
panel.logger.warn("Task %s revoked" % (task_id,))
return {"ok": "task %s revoked" % (task_id,)}
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)
def revoke(self, task_id, **kwargs):
"""Revoke task by task id."""
revoked.add(task_id)
self.logger.warn("Task %s revoked." % task_id)
def test_execute_using_pool_does_not_execute_revoked(self):
tw = TaskRequest(mytask.name, gen_unique_id(), [1], {"f": "x"})
revoked.add(tw.task_id)
tw.execute_using_pool(None)
def test_revoked(self):
tw = TaskRequest(mytask.name, gen_unique_id(), [1], {"f": "x"})
revoked.add(tw.task_id)
self.assertTrue(tw.revoked())
self.assertTrue(tw._already_revoked)
self.assertTrue(tw.acknowledged)
