コード例 #1
0
ファイル: test_request.py プロジェクト: EzyInsights/celery
 def test_execute_using_pool_does_not_execute_revoked(self):
     job = TaskRequest(
         self.mytask.name, uuid(), [1], {'f': 'x'}, app=self.app,
     )
     revoked.add(job.id)
     with self.assertRaises(TaskRevokedError):
         job.execute_using_pool(None)
コード例 #2
0
ファイル: test_request.py プロジェクト: uceo/uceo-2015
 def test_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     with assert_signal_called(task_revoked, sender=tw.task, terminated=False, expired=False, signum=None):
         revoked.add(tw.id)
         self.assertTrue(tw.revoked())
         self.assertTrue(tw._already_revoked)
         self.assertTrue(tw.acknowledged)
コード例 #3
0
ファイル: test_request.py プロジェクト: alekibango/celery
 def test_revoked(self):
     job = self.xRequest()
     with self.assert_signal_called(
             task_revoked, sender=job.task, request=job,
             terminated=False, expired=False, signum=None):
         revoked.add(job.id)
         assert job.revoked()
         assert job._already_revoked
         assert job.acknowledged
コード例 #4
0
ファイル: test_request.py プロジェクト: nonoDevil/celery
 def test_revoked(self):
     job = TaskRequest(self.mytask.name, uuid(), [1], {"f": "x"}, app=self.app)
     with assert_signal_called(
         task_revoked, sender=job.task, request=job, terminated=False, expired=False, signum=None
     ):
         revoked.add(job.id)
         self.assertTrue(job.revoked())
         self.assertTrue(job._already_revoked)
         self.assertTrue(job.acknowledged)
コード例 #5
0
ファイル: test_worker.py プロジェクト: OddBloke/celery
    def test_revoke(self):
        ready_queue = FastQueue()
        l = MyKombuConsumer(ready_queue, self.eta_schedule, self.logger, send_events=False)
        backend = Mock()
        id = gen_unique_id()
        t = create_message(backend, task=foo_task.name, args=[2, 4, 8], kwargs={}, id=id)
        from celery.worker.state import revoked

        revoked.add(id)

        l.receive_message(t.decode(), t)
        self.assertTrue(ready_queue.empty())
コード例 #6
0
ファイル: builtins.py プロジェクト: mootpointer/celery
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, )}
コード例 #7
0
ファイル: test_worker.py プロジェクト: AdrianRibao/celery
    def test_revoke(self):
        ready_queue = FastQueue()
        l = MyKombuConsumer(ready_queue, timer=self.timer)
        backend = Mock()
        id = uuid()
        t = create_message(backend, task=foo_task.name, args=[2, 4, 8],
                           kwargs={}, id=id)
        from celery.worker.state import revoked
        revoked.add(id)

        l.receive_message(t.decode(), t)
        self.assertTrue(ready_queue.empty())
コード例 #8
0
ファイル: test_worker.py プロジェクト: dn0/celery
    def test_revoke(self):
        l = _MyKombuConsumer(self.buffer.put, timer=self.timer, app=self.app)
        l.steps.pop()
        backend = Mock()
        id = uuid()
        t = create_message(backend, task=self.foo_task.name, args=[2, 4, 8],
                           kwargs={}, id=id)
        from celery.worker.state import revoked
        revoked.add(id)

        callback = self._get_on_message(l)
        callback(t.decode(), t)
        self.assertTrue(self.buffer.empty())
コード例 #9
0
ファイル: builtins.py プロジェクト: aleszoulek/celery
def revoke(panel, task_id, terminate=False, signal=None, **kwargs):
    """Revoke task by task id."""
    revoked.add(task_id)
    action = "revoked"
    if terminate:
        signum = get_signal(signal)
        for request in state.active_requests:
            if request.task_id == task_id:
                action = "terminated (%s)" % (signum, )
                request.terminate(panel.consumer.pool, signal=signum)
                break

    panel.logger.warn("Task %s %s." % (task_id, action))
    return {"ok": "task %s %s" % (task_id, action)}
コード例 #10
0
ファイル: test_worker.py プロジェクト: nmadhok/celery
    def test_revoke(self):
        l = _MyKombuConsumer(self.buffer.put, timer=self.timer, app=self.app)
        l.blueprint.state = RUN
        l.steps.pop()
        channel = Mock()
        id = uuid()
        t = create_task_message(channel, self.foo_task.name, args=[2, 4, 8], kwargs={}, id=id)
        from celery.worker.state import revoked

        revoked.add(id)

        callback = self._get_on_message(l)
        callback(t)
        self.assertTrue(self.buffer.empty())
コード例 #11
0
    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)
コード例 #12
0
    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)
コード例 #13
0
ファイル: test_request.py プロジェクト: arunsuresh/celery
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     with self.assertRaises(TaskRevokedError):
         tw.execute_using_pool(None)
コード例 #14
0
 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)
コード例 #15
0
ファイル: test_request.py プロジェクト: alekibango/celery
 def test_execute_does_not_execute_revoked(self):
     job = self.xRequest()
     revoked.add(job.id)
     job.execute()
コード例 #16
0
ファイル: test_request.py プロジェクト: alekibango/celery
 def test_execute_using_pool_does_not_execute_revoked(self):
     job = self.xRequest()
     revoked.add(job.id)
     with pytest.raises(TaskRevokedError):
         job.execute_using_pool(None)
コード例 #17
0
 def test_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     self.assertTrue(tw.revoked())
     self.assertTrue(tw._already_revoked)
     self.assertTrue(tw.acknowledged)
コード例 #18
0
 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)
コード例 #19
0
ファイル: test_request.py プロジェクト: EzyInsights/celery
 def test_execute_does_not_execute_revoked(self):
     job = TaskRequest(
         self.mytask.name, uuid(), [1], {'f': 'x'}, app=self.app,
     )
     revoked.add(job.id)
     job.execute()
コード例 #20
0
ファイル: builtins.py プロジェクト: jokar/minion
def revoke(panel, task_id, **kwargs):
    """Revoke task by task id."""
    revoked.add(task_id)
    panel.logger.warn("Task %s revoked" % (task_id, ))
    return {"ok": "task %s revoked" % (task_id, )}
コード例 #21
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     job = self.xRequest()
     revoked.add(job.id)
     with self.assertRaises(TaskRevokedError):
         job.execute_using_pool(None)
コード例 #22
0
ファイル: test_worker_job.py プロジェクト: sunliwen/celery
 def test_execute_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     revoked.add(tw.task_id)
     tw.execute()
コード例 #23
0
 def test_execute_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     tw.execute()
コード例 #24
0
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     with self.assertRaises(TaskRevokedError):
         tw.execute_using_pool(None)
コード例 #25
0
ファイル: test_worker_job.py プロジェクト: ivirabyan/celery
 def test_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     revoked.add(tw.id)
     self.assertTrue(tw.revoked())
     self.assertTrue(tw._already_revoked)
     self.assertTrue(tw.acknowledged)
コード例 #26
0
ファイル: test_request.py プロジェクト: arunsuresh/celery
 def test_execute_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {'f': 'x'})
     revoked.add(tw.id)
     tw.execute()
コード例 #27
0
ファイル: test_request.py プロジェクト: zubairalam/celery
 def test_execute_does_not_execute_revoked(self):
     job = self.xRequest()
     revoked.add(job.id)
     job.execute()
コード例 #28
0
ファイル: test_worker_job.py プロジェクト: ivirabyan/celery
 def test_execute_using_pool_does_not_execute_revoked(self):
     tw = TaskRequest(mytask.name, uuid(), [1], {"f": "x"})
     revoked.add(tw.id)
     tw.execute_using_pool(None)
コード例 #29
0
ファイル: builtins.py プロジェクト: dpwhite2/celery
def revoke(panel, task_id, **kwargs):
    """Revoke task by task id."""
    revoked.add(task_id)
    panel.logger.warn("Task %s revoked" % (task_id, ))
    return {"ok": "task %s revoked" % (task_id, )}