Fixes

Author: NoahStapp

pymongo/asynchronous/pool.py

@@ -1045,8 +1045,8 @@ async def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> A
             if self.handshake:
                 await conn.hello()
                 self.is_writable = conn.is_writable
-            # if handler:
-            #     handler.contribute_socket(conn, completed_handshake=False)
+            if handler:
+                handler.contribute_socket(conn, completed_handshake=False)
 
             await conn.authenticate()
         # Catch KeyboardInterrupt, CancelledError, etc. and cleanup.

pymongo/network_layer.py

@@ -529,19 +529,11 @@ async def read(
             self._debug = debug
             self._max_message_size = max_message_size
             self._request_id = request_id
-            (
-                self._length,
-                self._overflow_length,
-                self._body_length,
-                self._op_code,
-                self._overflow,
-            ) = (
-                0,
-                0,
-                0,
-                None,  # type: ignore[assignment]
-                None,
-            )
+            self._length = 0
+            self._overflow_length = 0
+            self._body_length = 0
+            self._op_code = None  # type: ignore[assignment]
+            self._overflow = None
             if self.transport.is_closing():
                 raise OSError("Connection is closed")
             read_waiter = asyncio.get_running_loop().create_future()

pymongo/synchronous/pool.py

@@ -1041,8 +1041,8 @@ def connect(self, handler: Optional[_MongoClientErrorHandler] = None) -> Connect
             if self.handshake:
                 conn.hello()
                 self.is_writable = conn.is_writable
-            # if handler:
-            #     handler.contribute_socket(conn, completed_handshake=False)
+            if handler:
+                handler.contribute_socket(conn, completed_handshake=False)
 
             conn.authenticate()
         # Catch KeyboardInterrupt, CancelledError, etc. and cleanup.