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_data(self, exit_mock, fork_mock):
app = MagicMock()
worker = _WorkerProcess(app, ['testqueue'], None, None, None, False)
worker._run = MagicMock()
master = WorkerMaster('testapp', worker_data={'test': 'data'})
fork_mock.return_value = 0
self.assertRaises(RuntimeError, get_worker_data, 'test')
master._start_worker(worker)
self.assertEquals('data', get_worker_data('test'))
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_restart_workers(self):
worker1 = MagicMock(queues="testqueues", pid="testpid1")
worker2 = MagicMock(queues="testqueues", pid=None)
master = WorkerMaster()
master._start_worker = MagicMock(return_value="testpid2")
master.workers = [worker1, worker2]
master._restart_workers()
master._start_worker.assert_called_once_with(worker2)
def test_restart_workers(self):
worker1 = MagicMock(queues='testqueues', pid='testpid1')
worker2 = MagicMock(queues='testqueues', pid=None)
master = WorkerMaster()
master._start_worker = MagicMock(return_value='testpid2')
master.workers = [worker1, worker2]
master._restart_workers()
master._start_worker.assert_called_once_with(worker2)
def test_restart_workers(self):
start_cb = MagicMock()
worker1 = MagicMock(queues='testqueues', pid='testpid1')
worker2 = MagicMock(queues='testqueues', pid=None)
master = WorkerMaster('testapp', start_callback=start_cb)
master._start_worker = MagicMock(return_value='testpid2')
master.workers = [worker1, worker2]
master._restart_workers()
master._start_worker.assert_called_once_with(worker2)
start_cb.assert_called_with('testqueues', 'testpid2')
def test_start_worker_parent(self, fork_mock):
worker = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 13
self.assertEqual(13, master._start_worker(worker))
fork_mock.assert_called_with()
def test_start_worker_parent(self, fork_mock):
worker = MagicMock()
master = WorkerMaster()
fork_mock.return_value = 13
self.assertEqual(13, master._start_worker(worker))
fork_mock.assert_called_with()
