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)