def test_main(self, parse_args_mock, join_mock, start_mock, Controller_init_mock, try_to_connect_mock, update_log_level_mock, write_pid_mock, perform_prestart_checks_mock, update_config_from_file_mock, stop_mock, bind_signal_handlers_mock, setup_logging_mock): Controller_init_mock.return_value = None options = MagicMock() parse_args_mock.return_value = (options, MagicMock) tmpdir = tempfile.gettempdir() #testing call without command-line arguments os.environ["AGENT_WORK_ROOT"] = os.path.join(tmpdir, "work") os.environ["AGENT_LOG_ROOT"] = os.path.join(tmpdir, "log") main.main() self.assertTrue(setup_logging_mock.called) self.assertTrue(bind_signal_handlers_mock.called) self.assertTrue(stop_mock.called) self.assertTrue(update_config_from_file_mock.called) self.assertTrue(perform_prestart_checks_mock.called) self.assertTrue(write_pid_mock.called) self.assertTrue(update_log_level_mock.called) try_to_connect_mock.assert_called_once_with(ANY, -1, ANY) self.assertTrue(start_mock.called)
def test_main(self, parse_args_mock, isAlive_mock, join_mock, start_mock, Controller_init_mock, AgentConfig_set_mock, try_to_connect_mock, update_log_level_mock, write_pid_mock, perform_prestart_checks_mock, update_config_from_file_mock, stop_mock, bind_signal_handlers_mock, setup_logging_mock): Controller_init_mock.return_value = None isAlive_mock.return_value = False parse_args_mock.return_value = (TestMain.AgentOptions( "agent", "host1", "8080", True), []) tmpdir = tempfile.gettempdir() #testing call without command-line arguments os.environ["AGENT_WORK_ROOT"] = os.path.join(tmpdir, "work") os.environ["AGENT_LOG_ROOT"] = os.path.join(tmpdir, "log") main.main() self.assertTrue(AgentConfig_set_mock.call_count == 2) AgentConfig_set_mock.assert_any_call("server", "hostname", "host1") AgentConfig_set_mock.assert_any_call("server", "port", "8080")
def test_main(self, parse_args_mock, join_mock, start_mock, Controller_init_mock, AgentConfig_set_mock, try_to_connect_mock, update_log_level_mock, write_pid_mock, perform_prestart_checks_mock, update_config_from_file_mock, stop_mock, bind_signal_handlers_mock, setup_logging_mock): Controller_init_mock.return_value = None parse_args_mock.return_value = ( TestMain.AgentOptions("agent", "host1", "8080", True), []) tmpdir = tempfile.gettempdir() #testing call without command-line arguments os.environ["AGENT_WORK_ROOT"] = os.path.join(tmpdir, "work") os.environ["AGENT_LOG_ROOT"] = os.path.join(tmpdir, "log") main.main() self.assertTrue(AgentConfig_set_mock.call_count == 2) AgentConfig_set_mock.assert_any_call("server", "hostname", "host1") AgentConfig_set_mock.assert_any_call("server", "port", "8080")