def setUp(self):
self.status = LoggerStatus(
unknown=sentinel.unknown,
errors=sentinel.errors,
warnings=sentinel.warnings,
important=sentinel.important,
)
class AddStuffTests(TestCase):
def setUp(self):
self.status = LoggerStatus(
unknown=sentinel.unknown,
errors=sentinel.errors,
warnings=sentinel.warnings,
important=sentinel.important,
)
def test_add_error(self):
status = self.status._replace(errors=())
status = status.add_error("<message_1>")
status = status.add_error("<message_2>")
self.assertSize(status.errors, 2)
self.assertEqual(status.errors[0], "<message_1>")
self.assertEqual(status.errors[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.warnings, sentinel.warnings)
self.assertEqual(status.important, sentinel.important)
def test_add_warning(self):
status = self.status._replace(warnings=())
status = status.add_warning("<message_1>")
status = status.add_warning("<message_2>")
self.assertSize(status.warnings, 2)
self.assertEqual(status.warnings[0], "<message_1>")
self.assertEqual(status.warnings[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.important, sentinel.important)
def test_add_important(self):
status = self.status._replace(important=())
status = status.add_important("<message_1>")
status = status.add_important("<message_2>")
self.assertSize(status.important, 2)
self.assertEqual(status.important[0], "<message_1>")
self.assertEqual(status.important[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.warnings, sentinel.warnings)
def test_set_unknown(self):
status = self.status._replace(unknown=False)
status = status.set_unknown()
self.assertEqual(status.unknown, True)
self.assertEqual(status.important, sentinel.important)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.warnings, sentinel.warnings)
class AddStuffTests(TestCase):
def setUp(self):
self.status = LoggerStatus(
unknown=sentinel.unknown,
errors=sentinel.errors,
warnings=sentinel.warnings,
important=sentinel.important,
)
def test_add_error(self):
status = self.status._replace(errors=())
status = status.add_error("<message_1>")
status = status.add_error("<message_2>")
self.assertSize(status.errors, 2)
self.assertEqual(status.errors[0], "<message_1>")
self.assertEqual(status.errors[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.warnings, sentinel.warnings)
self.assertEqual(status.important, sentinel.important)
def test_add_warning(self):
status = self.status._replace(warnings=())
status = status.add_warning("<message_1>")
status = status.add_warning("<message_2>")
self.assertSize(status.warnings, 2)
self.assertEqual(status.warnings[0], "<message_1>")
self.assertEqual(status.warnings[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.important, sentinel.important)
def test_add_important(self):
status = self.status._replace(important=())
status = status.add_important("<message_1>")
status = status.add_important("<message_2>")
self.assertSize(status.important, 2)
self.assertEqual(status.important[0], "<message_1>")
self.assertEqual(status.important[1], "<message_2>")
self.assertEqual(status.unknown, sentinel.unknown)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.warnings, sentinel.warnings)
def test_set_unknown(self):
status = self.status._replace(unknown=False)
status = status.set_unknown()
self.assertEqual(status.unknown, True)
self.assertEqual(status.important, sentinel.important)
self.assertEqual(status.errors, sentinel.errors)
self.assertEqual(status.warnings, sentinel.warnings)
def setUp(self):
self.status = LoggerStatus(
unknown=sentinel.unknown,
errors=sentinel.errors,
warnings=sentinel.warnings,
important=sentinel.important,
)
def setUp(self):
self.mock_list_messages = self.patch(
'pignacio_scripts.nagios.logger.list_messages', autospec=True)
self.mock_get_first_line = self.patch(
'pignacio_scripts.nagios.logger.get_first_line', autospec=True)
list_message_returns = {
sentinel.errors: [sentinel.listed_errors],
sentinel.warnings: [sentinel.listed_warnings],
sentinel.important: [sentinel.listed_important],
}
self.mock_list_messages.side_effect = (
lambda x, *a, **kw: list_message_returns.get(x, []))
self.mock_get_first_line.return_value = sentinel.first_line
self.status = LoggerStatus(
unknown=sentinel.unknown,
errors=sentinel.errors,
warnings=sentinel.warnings,
important=sentinel.important,
)
def setUp(self):
self.mock_get_output = self.patch(
'pignacio_scripts.nagios.logger.get_output', autospec=True)
self.mock_get_output.return_value = sentinel.output
self.mock_print_lines = self.patch(
'pignacio_scripts.nagios.logger.print_lines', autospec=True)
self.mock_sys_exit = self.patch(
'pignacio_scripts.nagios.logger.sys.exit', autospec=True)
self.mock_empty_to_whitespace = self.patch(
'pignacio_scripts.nagios.logger.empty_lines_to_whitespace',
autospec=True)
self.mock_empty_to_whitespace.return_value = sentinel.whitespace_output
self.status = LoggerStatus(
unknown=sentinel.unknown,
warnings=sentinel.warnings,
errors=sentinel.errors,
important=sentinel.important,
)
self.mock_exit_code = self.patch_object(self.status,
'exit_code',
autospec=True)
self.mock_exit_code.return_value = sentinel.exit_code
def setUp(self):
self.status = LoggerStatus.initial()
def setUp(self):
self.status = LoggerStatus.initial()
self.mock_exit_code = self.patch_object(self.status, 'exit_code',
autospec=True)
self.mock_exit_code.return_value = LoggerStatus.EXIT_OK
def setUp(self):
self.status = LoggerStatus.initial()
def setUp(self):
self.status = LoggerStatus.initial()
self.mock_exit_code = self.patch_object(self.status,
'exit_code',
autospec=True)
self.mock_exit_code.return_value = LoggerStatus.EXIT_OK
