class TestSummarisingLogger(TestCase):
def setUp(self):
removeHandlers()
DummySMTP.install()
def tearDown(self):
DummySMTP.remove()
exit_handler_manager.clear_at_exit_handlers()
def test_imports(self):
pass
def create(self, *args, **kw):
kw['atexit'] = False
self.handler = SummarisingLogger(*args, **kw)
self.logger = logging.getLogger('')
self.logger.addHandler(self.handler)
def test_do_send_empty(self):
self.create('*****@*****.**', ('*****@*****.**', ))
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
def test_dont_send_empty(self):
self.create('*****@*****.**', ('*****@*****.**', ),
send_empty_entries=False)
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 0)
def test_send_level_filters(self):
self.create('*****@*****.**', ('*****@*****.**', ),
send_level=logging.CRITICAL)
self.logger.warning('This line will not be sent')
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 0)
def test_send_level_includes_lower_level(self):
self.create('*****@*****.**', ('*****@*****.**', ),
send_level=logging.CRITICAL)
self.logger.warning('a warning')
self.logger.critical('something critical')
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
message_text = DummySMTP.sent[0].msg
_check_sent_message('a warning', message_text)
_check_sent_message('something critical', message_text)
def test_tmpfile_goes_away(self):
self.create('*****@*****.**', ('*****@*****.**', ))
os.remove(self.handler.filename)
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
def test_default_charset(self):
self.create(
'*****@*****.**',
('*****@*****.**', ),
)
self.logger.critical(u"accentu\u00E9")
logging.shutdown()
m = DummySMTP.sent[0].msg
# lovely, utf-8 encoded goodness
self.assertTrue('Subject: Summary of Log Messages (CRITICAL)' in m, m)
self.assertTrue('Content-Type: text/plain; charset="utf-8"' in m, m)
self.assertTrue('\nYWNjZW50dcOp' in m, m)
def test_specified_charset(self):
self.create('*****@*****.**', ('*****@*****.**', ),
charset='iso-8859-1')
self.logger.critical(u"accentu\u00E9")
logging.shutdown()
m = DummySMTP.sent[0].msg
# lovely, latin-1 encoded goodness
self.assertTrue('\naccentu=E9' in m, m)
self.assertTrue('Content-Type: text/plain; charset="iso-8859-1"' in m,
m)
self.assertTrue('Subject: Summary of Log Messages (CRITICAL)' in m, m)
def test_template(self):
self.create('*****@*****.**', ('*****@*****.**', ),
template='<before>%s<after>')
logging.critical('message')
logging.shutdown()
m = DummySMTP.sent[0].msg
self.assertTrue('Subject: Summary of Log Messages (CRITICAL)' in m, m)
_check_sent_message('<before>message\n<after>', m)
def test_specified_content_type(self):
self.create('*****@*****.**', ('*****@*****.**', ),
content_type='foo/bar')
self.logger.critical(u"message")
logging.shutdown()
m = DummySMTP.sent[0].msg
# NB: we drop the 'foo'
self.assertTrue('Content-Type: text/bar' in m, m)
def test_flood_level_exceeded(self):
self.create('*****@*****.**', ('*****@*****.**', ), flood_level=3)
self.handler.setFormatter(
logging.Formatter('%(levelname)s - %(message)s'))
for i in range(10):
logging.warning('message %s', i)
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
m = DummySMTP.sent[0].msg
self.assertTrue('Subject: Summary of Log Messages (WARNING)' in m, m)
_check_sent_message(
'\n'.join([
'WARNING - message 0',
'WARNING - message 1',
'WARNING - message 2',
'CRITICAL - 2 messages not included as flood limit of 3 exceeded',
'WARNING - message 5',
'WARNING - message 6',
'WARNING - message 7',
'WARNING - message 8',
]), m)
def test_flood_highest_level_still_recorded(self):
self.create('*****@*****.**', ('*****@*****.**', ), flood_level=1)
self.handler.setFormatter(
logging.Formatter('%(levelname)s - %(message)s'))
logging.info('included')
logging.warning('filtered')
for i in range(5):
logging.info('after %i', i)
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
m = DummySMTP.sent[0].msg
self.assertTrue('Subject: Summary of Log Messages (WARNING)' in m, m)
_check_sent_message(
'\n'.join([
'INFO - included',
'CRITICAL - 1 messages not included as flood limit of 1 exceeded',
'INFO - after 0',
]), m)
def test_flood_except_for_tail(self):
self.create('*****@*****.**', ('*****@*****.**', ), flood_level=1)
self.handler.setFormatter(
logging.Formatter('%(levelname)s - %(message)s'))
logging.warning('message 1')
logging.warning('message 2')
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 1)
m = DummySMTP.sent[0].msg
self.assertTrue('Subject: Summary of Log Messages (WARNING)' in m, m)
_check_sent_message(
'\n'.join([
'WARNING - message 1',
'WARNING - message 2',
]), m)
def test_reopen(self):
self.create('*****@*****.**', ('*****@*****.**', ))
self.handler.reopen()
logging.shutdown()
self.assertEqual(len(DummySMTP.sent), 2)
def test_safe_close(self):
self.create('*****@*****.**', ('*****@*****.**', ))
threads = []
for i in range(2):
t = threading.Thread(target=self.handler.close)
threads.append(t)
t.start()
[t.join() for t in threads]
self.assertEqual(len(DummySMTP.sent), 1)
