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_run(self):
master = WorkerMaster()
master._restart_workers = MagicMock(side_effect=[None, ValueError])
master._check_workers = MagicMock()
master._stop_workers = MagicMock()
self.assertRaises(ValueError, master.run)
master._restart_workers.assert_any_call()
master._restart_workers.assert_called_with()
master._check_workers.assert_called_with()
master._stop_workers.assert_called_with()
def test_run(self):
master = WorkerMaster()
master._restart_workers = MagicMock(side_effect=[None, ValueError])
master._check_workers = MagicMock()
master._stop_workers = MagicMock()
self.assertRaises(ValueError, master.run)
master._restart_workers.assert_any_call()
master._restart_workers.assert_called_with()
master._check_workers.assert_called_with()
master._stop_workers.assert_called_with()
