[PECO-1687] Fixed the issue of infinite blocking in case of invalid credentials

src/databricks/sql/auth/retry.py

@@ -19,6 +19,7 @@
     MaxRetryDurationError,
     NonRecoverableNetworkError,
     OperationalError,
+    AuthenticationFailureError,
     SessionAlreadyClosedError,
     UnsafeToRetryError,
 )
@@ -326,7 +327,7 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
                This limit prevents automatically retrying non-idempotent commands that could
                be destructive.
             5. The request received a 403 response, because this can never succeed.
-
+            6. The request received a 401 response, because the User credentials are invalid
 
         Q: What about OSErrors and Redirects?
         A: urllib3 automatically retries in both scenarios
@@ -339,6 +340,11 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
         if status_code == 200:
             return False, "200 codes are not retried"
 
+        # Don't retry as there is an authentication error
+        if status_code == 401:
+            raise AuthenticationFailureError(
+                "Received 401 - UNAUTHORIZED. Authentication is required and has failed or has not yet been provided."
+            )
         if status_code == 403:
             raise NonRecoverableNetworkError(
                 "Received 403 - FORBIDDEN. Confirm your authentication credentials."

src/databricks/sql/exc.py

@@ -105,6 +105,10 @@ class NonRecoverableNetworkError(RequestError):
     """Thrown if an HTTP code 501 is received"""
 
 
+class AuthenticationFailureError(RequestError):
+    """Thrown if an HTTP code 401 is received"""
+
+
 class UnsafeToRetryError(RequestError):
     """Thrown if ExecuteStatement request receives a code other than 200, 429, or 503"""