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, 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")
