Update test_import_dns_resolver for dnspython 1.X

NoahStapp · NoahStapp · commit d53eee5bc7b3 · 2025-06-27T10:03:50.000-04:00
diff --git a/test/asynchronous/test_srv_polling.py b/test/asynchronous/test_srv_polling.py
@@ -363,7 +363,7 @@ def test_import_dns_resolver(self):
         # Regression test for PYTHON-4407
         import dns.resolver
 
-        self.assertTrue(hasattr(dns.resolver, "resolve"))
+        self.assertTrue(hasattr(dns.resolver, "resolve") or hasattr(dns.resolver, "query"))
 
 
 if __name__ == "__main__":
diff --git a/test/test_srv_polling.py b/test/test_srv_polling.py
@@ -363,7 +363,7 @@ def test_import_dns_resolver(self):
         # Regression test for PYTHON-4407
         import dns.resolver
 
-        self.assertTrue(hasattr(dns.resolver, "resolve"))
+        self.assertTrue(hasattr(dns.resolver, "resolve") or hasattr(dns.resolver, "query"))
 
 
 if __name__ == "__main__":
