Merge branch '0.11'

pull/2337/merge
sebres 2021-03-25 12:14:11 +01:00
commit 80a33b1dee
3 changed files with 9 additions and 10 deletions

View File

@ -188,6 +188,7 @@ def _assert_correct_last_attempt(utest, filter_, output, count=None):
utest.assertEqual(len(found), 1) utest.assertEqual(len(found), 1)
_assert_equal_entries(utest, found[0], output, count) _assert_equal_entries(utest, found[0], output, count)
else: else:
utest.assertEqual(len(found), len(output))
# sort by string representation of ip (multiple failures with different ips): # sort by string representation of ip (multiple failures with different ips):
found = sorted(found, key=lambda x: str(x)) found = sorted(found, key=lambda x: str(x))
output = sorted(output, key=lambda x: str(x)) output = sorted(output, key=lambda x: str(x))

View File

@ -66,9 +66,12 @@ class TestServer(Server):
class TransmitterBase(LogCaptureTestCase): class TransmitterBase(LogCaptureTestCase):
TEST_SRV_CLASS = TestServer
def setUp(self): def setUp(self):
"""Call before every test case.""" """Call before every test case."""
super(TransmitterBase, self).setUp() super(TransmitterBase, self).setUp()
self.server = self.TEST_SRV_CLASS()
self.transm = self.server._Server__transm self.transm = self.server._Server__transm
# To test thransmitter we don't need to start server... # To test thransmitter we don't need to start server...
#self.server.start('/dev/null', '/dev/null', force=False) #self.server.start('/dev/null', '/dev/null', force=False)
@ -157,10 +160,6 @@ class TransmitterBase(LogCaptureTestCase):
class Transmitter(TransmitterBase): class Transmitter(TransmitterBase):
def setUp(self):
self.server = TestServer()
super(Transmitter, self).setUp()
def testServerIsNotStarted(self): def testServerIsNotStarted(self):
# so far isStarted only tested but not used otherwise # so far isStarted only tested but not used otherwise
# and here we don't really .start server # and here we don't really .start server
@ -937,8 +936,9 @@ class Transmitter(TransmitterBase):
class TransmitterLogging(TransmitterBase): class TransmitterLogging(TransmitterBase):
TEST_SRV_CLASS = Server
def setUp(self): def setUp(self):
self.server = Server()
super(TransmitterLogging, self).setUp() super(TransmitterLogging, self).setUp()
self.server.setLogTarget("/dev/null") self.server.setLogTarget("/dev/null")
self.server.setLogLevel("CRITICAL") self.server.setLogLevel("CRITICAL")

View File

@ -47,7 +47,7 @@ from ..server import asyncserver
from ..version import version from ..version import version
logSys = getLogger(__name__) logSys = getLogger("fail2ban")
TEST_NOW = 1124013600 TEST_NOW = 1124013600
@ -126,9 +126,6 @@ def getOptParser(doc=""):
def initProcess(opts): def initProcess(opts):
# Logger: # Logger:
global logSys
logSys = getLogger("fail2ban")
llev = None llev = None
if opts.log_level is not None: # pragma: no cover if opts.log_level is not None: # pragma: no cover
# so we had explicit settings # so we had explicit settings
@ -782,8 +779,9 @@ class LogCaptureTestCase(unittest.TestCase):
"""Call after every test case.""" """Call after every test case."""
# print "O: >>%s<<" % self._log.getvalue() # print "O: >>%s<<" % self._log.getvalue()
self.pruneLog() self.pruneLog()
self._log.close()
logSys.handlers = self._old_handlers logSys.handlers = self._old_handlers
logSys.level = self._old_level logSys.setLevel(self._old_level)
super(LogCaptureTestCase, self).tearDown() super(LogCaptureTestCase, self).tearDown()
def _is_logged(self, *s, **kwargs): def _is_logged(self, *s, **kwargs):