SNOW-91104: Retry IncompleteRead exception.

Author: sfc-gh-stakeda

compat.py

@@ -61,6 +61,7 @@
     FORBIDDEN = httplib.FORBIDDEN
     UNAUTHORIZED = httplib.UNAUTHORIZED
     INTERNAL_SERVER_ERROR = httplib.INTERNAL_SERVER_ERROR
+    IncompleteRead = httplib.IncompleteRead
     OK = httplib.OK
     BadStatusLine = httplib.BadStatusLine
 
@@ -105,6 +106,7 @@
     FORBIDDEN = http.client.FORBIDDEN
     UNAUTHORIZED = http.client.UNAUTHORIZED
     INTERNAL_SERVER_ERROR = http.client.INTERNAL_SERVER_ERROR
+    IncompleteRead = http.client.IncompleteRead
     OK = http.client.OK
     BadStatusLine = http.client.BadStatusLine
 

network.py

@@ -32,7 +32,7 @@
     METHOD_NOT_ALLOWED, BAD_REQUEST, SERVICE_UNAVAILABLE, GATEWAY_TIMEOUT,
     FORBIDDEN, BAD_GATEWAY, REQUEST_TIMEOUT,
     UNAUTHORIZED, INTERNAL_SERVER_ERROR, OK, BadStatusLine)
-from .compat import (TO_UNICODE, urlencode, urlparse)
+from .compat import (TO_UNICODE, urlencode, urlparse, IncompleteRead)
 from .constants import (
     HTTP_HEADER_CONTENT_TYPE,
     HTTP_HEADER_ACCEPT,
@@ -855,6 +855,7 @@ def _request_exec(
                 ReadTimeout,
                 BadStatusLine,
                 ConnectionError,
+                IncompleteRead,
                 SSLError,
                 ProtocolError,  # from urllib3
                 ReadTimeoutError,  # from urllib3

test/test_unit_retry_network.py

@@ -25,7 +25,8 @@
     BAD_REQUEST,
     BAD_GATEWAY,
     UNAUTHORIZED,
-    BadStatusLine)
+    BadStatusLine,
+    IncompleteRead)
 from snowflake.connector.errors import (
     InterfaceError, DatabaseError, OtherHTTPRetryableError)
 from snowflake.connector.network import (
@@ -117,10 +118,15 @@ def test_request_exec():
             session=session, catch_okta_unauthorized_error=True,
             **default_parameters)
 
+    class IncompleteReadMock(IncompleteRead):
+        def __init__(self):
+            IncompleteRead.__init__(self, "")
+
     # handle retryable exception
     for exc in [
         ConnectTimeout,
         ReadTimeout,
+        IncompleteReadMock,
         SSLError,
         ProtocolError,
         ConnectionError,