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')