PYTHON-4324 Wrap receive_data errors in pymongo errors

Author: ShaneHarvey

pymongo/asynchronous/pool.py

@@ -216,7 +216,10 @@ async def complete_pending(self) -> None:
             # TODO: respect deadline
             await self.receive_message(None, True)
         else:
-            network_layer.receive_data(self, self.pending_bytes, deadline)  # type:ignore[call-arg]
+            try:
+                network_layer.receive_data(self, self.pending_bytes, deadline)  # type:ignore[call-arg]
+            except BaseException as error:
+                await self._raise_connection_failure(error)
         self.pending_response = False
         self.pending_bytes = 0
         self.pending_deadline = 0.0

pymongo/synchronous/pool.py

@@ -216,7 +216,10 @@ def complete_pending(self) -> None:
             # TODO: respect deadline
             self.receive_message(None, True)
         else:
-            network_layer.receive_data(self, self.pending_bytes, deadline)  # type:ignore[call-arg]
+            try:
+                network_layer.receive_data(self, self.pending_bytes, deadline)  # type:ignore[call-arg]
+            except BaseException as error:
+                self._raise_connection_failure(error)
         self.pending_response = False
         self.pending_bytes = 0
         self.pending_deadline = 0.0