Cleanup part 2

mongodb · Sep 12, 2024 · 4df385c · 4df385c
1 parent 3f7dce5
commit 4df385c
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/test/asynchronous/test_client.py b/test/asynchronous/test_client.py
@@ -1881,11 +1881,11 @@ async def test_service_name_from_kwargs(self):
         self.assertEqual(client._topology_settings.srv_service_name, "customname")
 
     async def test_srv_max_hosts_kwarg(self):
-        client = AsyncMongoClient("mongodb+srv://test1.test.build.10gen.cc/")
+        client = self.simple_client("mongodb+srv://test1.test.build.10gen.cc/")
         self.assertGreater(len(client.topology_description.server_descriptions()), 1)
-        client = AsyncMongoClient("mongodb+srv://test1.test.build.10gen.cc/", srvmaxhosts=1)
+        client = self.simple_client("mongodb+srv://test1.test.build.10gen.cc/", srvmaxhosts=1)
         self.assertEqual(len(client.topology_description.server_descriptions()), 1)
-        client = AsyncMongoClient(
+        client = self.simple_client(
             "mongodb+srv://test1.test.build.10gen.cc/?srvMaxHosts=1", srvmaxhosts=2
         )
         self.assertEqual(len(client.topology_description.server_descriptions()), 2)

diff --git a/test/asynchronous/test_encryption.py b/test/asynchronous/test_encryption.py
@@ -230,7 +230,7 @@ def create_client_encryption(
         kms_providers: Mapping[str, Any],
         key_vault_namespace: str,
         key_vault_client: AsyncMongoClient,
-        codec_options: Mapping[str, Any],
+        codec_options: CodecOptions,
         kms_tls_options: Optional[Mapping[str, Any]] = None,
     ):
         client_encryption = AsyncClientEncryption(
@@ -245,7 +245,7 @@ def unmanaged_create_client_encryption(
         kms_providers: Mapping[str, Any],
         key_vault_namespace: str,
         key_vault_client: AsyncMongoClient,
-        codec_options: Mapping[str, Any],
+        codec_options: CodecOptions,
         kms_tls_options: Optional[Mapping[str, Any]] = None,
     ):
         client_encryption = AsyncClientEncryption(

diff --git a/test/test_client.py b/test/test_client.py
@@ -1839,11 +1839,11 @@ def test_service_name_from_kwargs(self):
         self.assertEqual(client._topology_settings.srv_service_name, "customname")
 
     def test_srv_max_hosts_kwarg(self):
-        client = MongoClient("mongodb+srv://test1.test.build.10gen.cc/")
+        client = self.simple_client("mongodb+srv://test1.test.build.10gen.cc/")
         self.assertGreater(len(client.topology_description.server_descriptions()), 1)
-        client = MongoClient("mongodb+srv://test1.test.build.10gen.cc/", srvmaxhosts=1)
+        client = self.simple_client("mongodb+srv://test1.test.build.10gen.cc/", srvmaxhosts=1)
         self.assertEqual(len(client.topology_description.server_descriptions()), 1)
-        client = MongoClient(
+        client = self.simple_client(
             "mongodb+srv://test1.test.build.10gen.cc/?srvMaxHosts=1", srvmaxhosts=2
         )
         self.assertEqual(len(client.topology_description.server_descriptions()), 2)

diff --git a/test/test_encryption.py b/test/test_encryption.py
@@ -230,7 +230,7 @@ def create_client_encryption(
         kms_providers: Mapping[str, Any],
         key_vault_namespace: str,
         key_vault_client: MongoClient,
-        codec_options: Mapping[str, Any],
+        codec_options: CodecOptions,
         kms_tls_options: Optional[Mapping[str, Any]] = None,
     ):
         client_encryption = ClientEncryption(
@@ -245,7 +245,7 @@ def unmanaged_create_client_encryption(
         kms_providers: Mapping[str, Any],
         key_vault_namespace: str,
         key_vault_client: MongoClient,
-        codec_options: Mapping[str, Any],
+        codec_options: CodecOptions,
         kms_tls_options: Optional[Mapping[str, Any]] = None,
     ):
         client_encryption = ClientEncryption(