eqdnsmessage: adjust to type changes in dnspython

(cherry picked from commit 255ec835c0c0475f86ec81ed838dd3014d365043)

diff --git a/contrib/assert-equal-DNSMessage/eqdnsmessage.py b/contrib/assert-equal-DNSMessage/eqdnsmessage.py
index 99a6784c993adc71c57ff4ee27342d6894bdca29..9a3605e0fce10302034cc9a67273e5cebdcbbe83 100644 (file)
--- a/contrib/assert-equal-DNSMessage/eqdnsmessage.py
+++ b/contrib/assert-equal-DNSMessage/eqdnsmessage.py
@@ -26,5 +26,6 @@ class AssertEqualDNSMessageMixin(unittest.TestCase):
 
     def setUp(self):
         self.addTypeEqualityFunc(dns.message.Message, self.assertEqualDNSMessage)
+        self.addTypeEqualityFunc(dns.message.QueryMessage, self.assertEqualDNSMessage)
 
-        super(AssertEqualDNSMessageMixin, self).setUp()
\ No newline at end of file
+        super(AssertEqualDNSMessageMixin, self).setUp()