Python WorkerMaster._check_workers Examples

Example #1

File: test_worker.py Project: icgood/provoke

 def test_check_workers(self, waitpid_mock):
     worker = MagicMock(queues="testqueues", pid="testpid")
     master = WorkerMaster()
     master.workers = [worker]
     waitpid_mock.return_value = ("testpid", 0)
     self.assertTrue(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     self.assertEqual(None, worker.pid)

Example #2

File: test_worker.py Project: icgood/provoke

 def test_check_workers_no_children(self, waitpid_mock):
     worker = MagicMock(queues="testqueues", pid="testpid")
     master = WorkerMaster()
     master.workers = [worker]
     waitpid_mock.side_effect = OSError(errno.ECHILD, "No child processes")
     self.assertFalse(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     self.assertEqual(None, worker.pid)

Example #3

File: test_worker.py Project: jordandeang/provoke

 def test_check_workers_no_children(self, waitpid_mock):
     worker = MagicMock(queues='testqueues', pid='testpid')
     master = WorkerMaster()
     master.workers = [worker]
     waitpid_mock.side_effect = OSError(errno.ECHILD, 'No child processes')
     self.assertFalse(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     self.assertEqual(None, worker.pid)

Example #4

File: test_worker.py Project: jordandeang/provoke

 def test_check_workers(self, waitpid_mock):
     worker = MagicMock(queues='testqueues', pid='testpid')
     master = WorkerMaster()
     master.workers = [worker]
     waitpid_mock.return_value = ('testpid', 0)
     self.assertTrue(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     self.assertEqual(None, worker.pid)

Example #5

File: test_worker.py Project: icgood/provoke

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

Example #6

File: test_worker.py Project: jon-armstrong/provoke

 def test_check_workers_no_children(self, waitpid_mock):
     exit_cb = MagicMock()
     worker = MagicMock(queues='testqueues', pid='testpid')
     master = WorkerMaster('testapp', exit_callback=exit_cb)
     master.workers = [worker]
     waitpid_mock.side_effect = OSError(errno.ECHILD, 'No child processes')
     self.assertFalse(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     exit_cb.assert_called_with('testqueues', 'testpid', None)
     self.assertEqual(None, worker.pid)

Example #7

File: test_worker.py Project: jon-armstrong/provoke

 def test_check_workers(self, waitpid_mock):
     exit_cb = MagicMock()
     worker = MagicMock(queues='testqueues', pid='testpid')
     master = WorkerMaster('testapp', exit_callback=exit_cb)
     master.workers = [worker]
     waitpid_mock.return_value = ('testpid', 0)
     self.assertTrue(master._check_workers())
     waitpid_mock.assert_called_with(0, 0)
     exit_cb.assert_called_with('testqueues', 'testpid', 0)
     self.assertEqual(None, worker.pid)

Example #8

File: test_worker.py Project: jordandeang/provoke

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