prehistoric-systems/fail2ban
1
0
Fork 0
mirror of https://github.com/fail2ban/fail2ban.git synced 2026-03-11 08:55:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

code review

Browse source
This commit is contained in:
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 _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):
self._recs = list()
self._strm = StringIO()
logging.StreamHandler.__init__(self, self._strm)
#
def truncate(self, size=None):
"""Truncate the internal buffer and records."""
if size:
@ -461,7 +469,7 @@ class LogCaptureTestCase(unittest.TestCase):
self._strm.truncate(0)
self._val = None
self._recs = list()
#
def getvalue(self):
"""Return current buffer as whole string."""
# cached:
@ -474,7 +482,7 @@ class LogCaptureTestCase(unittest.TestCase):
# cache and return:
self._val = self._strm.getvalue()
return self._val
#
def handle(self, record):
"""Handle the specified record."""
self._val = None