Merge pull request #833 from adamtheturtle/a-few-more-exceptions

Add a few more exception tests

Author: adamtheturtle

src/vws/exceptions.py

@@ -20,6 +20,21 @@ def __init__(self, response: Response) -> None:
         self.response = response
 
 
+class Fail(Exception):
+    """
+    Exception raised when Vuforia returns a response with a result code
+    'Fail'.
+    """
+
+    def __init__(self, response: Response) -> None:
+        """
+        Args:
+            response: The response to a request to Vuforia.
+        """
+        super().__init__()
+        self.response = response
+
+
 class BadImage(Exception):
     """
     Exception raised when Vuforia returns a response with a result code

src/vws/vws.py

@@ -16,6 +16,7 @@
 from vws._authorization import authorization_header, rfc_1123_date
 from vws.exceptions import (
     BadImage,
+    Fail,
     ImageTooLarge,
     MetadataTooLarge,
     TargetNameExist,
@@ -101,6 +102,7 @@ def _raise_for_result_code(
 
     exception = {
         'BadImage': BadImage,
+        'Fail': Fail,
         'ImageTooLarge': ImageTooLarge,
         'MetadataTooLarge': MetadataTooLarge,
         'TargetNameExist': TargetNameExist,

tests/test_exceptions.py

@@ -6,12 +6,14 @@
 import random
 
 import pytest
+from mock_vws import MockVWS
 from PIL import Image
 from requests import codes
 
 from vws import VWS
 from vws.exceptions import (
     BadImage,
+    Fail,
     ImageTooLarge,
     MetadataTooLarge,
     TargetNameExist,
@@ -90,6 +92,26 @@ def test_request_quota_reached() -> None:
     """
 
 
+def test_fail(high_quality_image: io.BytesIO) -> None:
+    """
+    A ``Fail`` exception is raised when there are authentication issues.
+    """
+    with MockVWS() as mock:
+        client = VWS(
+            server_access_key='a',
+            server_secret_key=mock.server_secret_key,
+        )
+
+        with pytest.raises(Fail) as exc:
+            client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )
+
+        assert exc.value.response.status_code == codes.BAD_REQUEST
+
+
 def test_bad_image(client: VWS) -> None:
     """
     A ``BadImage`` exception is raised when a non-image is given.