feat: Update known errors and tests

Author: GeekMasher

src/ghastoolkit/errors.py

@@ -12,10 +12,12 @@ def __init__(
         message: Optional[str] = None,
         docs: Optional[str] = None,
         permissions: Optional[List[str]] = [],
+        status: Optional[int] = None,
     ) -> None:
         self.message = message
         self.docs = docs
         self.permissions = permissions
+        self.status = status
 
         super().__init__(message)
 
@@ -27,6 +29,9 @@ def __str__(self) -> str:
         else:
             msg = "An error occurred"
 
+        if status := self.status:
+            msg += f" (status code: {status})"
+
         if permissions := self.permissions:
             msg += "\n\nPermissions Required:"
             for perm in permissions:

src/ghastoolkit/octokit/octokit.py

@@ -21,11 +21,15 @@
 __OCTOKIT_PATH__ = os.path.dirname(os.path.realpath(__file__))
 
 __OCTOKIT_ERRORS__ = {
-    401: GHASToolkitAuthenticationError("Authentication / Permission Issue"),
-    403: GHASToolkitAuthenticationError("Authentication / Permission Issue"),
-    404: GHASToolkitError("Not Found"),
-    429: GHASToolkitError("Rate limit hit"),
-    500: GHASToolkitError("GitHub Server Error"),
+    401: GHASToolkitAuthenticationError(
+        "Authentication / Permission Issue", status=401
+    ),
+    403: GHASToolkitAuthenticationError(
+        "Authentication / Permission Issue", status=403
+    ),
+    404: GHASToolkitError("Not Found", status=404),
+    429: GHASToolkitError("Rate limit hit", status=429),
+    500: GHASToolkitError("GitHub Server Error", status=500),
 }
 
 
@@ -228,6 +232,9 @@ def get(
                 if display_errors:
                     logger.error(f"Error code from server :: {response.status_code}")
 
+                if error_handler:
+                    return error_handler(response.status_code, response_json)
+
                 known_error = __OCTOKIT_ERRORS__.get(response.status_code)
                 if known_error:
                     raise known_error

tests/test_restrequest.py

@@ -1,5 +1,7 @@
 """Test RestRequest class."""
+
 import unittest
+from ghastoolkit.errors import GHASToolkitError
 import responses
 
 from ghastoolkit.octokit.octokit import RestRequest
@@ -35,10 +37,10 @@ def test_errors(self):
             status=404,
         )
 
-        with self.assertRaises(Exception):
+        with self.assertRaises(GHASToolkitError):
             self.rest.get("/repos/{owner}/{repo}/secret-scanning/alerts")
 
-        with self.assertRaises(Exception):
+        with self.assertRaises(GHASToolkitError):
             self.rest.get("/repos/{owner}/{repo}/secret-scanning/alerts/1")
 
     @responses.activate