fix test

Author: blink1073

test/asynchronous/test_auth.py

@@ -35,7 +35,7 @@
 import pytest
 
 from pymongo import AsyncMongoClient, monitoring
-from pymongo.asynchronous.auth import HAVE_KERBEROS
+from pymongo.asynchronous.auth import HAVE_KERBEROS, _canonicalize_hostname
 from pymongo.auth_shared import _build_credentials_tuple
 from pymongo.errors import OperationFailure
 from pymongo.hello import HelloCompat
@@ -331,6 +331,12 @@ async def test_gssapi_host_name(self):
         )
         await client.server_info()
 
+    def test_canonicalize_host_name(self):
+        result = _canonicalize_hostname(GSSAPI_HOST, "forward")
+        self.assertIn("compute-1.amazonaws.com", result)
+        result = _canonicalize_hostname(GSSAPI_HOST, "forwardAndReverse")
+        self.assertEqual(result, GSSAPI_HOST)
+
 
 class TestSASLPlain(AsyncPyMongoTestCase):
     @classmethod

test/test_auth.py

@@ -40,7 +40,7 @@
 from pymongo.hello import HelloCompat
 from pymongo.read_preferences import ReadPreference
 from pymongo.saslprep import HAVE_STRINGPREP
-from pymongo.synchronous.auth import HAVE_KERBEROS
+from pymongo.synchronous.auth import HAVE_KERBEROS, _canonicalize_hostname
 
 _IS_SYNC = True
 
@@ -331,6 +331,12 @@ def test_gssapi_host_name(self):
         )
         client.server_info()
 
+    def test_canonicalize_host_name(self):
+        result = _canonicalize_hostname(GSSAPI_HOST, "forward")
+        self.assertIn("compute-1.amazonaws.com", result)
+        result = _canonicalize_hostname(GSSAPI_HOST, "forwardAndReverse")
+        self.assertEqual(result, GSSAPI_HOST)
+
 
 class TestSASLPlain(PyMongoTestCase):
     @classmethod