Exemple #1
0
 def test_ready(self, requests=[SimpleReq('foo'), SimpleReq('bar')]):
     for request in requests:
         state.task_accepted(request)
     assert len(state.active_requests) == 2
     for request in requests:
         state.task_ready(request)
     assert len(state.active_requests) == 0
Exemple #2
0
 def on_accepted(self):
     state.task_accepted(self)
     if not self.task.acks_late:
         self.acknowledge()
     self.send_event("task-started", uuid=self.task_id)
     self.logger.debug("Task accepted: %s[%s]" % (
         self.task_name, self.task_id))
Exemple #3
0
 def test_ready(self, requests=[SimpleReq('foo'), SimpleReq('bar')]):
     for request in requests:
         state.task_accepted(request)
     self.assertEqual(len(state.active_requests), 2)
     for request in requests:
         state.task_ready(request)
     self.assertEqual(len(state.active_requests), 0)
 def on_accepted(self):
     state.task_accepted(self)
     if not self.task.acks_late:
         self.acknowledge()
     self.send_event("task-started", uuid=self.task_id)
     self.logger.debug("Task accepted: %s[%s]" %
                       (self.task_name, self.task_id))
 def test_ready(self, requests=[SimpleReq('foo'),
                                SimpleReq('bar')]):
     for request in requests:
         state.task_accepted(request)
     self.assertEqual(len(state.active_requests), 2)
     for request in requests:
         state.task_ready(request)
     self.assertEqual(len(state.active_requests), 0)
Exemple #6
0
 def test_ready(self, requests=[SimpleReq('foo'),
                                SimpleReq('bar')]):
     for request in requests:
         state.task_accepted(request)
     assert len(state.active_requests) == 2
     for request in requests:
         state.task_ready(request)
     assert len(state.active_requests) == 0
Exemple #7
0
 def on_accepted(self):
     """Handler called when task is accepted by worker pool."""
     self.time_start = time.time()
     state.task_accepted(self)
     if not self.task.acks_late:
         self.acknowledge()
     self.send_event("task-started", uuid=self.task_id)
     self.logger.debug("Task accepted: %s[%s]" % (
         self.task_name, self.task_id))
Exemple #8
0
 def test_accepted(self, requests=[SimpleReq('foo'),
                                   SimpleReq('bar'),
                                   SimpleReq('baz'),
                                   SimpleReq('baz')]):
     for request in requests:
         state.task_accepted(request)
     for req in requests:
         self.assertIn(req, state.active_requests)
     self.assertEqual(state.total_count['foo'], 1)
     self.assertEqual(state.total_count['bar'], 1)
     self.assertEqual(state.total_count['baz'], 2)
Exemple #9
0
 def test_accepted(self, requests=[SimpleReq("foo"),
                                   SimpleReq("bar"),
                                   SimpleReq("baz"),
                                   SimpleReq("baz")]):
     for request in requests:
         state.task_accepted(request)
     for req in requests:
         self.assertIn(req, state.active_requests)
     self.assertEqual(state.total_count["foo"], 1)
     self.assertEqual(state.total_count["bar"], 1)
     self.assertEqual(state.total_count["baz"], 2)
Exemple #10
0
 def test_accepted(self, requests=[SimpleReq("foo"),
                                   SimpleReq("bar"),
                                   SimpleReq("baz"),
                                   SimpleReq("baz")]):
     for request in requests:
         state.task_accepted(request)
     for req in requests:
         self.assertIn(req, state.active_requests)
     self.assertEqual(state.total_count["foo"], 1)
     self.assertEqual(state.total_count["bar"], 1)
     self.assertEqual(state.total_count["baz"], 2)
Exemple #11
0
 def test_accepted(self, requests=[SimpleReq('foo'),
                                   SimpleReq('bar'),
                                   SimpleReq('baz'),
                                   SimpleReq('baz')]):
     for request in requests:
         state.task_accepted(request)
     for req in requests:
         assert req in state.active_requests
     assert state.total_count['foo'] == 1
     assert state.total_count['bar'] == 1
     assert state.total_count['baz'] == 2
Exemple #12
0
 def test_accepted(self, requests=[SimpleReq('foo'),
                                   SimpleReq('bar'),
                                   SimpleReq('baz'),
                                   SimpleReq('baz')]):
     for request in requests:
         state.task_accepted(request)
     for req in requests:
         assert req in state.active_requests
     assert state.total_count['foo'] == 1
     assert state.total_count['bar'] == 1
     assert state.total_count['baz'] == 2
Exemple #13
0
 def on_accepted(self, pid, time_accepted):
     """Handler called when task is accepted by worker pool."""
     self.worker_pid = pid
     self.time_start = time_accepted
     state.task_accepted(self)
     if not self.task.acks_late:
         self.acknowledge()
     self.send_event("task-started", uuid=self.task_id, pid=pid)
     self.logger.debug("Task accepted: %s[%s] pid:%r" % (
         self.task_name, self.task_id, pid))
     if self._terminate_on_ack is not None:
         _, pool, signal = self._terminate_on_ack
         self.terminate(pool, signal)
Exemple #14
0
 def on_accepted(self, pid, time_accepted):
     """Handler called when task is accepted by worker pool."""
     self.worker_pid = pid
     self.time_start = time_accepted
     state.task_accepted(self)
     if not self.task.acks_late:
         self.acknowledge()
     self.send_event("task-started", uuid=self.task_id, pid=pid)
     self.logger.debug("Task accepted: %s[%s] pid:%r" %
                       (self.task_name, self.task_id, pid))
     if self._terminate_on_ack is not None:
         _, pool, signal = self._terminate_on_ack
         self.terminate(pool, signal)