Fixing liter error

Author: apetenchea

arangoasync/connection.py

@@ -12,8 +12,9 @@
 from arangoasync.auth import Auth, JwtToken
 from arangoasync.compression import CompressionManager, DefaultCompressionManager
 from arangoasync.exceptions import (
+    AuthHeaderError,
+    ClientConnectionAbortedError,
     ClientConnectionError,
-    ConnectionAbortedError,
     JWTRefreshError,
     ServerConnectionError,
 )
@@ -102,7 +103,7 @@ async def process_request(self, request: Request) -> Response:
                     self._host_resolver.change_host()
                     host_index = self._host_resolver.get_host_index()
 
-        raise ConnectionAbortedError(
+        raise ClientConnectionAbortedError(
             f"Can't connect to host(s) within limit ({self._host_resolver.max_tries})"
         )
 
@@ -226,6 +227,27 @@ def __init__(
         if self._token is None and self._auth is None:
             raise ValueError("Either token or auth must be provided.")
 
+    @property
+    def token(self) -> Optional[JwtToken]:
+        """Get the JWT token.
+
+        Returns:
+            JwtToken | None: JWT token.
+        """
+        return self._token
+
+    @token.setter
+    def token(self, token: Optional[JwtToken]) -> None:
+        """Set the JWT token.
+
+        Args:
+            token (JwtToken | None): JWT token.
+                Setting it to None will cause the token to be automatically
+                refreshed on the next request, if auth information is provided.
+        """
+        self._token = token
+        self._auth_header = f"bearer {self._token.token}" if self._token else None
+
     async def refresh_token(self) -> None:
         """Refresh the JWT token.
 
@@ -248,7 +270,7 @@ async def refresh_token(self) -> None:
 
         try:
             resp = await self.process_request(request)
-        except ConnectionAbortedError as e:
+        except ClientConnectionAbortedError as e:
             raise JWTRefreshError(str(e)) from e
         except ServerConnectionError as e:
             raise JWTRefreshError(str(e)) from e
@@ -267,27 +289,6 @@ async def refresh_token(self) -> None:
                 "Failed to refresh the JWT token: got an expired token"
             ) from e
 
-    @property
-    def token(self) -> Optional[JwtToken]:
-        """Get the JWT token.
-
-        Returns:
-            JwtToken | None: JWT token.
-        """
-        return self._token
-
-    @token.setter
-    def token(self, token: Optional[JwtToken]) -> None:
-        """Set the JWT token.
-
-        Args:
-            token (JwtToken | None): JWT token.
-                Setting it to None will cause the token to be automatically
-                refreshed on the next request, if auth information is provided.
-        """
-        self._token = token
-        self._auth_header = f"bearer {self._token.token}" if self._token else None
-
     async def send_request(self, request: Request) -> Response:
         """Send an HTTP request to the ArangoDB server.
 
@@ -303,6 +304,10 @@ async def send_request(self, request: Request) -> Response:
         """
         if self._auth_header is None:
             await self.refresh_token()
+
+        if self._auth_header is None:
+            raise AuthHeaderError("Failed to generate authorization header.")
+
         request.headers["authorization"] = self._auth_header
 
         try:

arangoasync/exceptions.py

@@ -68,14 +68,18 @@ def __init__(
         self.http_headers = resp.headers
 
 
-class ConnectionAbortedError(ArangoClientError):
+class ClientConnectionAbortedError(ArangoClientError):
     """The connection was aborted."""
 
 
 class ClientConnectionError(ArangoClientError):
     """The request was unable to reach the server."""
 
 
+class AuthHeaderError(ArangoClientError):
+    """The authentication header could not be determined."""
+
+
 class JWTExpiredError(ArangoClientError):
     """JWT token has expired."""
 

tests/test_connection.py

@@ -6,8 +6,8 @@
 from arangoasync.compression import AcceptEncoding, DefaultCompressionManager
 from arangoasync.connection import BasicConnection, JwtConnection
 from arangoasync.exceptions import (
+    ClientConnectionAbortedError,
     ClientConnectionError,
-    ConnectionAbortedError,
     ServerConnectionError,
 )
 from arangoasync.http import AioHTTPClient
@@ -128,7 +128,7 @@ async def mock_send_request(*args, **kwargs):
         auth=Auth(username=root, password=password),
     )
 
-    with pytest.raises(ConnectionAbortedError):
+    with pytest.raises(ClientConnectionAbortedError):
         await connection.process_request(request)