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()