def test_process_task(self):
worker = self.worker
worker.pool = Mock()
backend = Mock()
m = create_message(backend, task=self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request.from_message(m, m.decode(), app=self.app)
worker._process_task(task)
self.assertEqual(worker.pool.apply_async.call_count, 1)
worker.pool.stop()
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=self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request.from_message(m, m.decode(), app=self.app)
worker._process_task(task)
worker.pool.stop()
def test_process_task_raise_regular(self):
worker = self.worker
worker.pool = Mock()
worker.pool.apply_async.side_effect = KeyError("some exception")
channel = Mock()
m = create_task_message(channel, self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request(m, app=self.app)
worker._process_task(task)
worker.pool.stop()
def test_process_task(self):
worker = self.worker
worker.pool = Mock()
channel = Mock()
m = create_task_message(channel, self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request(m, app=self.app)
worker._process_task(task)
self.assertEqual(worker.pool.apply_async.call_count, 1)
worker.pool.stop()
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=self.foo_task.name, args=[4, 8, 10],
kwargs={})
task = Request(m.decode(), message=m, app=self.app)
worker._process_task(task)
worker.pool.stop()
def test_process_task(self):
worker = self.worker
worker.pool = Mock()
backend = Mock()
m = create_message(backend, task=self.foo_task.name, args=[4, 8, 10],
kwargs={})
task = Request(m.decode(), message=m, app=self.app)
worker._process_task(task)
self.assertEqual(worker.pool.apply_async.call_count, 1)
worker.pool.stop()
def test_process_task_raise_WorkerTerminate(self):
worker = self.worker
worker.pool = Mock()
worker.pool.apply_async.side_effect = WorkerTerminate()
channel = Mock()
m = create_task_message(channel, self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request(m, app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(SystemExit):
worker._process_task(task)
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=self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request.from_message(m, m.decode(), app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(KeyboardInterrupt):
worker._process_task(task)
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=self.foo_task.name, args=[4, 8, 10], kwargs={})
task = Request.from_message(m, m.decode(), app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(SystemExit):
worker._process_task(task)
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=self.foo_task.name, args=[4, 8, 10],
kwargs={})
task = Request(m.decode(), message=m, app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(SystemExit):
worker._process_task(task)
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=self.foo_task.name, args=[4, 8, 10],
kwargs={})
task = Request(m.decode(), message=m, app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(KeyboardInterrupt):
worker._process_task(task)
def test_process_task_raise_regular(self):
worker = self.worker
worker.pool = Mock()
worker.pool.apply_async.side_effect = KeyError('some exception')
channel = Mock()
m = create_task_message(
channel, self.foo_task.name,
args=[4, 8, 10], kwargs={},
)
task = Request(m, app=self.app)
worker._process_task(task)
worker.pool.stop()
def test_process_task(self):
worker = self.worker
worker.pool = Mock()
channel = Mock()
m = create_task_message(
channel, self.foo_task.name,
args=[4, 8, 10], kwargs={},
)
task = Request(m, app=self.app)
worker._process_task(task)
self.assertEqual(worker.pool.apply_async.call_count, 1)
worker.pool.stop()
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.steps = []
worker.namespace.state = RUN
with self.assertRaises(KeyboardInterrupt):
worker._process_task(task)
self.assertEqual(worker.namespace.state, TERMINATE)
def test_process_task_raise_WorkerTerminate(self):
worker = self.worker
worker.pool = Mock()
worker.pool.apply_async.side_effect = WorkerTerminate()
channel = Mock()
m = create_task_message(
channel, self.foo_task.name,
args=[4, 8, 10], kwargs={},
)
task = Request(m, app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(SystemExit):
worker._process_task(task)
def test_process_task_raise_base(self):
worker = self.worker
worker.pool = Mock()
worker.pool.apply_async.side_effect = KeyboardInterrupt('Ctrl+C')
channel = Mock()
m = create_task_message(
channel, self.foo_task.name,
args=[4, 8, 10], kwargs={},
)
task = Request(m, app=self.app)
worker.steps = []
worker.blueprint.state = RUN
with self.assertRaises(KeyboardInterrupt):
worker._process_task(task)
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)
