mirror of https://github.com/fail2ban/fail2ban
TST: Sample log regex test now warns if no log for a filter
Also checks that at least some tests are presentpull/290/head
parent
94376bfbe1
commit
1116f23151
|
@ -183,6 +183,8 @@ tests.addTest(unittest.makeSuite(datedetectortestcase.DateDetectorTest))
|
||||||
if json:
|
if json:
|
||||||
# Filter Regex tests with sample logs
|
# Filter Regex tests with sample logs
|
||||||
tests.addTest(unittest.makeSuite(samplestestcase.FilterSamplesRegex))
|
tests.addTest(unittest.makeSuite(samplestestcase.FilterSamplesRegex))
|
||||||
|
else:
|
||||||
|
print "I: Skipping filter samples testing. No simplejson/json module"
|
||||||
|
|
||||||
#
|
#
|
||||||
# Extensive use-tests of different available filters backends
|
# Extensive use-tests of different available filters backends
|
||||||
|
|
|
@ -22,7 +22,7 @@
|
||||||
__copyright__ = "Copyright (c) 2013 Steven Hiscocks"
|
__copyright__ = "Copyright (c) 2013 Steven Hiscocks"
|
||||||
__license__ = "GPL"
|
__license__ = "GPL"
|
||||||
|
|
||||||
import unittest, sys, os, fileinput, re, datetime
|
import unittest, sys, os, fileinput, re, datetime, inspect
|
||||||
|
|
||||||
if sys.version_info >= (2, 6):
|
if sys.version_info >= (2, 6):
|
||||||
import json
|
import json
|
||||||
|
@ -46,6 +46,14 @@ class FilterSamplesRegex(unittest.TestCase):
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
"""Call after every test case."""
|
"""Call after every test case."""
|
||||||
|
|
||||||
|
def testFiltersPresent(self):
|
||||||
|
"""Check to ensure some tests exist"""
|
||||||
|
self.assertTrue(
|
||||||
|
len([test for test in inspect.getmembers(self)
|
||||||
|
if test[0].startswith('testSampleRegexs')])
|
||||||
|
>= 10,
|
||||||
|
"Expected more FilterSampleRegexs tests")
|
||||||
|
|
||||||
def testSampleRegexsFactory(name):
|
def testSampleRegexsFactory(name):
|
||||||
def testFilter(self):
|
def testFilter(self):
|
||||||
|
|
||||||
|
@ -58,8 +66,22 @@ def testSampleRegexsFactory(name):
|
||||||
if opt[2] == "addfailregex":
|
if opt[2] == "addfailregex":
|
||||||
self.filter.addFailRegex(opt[3])
|
self.filter.addFailRegex(opt[3])
|
||||||
|
|
||||||
|
if not self.filter.getFailRegex():
|
||||||
|
# No fail regexs set: likely just common file for includes.
|
||||||
|
return
|
||||||
|
|
||||||
|
# TODO: Remove exception handling once sample logs obtained for all
|
||||||
|
try:
|
||||||
|
self.assertTrue(
|
||||||
|
os.path.isfile(os.path.join(TEST_FILES_DIR, "logs", name)),
|
||||||
|
"No sample log file available for '%s' filter" % name)
|
||||||
|
except AssertionError:
|
||||||
|
print "I: No sample log file available for '%s' filter" % name
|
||||||
|
return
|
||||||
|
|
||||||
logFile = fileinput.FileInput(
|
logFile = fileinput.FileInput(
|
||||||
os.path.join(TEST_FILES_DIR, "logs", name))
|
os.path.join(TEST_FILES_DIR, "logs", name))
|
||||||
|
|
||||||
for line in logFile:
|
for line in logFile:
|
||||||
jsonREMatch = re.match("^# ?failJSON:(.+)$", line)
|
jsonREMatch = re.match("^# ?failJSON:(.+)$", line)
|
||||||
if jsonREMatch:
|
if jsonREMatch:
|
||||||
|
@ -78,13 +100,13 @@ def testSampleRegexsFactory(name):
|
||||||
if not ret:
|
if not ret:
|
||||||
# Check line is flagged as none match
|
# Check line is flagged as none match
|
||||||
self.assertFalse(faildata.get('match', True),
|
self.assertFalse(faildata.get('match', True),
|
||||||
"Line not matched when should have: %s:%i" %
|
"Line not matched when should have: %s:%i %r" %
|
||||||
(logFile.filename(), logFile.filelineno()))
|
(logFile.filename(), logFile.filelineno(), line))
|
||||||
elif ret:
|
elif ret:
|
||||||
# Check line is flagged to match
|
# Check line is flagged to match
|
||||||
self.assertTrue(faildata.get('match', False),
|
self.assertTrue(faildata.get('match', False),
|
||||||
"Line matched when shouldn't have: %s:%i" %
|
"Line matched when shouldn't have: %s:%i %r" %
|
||||||
(logFile.filename(), logFile.filelineno()))
|
(logFile.filename(), logFile.filelineno(), line))
|
||||||
self.assertEqual(len(ret), 1)
|
self.assertEqual(len(ret), 1)
|
||||||
# Verify timestamp and host as expected
|
# Verify timestamp and host as expected
|
||||||
host, time = ret[0]
|
host, time = ret[0]
|
||||||
|
@ -96,9 +118,9 @@ def testSampleRegexsFactory(name):
|
||||||
|
|
||||||
return testFilter
|
return testFilter
|
||||||
|
|
||||||
for filter_ in os.listdir(os.path.join(TEST_FILES_DIR, "logs")):
|
for filter_ in os.listdir(os.path.join(CONFIG_DIR, "filter.d")):
|
||||||
if os.path.isfile(os.path.join(TEST_FILES_DIR, "logs", filter_)):
|
filterName = filter_.rpartition(".")[0]
|
||||||
setattr(
|
setattr(
|
||||||
FilterSamplesRegex,
|
FilterSamplesRegex,
|
||||||
"testSampleRegexs%s" % filter_.upper(),
|
"testSampleRegexs%s" % filterName.upper(),
|
||||||
testSampleRegexsFactory(filter_))
|
testSampleRegexsFactory(filterName))
|
||||||
|
|
Loading…
Reference in New Issue