if not found :
raise AssertionError("RRset not found in answer\n\n%s" % ret)
+ def sortRRsets(self, rrsets):
+ """Sorts RRsets in a more useful way than dnspython's default behaviour
+
+ @param rrsets: an array of dns.rrset.RRset objects"""
+
+ return sorted(rrsets, key=lambda rrset: (rrset.name, rrset.rdtype))
+
def assertAnyRRsetInAnswer(self, msg, rrsets):
"""Asserts that any of the supplied rrsets exists (without comparing TTL)
in the answer section of msg
res = self.sendUDPQuery(query)
self.assertRcodeEqual(res, dns.rcode.NOERROR)
- self.assertEqual(sorted(res.answer), sorted(response.answer))
+ self.assertEqual(self.sortRRsets(res.answer), self.sortRRsets(response.answer))
if __name__ == '__main__':
unittest.main()