Progress towards test

Author: adamtheturtle

src/vws/exceptions.py

@@ -63,3 +63,18 @@ def __init__(self, response: Response) -> None:
         """
         super().__init__()
         self.response = response
+
+
+class ImageTooLarge(Exception):
+    """
+    Exception raised when Vuforia returns a response with a result code
+    'ImageTooLarge'.
+    """
+
+    def __init__(self, response: Response) -> None:
+        """
+        Args:
+            response: The response to a request to Vuforia.
+        """
+        super().__init__()
+        self.response = response

src/vws/vws.py

@@ -16,6 +16,7 @@
 
 from vws._authorization import authorization_header, rfc_1123_date
 from vws.exceptions import (
+    ImageTooLarge,
     MetadataTooLarge,
     TargetNameExist,
     TargetStatusProcessing,
@@ -110,6 +111,7 @@ class _ResultCodes(Enum):
 
 
 _EXCEPTIONS = {
+    _ResultCodes.IMAGE_TOO_LARGE: ImageTooLarge,
     _ResultCodes.METADATA_TOO_LARGE: MetadataTooLarge,
     _ResultCodes.TARGET_NAME_EXIST: TargetNameExist,
     _ResultCodes.TARGET_STATUS_PROCESSING: TargetStatusProcessing,

tests/test_exceptions/test_image_too_large.py

@@ -7,8 +7,12 @@
 from pathlib import Path
 
 import pytest
+from requests import codes
 from PIL import Image
 
+from vws import VWS
+from vws.exceptions import ImageTooLarge
+
 
 def make_image_file(
     file_format: str,
@@ -43,7 +47,7 @@ def make_image_file(
     return image_buffer
 
 
-def test_foo():
+def test_foo(client: VWS):
     width = height = 890
 
     png_too_large = make_image_file(
@@ -52,6 +56,8 @@ def test_foo():
         width=width,
         height=height,
     )
-    #
-    # with pytest.raises(ImageTooLarge):
-    #     client.add_target(name='x', width=1, image=png_too_large)
+
+    with pytest.raises(ImageTooLarge) as exc:
+        client.add_target(name='x', width=1, image=png_too_large)
+
+    assert exc.value.response.status_code == codes.UNPROCESSABLE_ENTITY