def test_validate_ips_fail(self):
xml_source = ('<?xml version="1.0" encoding="UTF-8" ?>'
'<feedback>'
' <version>1.0</version>'
' <record>'
' <row>'
' <source_ip>10.10.10.10</source_ip>'
' </row>'
' </record>'
'</feedback>')
xml_tree = xml.etree.ElementTree.parse(io.StringIO(xml_source))
domain_ips = set(['10.10.10.1'])
report = RuaReport(xml_tree, domain_ips)
report.validate_ips()
self.assertFalse(report.ok(),
msg='Domain and report IPs matched unexpectedly')
self.assertEqual(len(report.errors), 1, msg='Unexpected errors count')
self.assertTrue('do not match expected IPs' in report.errors[0],
msg='Error string mismatch on IPs check')
def test_validate_ips_pass(self):
xml_source = ('<?xml version="1.0" encoding="UTF-8" ?>'
'<feedback>'
' <version>1.0</version>'
' <record>'
' <row>'
' <source_ip>10.10.10.10</source_ip>'
' </row>'
' </record>'
' <record>'
' <row>'
' <source_ip>192.168.1.1</source_ip>'
' </row>'
' </record>'
'</feedback>')
xml_tree = xml.etree.ElementTree.parse(io.StringIO(xml_source))
domain_ips = set(['192.168.1.1', '10.10.10.10'])
report = RuaReport(xml_tree, domain_ips)
report.validate_ips()
self.assertTrue(report.ok(),
msg='Domain and report IPs sets did not match')
self.assertFalse(len(report.errors), msg='Unexpected errors')