def test_run(self, thread_cls_mock):
thread_cls_mock.return_value = thread_mock = MagicMock()
worker = WorkerProcess("testapp", ["testqueue"], 10)
worker._consume = MagicMock(side_effect=SystemExit)
worker._run()
thread_cls_mock.assert_called_with(target=ANY)
thread_mock.start.assert_called_with()
worker._consume.assert_called_with()
def test_run(self, thread_cls_mock):
thread_cls_mock.return_value = thread_mock = MagicMock()
worker = WorkerProcess('testapp', ['testqueue'], 10)
worker._consume = MagicMock(side_effect=SystemExit)
worker._run()
thread_cls_mock.assert_called_with(target=ANY)
thread_mock.start.assert_called_with()
worker._consume.assert_called_with()
def test_run_accessrefused(self, thread_cls_mock, sleep_mock):
thread_cls_mock.return_value = thread_mock = MagicMock()
worker = WorkerProcess("testapp", ["testqueue"], 10)
worker._consume = MagicMock(side_effect=[AccessRefused, None])
worker._run()
thread_cls_mock.assert_called_with(target=ANY)
thread_mock.start.assert_called_with()
sleep_mock.assert_called_once_with(ANY)
worker._consume.assert_called_with()
def test_run_accessrefused(self, thread_cls_mock, sleep_mock):
thread_cls_mock.return_value = thread_mock = MagicMock()
worker = WorkerProcess('testapp', ['testqueue'], 10)
worker._consume = MagicMock(side_effect=[AccessRefused, None])
worker._run()
thread_cls_mock.assert_called_with(target=ANY)
thread_mock.start.assert_called_with()
sleep_mock.assert_called_once_with(ANY)
worker._consume.assert_called_with()
def test_start_worker_child(self, exit_mock, fork_mock):
app = MagicMock()
worker = WorkerProcess(app, ["testqueue"])
worker._run = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 0
master._start_worker(worker)
fork_mock.assert_called_with()
worker._run.assert_called_with()
exit_mock.assert_called_with(os.EX_OK)
def test_start_worker_child(self, exit_mock, fork_mock):
app = MagicMock()
worker = WorkerProcess(app, ['testqueue'])
worker._run = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 0
master._start_worker(worker)
fork_mock.assert_called_with()
worker._run.assert_called_with()
exit_mock.assert_called_with(os.EX_OK)
def test_get_worker_app(self, exit_mock, fork_mock):
app = MagicMock()
worker = WorkerProcess(app, ["testqueue"])
worker._run = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 0
self.assertRaises(RuntimeError, get_worker_app)
master._start_worker(worker)
self.assertEquals(app, get_worker_app())
fork_mock.assert_called_with()
worker._run.assert_called_with()
exit_mock.assert_called_with(os.EX_OK)
def test_get_worker_app(self, exit_mock, fork_mock):
app = MagicMock()
worker = WorkerProcess(app, ['testqueue'])
worker._run = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 0
self.assertRaises(RuntimeError, get_worker_app)
master._start_worker(worker)
self.assertEquals(app, get_worker_app())
fork_mock.assert_called_with()
worker._run.assert_called_with()
exit_mock.assert_called_with(os.EX_OK)
