Python process_task示例

示例#1

文件： test_worker.py 项目： Yight/InfoSecurity

 def test_process_task_raise_regular(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = KeyError("some exception")
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10], kwargs={})
     task = Request.from_message(m, m.decode())
     worker.process_task(task)
     worker.pool.stop()

示例#2

文件： test_worker.py 项目： Yight/InfoSecurity

 def test_process_task(self):
     worker = self.worker
     worker.pool = Mock()
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10], kwargs={})
     task = Request.from_message(m, m.decode())
     worker.process_task(task)
     self.assertEqual(worker.pool.apply_async.call_count, 1)
     worker.pool.stop()

示例#3

文件： test_worker.py 项目： jsheffie/celery

 def test_process_task_raise_regular(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = KeyError('some exception')
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10],
                        kwargs={})
     task = Request.from_message(m, m.decode())
     worker.process_task(task)
     worker.pool.stop()

示例#4

文件： test_worker.py 项目： jsheffie/celery

 def test_process_task(self):
     worker = self.worker
     worker.pool = Mock()
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10],
                        kwargs={})
     task = Request.from_message(m, m.decode())
     worker.process_task(task)
     self.assertEqual(worker.pool.apply_async.call_count, 1)
     worker.pool.stop()

示例#5

文件： test_worker.py 项目： Yight/InfoSecurity

 def test_process_task_raise_SystemTerminate(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = SystemTerminate()
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10], kwargs={})
     task = Request.from_message(m, m.decode())
     worker.components = []
     worker._state = worker.RUN
     with self.assertRaises(SystemExit):
         worker.process_task(task)
     self.assertEqual(worker._state, worker.TERMINATE)

示例#6

文件： test_worker.py 项目： quoter/celery

 def test_process_task_raise_base(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = KeyboardInterrupt("Ctrl+C")
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10], kwargs={})
     task = Request.from_message(m, m.decode())
     worker.components = []
     worker.namespace.state = RUN
     with self.assertRaises(KeyboardInterrupt):
         worker.process_task(task)
     self.assertEqual(worker.namespace.state, TERMINATE)

示例#7

文件： test_worker.py 项目： jsheffie/celery

 def test_process_task_raise_SystemTerminate(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = SystemTerminate()
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10],
                        kwargs={})
     task = Request.from_message(m, m.decode())
     worker.steps = []
     worker.namespace.state = RUN
     with self.assertRaises(SystemExit):
         worker.process_task(task)
     self.assertEqual(worker.namespace.state, TERMINATE)

示例#8

 def test_process_task_raise_base(self):
     worker = self.worker
     worker.pool = Mock()
     worker.pool.apply_async.side_effect = KeyboardInterrupt('Ctrl+C')
     backend = Mock()
     m = create_message(backend, task=foo_task.name, args=[4, 8, 10],
                        kwargs={})
     task = Request.from_message(m, m.decode())
     worker.components = []
     worker._state = worker.RUN
     with self.assertRaises(KeyboardInterrupt):
         worker.process_task(task)
     self.assertEqual(worker._state, worker.TERMINATE)