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 f6186eff14
commit cfb2c75b49
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
testcases/filtertestcase.py
View File

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