fix: add TimeoutError handling in get_connection() (#1485)

* fix: add TimeoutError handling in get_connection()

In get_connection() we can implicitly call read on
a connection. This can timeout of the underlying TCP
session is gone. With this change we remove it
from the connection pool and get a new connection.

* fix: add TimeoutError handling in get_connection() sync/async

In get_connection() we can implicitly call read on a connection. This
can timeout of the underlying TCP session is gone. With this change we
remove it from the connection pool and get a new connection.

* fix: update version number to match test expectations

* fix: revert version number to 5.2.1, manually uninstall entraid

Author: donbowman

redis/asyncio/connection.py

@@ -1157,7 +1157,7 @@ async def ensure_connection(self, connection: AbstractConnection):
         try:
             if await connection.can_read_destructive():
                 raise ConnectionError("Connection has data") from None
-        except (ConnectionError, OSError):
+        except (ConnectionError, TimeoutError, OSError):
             await connection.disconnect()
             await connection.connect()
             if await connection.can_read_destructive():

redis/connection.py

@@ -1494,7 +1494,7 @@ def get_connection(self, command_name=None, *keys, **options) -> "Connection":
             try:
                 if connection.can_read() and self.cache is None:
                     raise ConnectionError("Connection has data")
-            except (ConnectionError, OSError):
+            except (ConnectionError, TimeoutError, OSError):
                 connection.disconnect()
                 connection.connect()
                 if connection.can_read():
@@ -1741,7 +1741,7 @@ def get_connection(self, command_name=None, *keys, **options):
             try:
                 if connection.can_read():
                     raise ConnectionError("Connection has data")
-            except (ConnectionError, OSError):
+            except (ConnectionError, TimeoutError, OSError):
                 connection.disconnect()
                 connection.connect()
                 if connection.can_read():