def test_os_exit_didnt_call_when_os_fork_returned_zero_both_times(self): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)): with mock.patch('os.setsid'): with mock.patch('os._exit') as os_exit_mock: utils.daemonize() self.assertEqual(0, os_exit_mock.call_count)
def test_daemonize_child_parent(self): with mock.patch('os.fork', mock.Mock(side_effect=[0, 42])): with mock.patch('os._exit', mock.Mock()) as os_exit: with mock.patch('os.setsid', mock.Mock()): utils.daemonize() os_exit.assert_called_once_with(0)
def test_daemonize_from_child_unsuccess(self, m_exit): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)) as m_fork: with mock.patch('os.setsid', mock.Mock()) as m_setsid: daemonize() m_setsid.assert_called_once_with() self.assertEqual(m_fork.call_count, 2) self.assertFalse(m_exit.called)
def test_os_exit_called_once_when_os_fork_returned_zero_first_time_and_os_fork_return_valid_pid_second_time(self): invalid_pid = 0 valid_pid = 1 with mock.patch('os.fork', mock.Mock(side_effect=[invalid_pid, valid_pid])): with mock.patch('os.setsid'): with mock.patch('os._exit') as os_exit_mock: utils.daemonize() os_exit_mock.assert_called_once_with(0)
def test_daemonize_parent(self): test_pid = 42 with mock.patch('os.fork', mock.Mock(return_value=test_pid)): with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize() os_exit.assert_called_once_with(0)
def test_daemonize_pid_not_0(self): """ pid всегда != 0. os_exit должен вызваться :return: """ with mock.patch('source.lib.utils.try_fork', mock.Mock(return_value=100)): with mock.patch('os._exit', mock.Mock()) as exit: utils.daemonize() assert exit.called
def test_daemonize_pid_not_0_second_time(self): """ pid всегда = 0 при первом вызове и != 0 при втором. os_exit должен вызваться :return: """ with mock.patch('source.lib.utils.try_fork', mock.Mock(side_effect=[0, 1])): with mock.patch('os._exit', mock.Mock()) as exit: utils.daemonize() assert exit.called
def _daemonize(self, m_setsid=Mock(), m_exit=Mock(), m_fork=Mock(return_value=42)): with patch('os.fork', m_fork), patch('os._exit', m_exit) as mock_os_exit,\ mock.patch('os.setsid', m_setsid): utils.daemonize() return mock_os_exit
def test_daemonize_from_child_success(self, m_exit): master_pid = 0 child_pid = 1 with mock.patch('os.fork', mock.Mock(side_effect=[master_pid, child_pid])) as m_fork: with mock.patch('os.setsid', mock.Mock()) as m_setsid: daemonize() m_setsid.assert_called_once_with() self.assertEqual(m_fork.call_count, 2) m_exit.assert_called_once_with(0)
def test_daemonize_parent(self): pid = 12 mock_os_fork = mock.Mock(return_value=pid) mock_os_exit = mock.Mock() with mock.patch('source.notification_pusher.os.fork', mock_os_fork): with mock.patch('source.notification_pusher.os._exit', mock_os_exit): utils.daemonize() mock_os_exit.assert_called_once_with(0) mock_os_fork.assert_called_once_with()
def test_daemonize_parent(self): pid = 12345 with mock.patch('os.fork', mock.Mock(return_value=pid)) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize() self.assertEqual(os_fork.call_count, 1) os_exit.assert_called_once_with(0) self.assertEqual(os_setsid.call_count, 0) pass
def test_daemonize_child(self): pid = 12345 with mock.patch('os.fork', mock.Mock(side_effect=[0,pid])) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize(); os_exit.assert_called_once_with(0) self.assertEqual(os_fork.call_count, 2) self.assertEqual(os_setsid.call_count, 1) pass
def test_daemonize_parent(self): pid = 12345 with mock.patch('os.fork', mock.Mock(return_value=pid)) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize(); self.assertEqual(os_fork.call_count, 1) os_exit.assert_called_once_with(0) self.assertEqual(os_setsid.call_count, 0) pass
def test_daemonize_child_child(self): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize(); self.assertEqual(os_fork.call_count, 2) self.assertEqual(os_exit.call_count, 0) self.assertEqual(os_setsid.call_count, 1) pass
def test_daemonize_child(self): pid = 12345 with mock.patch('os.fork', mock.Mock(side_effect=[0, pid])) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize() os_exit.assert_called_once_with(0) self.assertEqual(os_fork.call_count, 2) self.assertEqual(os_setsid.call_count, 1) pass
def test_daemonize_child_child(self): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)) as os_fork: with mock.patch('os.setsid', mock.Mock()) as os_setsid: with mock.patch('os._exit', mock.Mock()) as os_exit: utils.daemonize() self.assertEqual(os_fork.call_count, 2) self.assertEqual(os_exit.call_count, 0) self.assertEqual(os_setsid.call_count, 1) pass
def test_daemonize_child(self): test_pid = 0 with mock.patch('os.fork', mock.Mock(return_value=test_pid)) as os_fork: with mock.patch('os._exit', mock.Mock()) as os_exit: with mock.patch('os.setsid', mock.Mock()) as os_setsid: utils.daemonize() os_setsid.assert_called_once_with() self.assertTrue(os_fork.called, 'fork() has not been called') self.assertFalse(os_exit.called, 'exit should not have been called')
def test_daemonize_child_parent(self): pid = 0 mock_os_fork = mock.Mock(side_effect=[pid, 12]) mock_os_exit = mock.Mock() mock_os_setsid = mock.Mock() with mock.patch('source.notification_pusher.os.fork', mock_os_fork): with mock.patch('source.notification_pusher.os._exit', mock_os_exit): with mock.patch('source.notification_pusher.os.setsid', mock_os_setsid): utils.daemonize() mock_os_exit.assert_called_once_with(0) mock_os_setsid.assert_called_once_with() self.assertTrue(mock_os_fork.call_count, 2)
def test_daemonize_child_child(self): pid = 0 mock_os_fork = mock.Mock(return_value=pid) mock_os_exit = mock.Mock() mock_os_setsid = mock.Mock() with mock.patch('source.notification_pusher.os.fork', mock_os_fork): with mock.patch('source.notification_pusher.os._exit', mock_os_exit): with mock.patch('source.notification_pusher.os.setsid', mock_os_setsid): utils.daemonize() self.assertFalse(mock_os_exit.called) mock_os_setsid.assert_called_once_with() self.assertTrue(mock_os_fork.call_count, 2)
def main(argv): """ Точка входа в приложение. В случае возникновения ошибки в приложении, оно засыпает на config.SLEEP_ON_FAIL секунд. :param argv: агрументы командной строки. :type argv: list """ args = parse_cmd_args(argv[1:]) if args.daemon: daemonize() if args.pidfile: create_pidfile(args.pidfile) config = load_config_from_pyfile( os.path.realpath(os.path.expanduser(args.config)) ) patch_all() dictConfig(config.LOGGING) current_thread().name = 'pusher.main' install_signal_handlers() while run: try: main_loop(config) except Exception as exc: logger.error( 'Error in main loop. Go to sleep on {} second(s).'.format(config.SLEEP_ON_FAIL) ) logger.exception(exc) sleep(config.SLEEP_ON_FAIL) logger.info('Stop application loop in main.') return exit_code
def test_os_fork_called_two_times_when_os_fork_returned_zero_first_time(self): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)) as os_fork_mock: with mock.patch('os.setsid'): utils.daemonize() self.assertEqual(2, os_fork_mock.call_count)
def test_daemonize_from_master(self, m_exit): pid = 1 with mock.patch('os.fork', mock.Mock(return_value=pid)): daemonize() m_exit.assert_called_once_with(0)
def test_os_setsid_called_when_os_fork_returned_zero_first_time(self): pid = 0 with mock.patch('os.fork', mock.Mock(return_value=pid)): with mock.patch('os.setsid') as os_setsid_mock: utils.daemonize() os_setsid_mock.assert_calls_once_with()
def _daemonize(self, m_setsid=Mock(), m_exit=Mock(), m_fork=Mock(return_value=42)): with patch('os.fork', m_fork), patch('os._exit', m_exit) as mock_os_exit,\ mock.patch('os.setsid', m_setsid): utils.daemonize() return mock_os_exit
def test_os_exit_called_once_when_os_fork_returned_valid_pid_first_time(self): pid = 1 with mock.patch('os.fork', mock.Mock(return_value=pid)): with mock.patch('os._exit') as os_exit_mock: utils.daemonize() os_exit_mock.assert_called_once_with(0)