PYTHON-4493 Remove unused buffer size arg, await future in close()

Author: ShaneHarvey

pymongo/network_layer.py

@@ -466,7 +466,7 @@ def recv_into(self, buffer: bytes) -> int:
 
 
 class PyMongoProtocol(BufferedProtocol):
-    def __init__(self, timeout: Optional[float] = None, buffer_size: int = 2**10):
+    def __init__(self, timeout: Optional[float] = None):
         self.transport: Transport = None  # type: ignore[assignment]
         # Each message is reader in 2-3 parts: header, compression header, and message body
         # The message buffer is allocated after the header is read.
@@ -655,13 +655,10 @@ def connection_lost(self, exc: Exception | None) -> None:
             self._done_messages.append(msg)
 
         if not self._closed.done():
-            if exc is None:
-                self._closed.set_result(None)
-            else:
-                self._closed.set_exception(exc)
+            self._closed.set_result(None)
 
     async def wait_closed(self) -> None:
-        await asyncio.wait([self._closed])
+        await self._closed
 
 
 async def async_sendall(conn: PyMongoProtocol, buf: bytes) -> None:

pymongo/pool_shared.py

@@ -337,7 +337,7 @@ async def _configured_protocol_interface(
     if ssl_context is None:
         return AsyncNetworkingInterface(
             await asyncio.get_running_loop().create_connection(
-                lambda: PyMongoProtocol(timeout=timeout, buffer_size=2**16), sock=sock
+                lambda: PyMongoProtocol(timeout=timeout), sock=sock
             )
         )
 
@@ -346,7 +346,7 @@ async def _configured_protocol_interface(
         # We have to pass hostname / ip address to wrap_socket
         # to use SSLContext.check_hostname.
         transport, protocol = await asyncio.get_running_loop().create_connection(  # type: ignore[call-overload]
-            lambda: PyMongoProtocol(timeout=timeout, buffer_size=2**14),
+            lambda: PyMongoProtocol(timeout=timeout),
             sock=sock,
             server_hostname=host,
             ssl=ssl_context,