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