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)