Fix deadline

NoahStapp · NoahStapp · commit 3bdd38113d0a · 2024-10-07T09:46:09.000-04:00
diff --git a/pymongo/asynchronous/encryption.py b/pymongo/asynchronous/encryption.py
@@ -154,7 +154,7 @@ async def _async_kms_request(
                 while kms_context.bytes_needed > 0:
                     # CSOT: update timeout.
                     conn.settimeout(max(_csot.clamp_remaining(_KMS_CONNECT_TIMEOUT), 0))
-                    data = await _async_receive_data_socket(conn, kms_context.bytes_needed)
+                    data = await _async_receive_data_socket(conn, kms_context.bytes_needed, None)
                     if not data:
                         raise OSError("KMS connection closed")
                     kms_context.feed(data)
diff --git a/pymongo/synchronous/encryption.py b/pymongo/synchronous/encryption.py
@@ -154,7 +154,7 @@ def _async_kms_request(
                 while kms_context.bytes_needed > 0:
                     # CSOT: update timeout.
                     conn.settimeout(max(_csot.clamp_remaining(_KMS_CONNECT_TIMEOUT), 0))
-                    data = _receive_data_socket(conn, kms_context.bytes_needed)
+                    data = _receive_data_socket(conn, kms_context.bytes_needed, None)
                     if not data:
                         raise OSError("KMS connection closed")
                     kms_context.feed(data)
