fix tests

Author: sleepyStick

pymongo/asynchronous/srv_resolver.py

@@ -56,9 +56,9 @@ async def _resolve(*args: Any, **kwargs: Any) -> resolver.Answer:
         # dnspython 1.X
         return resolver.query(*args, **kwargs)
     else:
-        from dns.asyncresolver import Resolver
+        from dns import asyncresolver
 
-        return await Resolver.resolve(*args, **kwargs)  # type:ignore[return-value]
+        return await asyncresolver.resolve(*args, **kwargs)  # type:ignore[return-value]
 
 
 _INVALID_HOST_MSG = (

pymongo/synchronous/srv_resolver.py

@@ -56,9 +56,9 @@ def _resolve(*args: Any, **kwargs: Any) -> resolver.Answer:
         # dnspython 1.X
         return resolver.query(*args, **kwargs)
     else:
-        from dns.asyncresolver import Resolver
+        from dns import asyncresolver
 
-        return Resolver.resolve(*args, **kwargs)  # type:ignore[return-value]
+        return asyncresolver.resolve(*args, **kwargs)  # type:ignore[return-value]
 
 
 _INVALID_HOST_MSG = (

test/asynchronous/test_client.py

@@ -607,7 +607,7 @@ def test_validate_suggestion(self):
             with self.assertRaisesRegex(ConfigurationError, expected):
                 AsyncMongoClient(**{typo: "standard"})  # type: ignore[arg-type]
 
-    @patch("pymongo.srv_resolver._SrvResolver.get_hosts")
+    @patch("pymongo.asynchronous.srv_resolver._SrvResolver.get_hosts")
     def test_detected_environment_logging(self, mock_get_hosts):
         normal_hosts = [
             "normal.host.com",
@@ -629,7 +629,7 @@ def test_detected_environment_logging(self, mock_get_hosts):
             logs = [record.getMessage() for record in cm.records if record.name == "pymongo.client"]
             self.assertEqual(len(logs), 7)
 
-    @patch("pymongo.srv_resolver._SrvResolver.get_hosts")
+    @patch("pymongo.asynchronous.srv_resolver._SrvResolver.get_hosts")
     async def test_detected_environment_warning(self, mock_get_hosts):
         with self._caplog.at_level(logging.WARN):
             normal_hosts = [