github
/
fail2ban
mirror of https://github.com/fail2ban/fail2ban
1
0
Fork 0
Code Issues Releases Wiki Activity

code review

pull/1510/head
sebres 2016-08-15 13:51:50 +02:00
parent 83029b89b9
commit 6fcf8192f3
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
fail2ban/tests/utils.py
View File

@ -449,11 +449,19 @@ if True: ## if not hasattr(unittest.TestCase, 'assertIn'):
class LogCaptureTestCase(unittest.TestCase): class LogCaptureTestCase(unittest.TestCase):
class _MemHandler(logging.StreamHandler): class _MemHandler(logging.StreamHandler):
"""Logging handler helper
Affords not to delegate logging to StreamHandler at all,
format lazily on demand in getvalue.
Increases performance inside the LogCaptureTestCase tests, because there
the log level set to DEBUG.
"""
def __init__(self): def __init__(self):
self._recs = list() self._recs = list()
self._strm = StringIO() self._strm = StringIO()
logging.StreamHandler.__init__(self, self._strm) logging.StreamHandler.__init__(self, self._strm)
#
def truncate(self, size=None): def truncate(self, size=None):
"""Truncate the internal buffer and records.""" """Truncate the internal buffer and records."""
if size: if size:
@ -461,7 +469,7 @@ class LogCaptureTestCase(unittest.TestCase):
self._strm.truncate(0) self._strm.truncate(0)
self._val = None self._val = None
self._recs = list() self._recs = list()
#
def getvalue(self): def getvalue(self):
"""Return current buffer as whole string.""" """Return current buffer as whole string."""
# cached: # cached:
@ -474,7 +482,7 @@ class LogCaptureTestCase(unittest.TestCase):
# cache and return: # cache and return:
self._val = self._strm.getvalue() self._val = self._strm.getvalue()
return self._val return self._val
#
def handle(self, record): def handle(self, record):
"""Handle the specified record.""" """Handle the specified record."""
self._val = None self._val = None