increase code coverage

pull/1346/head
sebres 2014-10-28 21:57:16 +01:00
parent d22b2498d4
commit db5ee0842a
3 changed files with 16 additions and 5 deletions

View File

@ -194,7 +194,7 @@ class DateDetector(object):
logSys.log(logLevel, "Got time %f for \"%r\" using template %s", logSys.log(logLevel, "Got time %f for \"%r\" using template %s",
date[0], date[1].group(), template.name) date[0], date[1].group(), template.name)
return date return date
except ValueError: except ValueError: # pragma: no cover
pass pass
return None return None
finally: finally:

View File

@ -76,7 +76,8 @@ class Jail:
self.__queue = Queue.Queue() self.__queue = Queue.Queue()
self.__filter = None self.__filter = None
logSys.info("Creating new jail '%s'" % self.name) logSys.info("Creating new jail '%s'" % self.name)
self._setBackend(backend) if backend is not None:
self._setBackend(backend)
def __repr__(self): def __repr__(self):
return "%s(%r)" % (self.__class__.__name__, self.name) return "%s(%r)" % (self.__class__.__name__, self.name)
@ -106,11 +107,12 @@ class Jail:
logSys.info("Initiated %r backend" % b) logSys.info("Initiated %r backend" % b)
self.__actions = Actions(self) self.__actions = Actions(self)
return # we are done return # we are done
except ImportError, e: except ImportError, e: # pragma: no cover
# Log debug if auto, but error if specific # Log debug if auto, but error if specific
logSys.log( logSys.log(
logging.DEBUG if backend == "auto" else logging.ERROR, logging.DEBUG if backend == "auto" else logging.ERROR,
"Backend %r failed to initialize due to %s" % (b, e)) "Backend %r failed to initialize due to %s" % (b, e))
# pragma: no cover
# log error since runtime error message isn't printed, INVALID COMMAND # log error since runtime error message isn't printed, INVALID COMMAND
logSys.error( logSys.error(
"Failed to initialize any backend for Jail %r" % self.name) "Failed to initialize any backend for Jail %r" % self.name)

View File

@ -29,20 +29,29 @@ import time
import datetime import datetime
from ..server.datedetector import DateDetector from ..server.datedetector import DateDetector
from ..server import datedetector
from ..server.datetemplate import DateTemplate from ..server.datetemplate import DateTemplate
from .utils import setUpMyTime, tearDownMyTime from .utils import setUpMyTime, tearDownMyTime, LogCaptureTestCase
from ..helpers import getLogger
logSys = getLogger("fail2ban")
class DateDetectorTest(unittest.TestCase): class DateDetectorTest(LogCaptureTestCase):
def setUp(self): def setUp(self):
"""Call before every test case.""" """Call before every test case."""
LogCaptureTestCase.setUp(self)
self.__old_eff_level = datedetector.logLevel
datedetector.logLevel = logSys.getEffectiveLevel()
setUpMyTime() setUpMyTime()
self.__datedetector = DateDetector() self.__datedetector = DateDetector()
self.__datedetector.addDefaultTemplate() self.__datedetector.addDefaultTemplate()
def tearDown(self): def tearDown(self):
"""Call after every test case.""" """Call after every test case."""
LogCaptureTestCase.tearDown(self)
datedetector.logLevel = self.__old_eff_level
tearDownMyTime() tearDownMyTime()
def testGetEpochTime(self): def testGetEpochTime(self):