TST: get/set use DNS on Filters

pull/483/head
Daniel Black 2013-12-11 10:43:47 +11:00
parent a8b5c5b5f3
commit ebf4a02004
2 changed files with 16 additions and 1 deletions

View File

@ -174,6 +174,7 @@ tests.addTest(unittest.makeSuite(misctestcase.CustomDateFormatsTest))
# Filter # Filter
if not opts.no_network: if not opts.no_network:
tests.addTest(unittest.makeSuite(filtertestcase.IgnoreIP)) tests.addTest(unittest.makeSuite(filtertestcase.IgnoreIP))
tests.addTest(unittest.makeSuite(filtertestcase.BasicFilter))
tests.addTest(unittest.makeSuite(filtertestcase.LogFile)) tests.addTest(unittest.makeSuite(filtertestcase.LogFile))
tests.addTest(unittest.makeSuite(filtertestcase.LogFileMonitor)) tests.addTest(unittest.makeSuite(filtertestcase.LogFileMonitor))
if not opts.no_network: if not opts.no_network:

View File

@ -31,7 +31,7 @@ import tempfile
from server.jail import Jail from server.jail import Jail
from server.filterpoll import FilterPoll from server.filterpoll import FilterPoll
from server.filter import FileFilter, DNSUtils from server.filter import Filter, FileFilter, DNSUtils
from server.failmanager import FailManager from server.failmanager import FailManager
from server.failmanager import FailManagerEmpty from server.failmanager import FailManagerEmpty
@ -144,6 +144,20 @@ def _copy_lines_between_files(fin, fout, n=None, skip=0, mode='a', terminal_line
# Actual tests # Actual tests
# #
class BasicFilter(unittest.TestCase):
def setUp(self):
self.filter = Filter('name')
def testGetSetUseDNS(self):
# default is warn
self.assertEqual(self.filter.getUseDns(), 'warn')
self.filter.setUseDns(True)
self.assertEqual(self.filter.getUseDns(), 'yes')
self.filter.setUseDns(False)
self.assertEqual(self.filter.getUseDns(), 'no')
class IgnoreIP(LogCaptureTestCase): class IgnoreIP(LogCaptureTestCase):
def setUp(self): def setUp(self):