Add headers to Auth0Error

Author: adamjmcgrath

auth0/exceptions.py

@@ -10,11 +10,13 @@ def __init__(
         error_code: str,
         message: str,
         content: Any | None = None,
+        headers: Any | None = None,
     ) -> None:
         self.status_code = status_code
         self.error_code = error_code
         self.message = message
         self.content = content
+        self.headers = headers
 
     def __str__(self) -> str:
         return f"{self.status_code}: {self.message}"

auth0/rest.py

@@ -296,12 +296,14 @@ def content(self) -> Any:
                     error_code=self._error_code(),
                     message=self._error_message(),
                     content=self._content,
+                    headers=self._headers
                 )
 
             raise Auth0Error(
                 status_code=self._status_code,
                 error_code=self._error_code(),
                 message=self._error_message(),
+                headers=self._headers
             )
         else:
             return self._content

auth0/test/authentication/test_get_token.py

@@ -1,4 +1,5 @@
 import unittest
+import requests
 from fnmatch import fnmatch
 from unittest import mock
 from unittest.mock import ANY
@@ -335,7 +336,26 @@ def test_backchannel_login(self, mock_post):
                 "grant_type": "urn:openid:params:grant-type:ciba",
             },
         )
-        
+
+    @mock.patch("auth0.rest.RestClient.post")
+    def test_backchannel_login_headers_on_failure(self, mock_post):
+        response = requests.Response()
+        response.status_code = 400
+        response.headers = {"Retry-After": 100}
+        response._content = b'{"error":"slow_down"}'
+        mock_post.side_effect = requests.exceptions.HTTPError(response=response)
+
+        g = GetToken("my.domain.com", "cid", client_secret="csec")
+
+        try:
+            g.backchannel_login(
+                auth_req_id="reqid",
+                grant_type="urn:openid:params:grant-type:ciba",
+            )
+        except requests.exceptions.HTTPError as e:
+            self.assertEqual(e.response.headers["Retry-After"], 100)
+            self.assertEqual(e.response.status_code, 400)
+
     @mock.patch("auth0.rest.RestClient.post")
     def test_connection_login(self, mock_post):
         g = GetToken("my.domain.com", "cid", client_secret="csec")
@@ -367,7 +387,7 @@ def test_connection_login(self, mock_post):
         )
 
     @mock.patch("auth0.rest.RestClient.post")
-    def test_connection_loginwith_login_hint(self, mock_post):
+    def test_connection_login_with_login_hint(self, mock_post):
         g = GetToken("my.domain.com", "cid", client_secret="csec")
 
         g.access_token_for_connection(