github
/
fail2ban
mirror of https://github.com/fail2ban/fail2ban
1
0
Fork 0
Code Issues Releases Wiki Activity

Updated DNSUtilsTests to test use_dns and added positive test to testTextToIp

pull/27/head
Lee Clemens 2012-01-10 19:18:41 -05:00
parent 4fae37e46f
commit 4b18afb28a
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
testcases/filtertestcase.py
View File

@ -222,12 +222,24 @@ class GetFailures(unittest.TestCase):
class DNSUtilsTests(unittest.TestCase): class DNSUtilsTests(unittest.TestCase):
def testTextToIp(self, useDns): def testUseDns(self):
# Bogus addresses which should have no DNS matches res = DNSUtils.textToIp('www.example.com', 'no')
bogus = [ self.assertEqual(res, None)
res = DNSUtils.textToIp('www.example.com', 'warn')
self.assertEqual(res, ['192.0.43.10'])
res = DNSUtils.textToIp('www.example.com', 'yes')
self.assertEqual(res, ['192.0.43.10'])
def testTextToIp(self):
# Test hostnames
hostnames = [
'www.example.com'
'doh1.2.3.4.buga.xxxxx.yyy.invalid', 'doh1.2.3.4.buga.xxxxx.yyy.invalid',
'1.2.3.4.buga.xxxxx.yyy.invalid', '1.2.3.4.buga.xxxxx.yyy.invalid',
] ]
for s in bogus: for s in hostnames:
res = DNSUtils.textToIp(s, useDns) res = DNSUtils.textToIp(s, 'yes')
if s is 'www.example.com':
self.assertEqual(res, ['192.0.43.10'])
else:
self.assertEqual(res, []) self.assertEqual(res, [])