Passing tests on the mock

Author: adamtheturtle

src/mock_vws/_query_validators/exceptions.py

@@ -621,6 +621,33 @@ def __init__(self) -> None:
         }
 
 
+class RequestEntityTooLarge(ValidatorException):
+    """
+    Exception raised when the given image file size is too large.
+    """
+
+    def __init__(self) -> None:
+        """
+        Attributes:
+            status_code: The status code to use in a response if this is
+                raised.
+            response_text: The response text to use in a response if this is
+                raised.
+        """
+        super().__init__()
+        self.status_code = HTTPStatus.REQUEST_ENTITY_TOO_LARGE
+        # TODO assert this!
+        self.response_text = ''
+        date = email.utils.formatdate(None, localtime=False, usegmt=True)
+        self.headers = {
+            'Connection': 'Close',
+            'Date': date,
+            'Server': 'nginx',
+            'Content-Type': 'text/html',
+            'Content-Length': str(len(self.response_text)),
+        }
+
+
 class MatchProcessing(ValidatorException):
     """
     Exception raised a target is matched which is processing or recently

src/mock_vws/_query_validators/image_validators.py

@@ -9,7 +9,7 @@
 import requests
 from PIL import Image
 
-from mock_vws._query_validators.exceptions import BadImage, ImageNotGiven
+from mock_vws._query_validators.exceptions import BadImage, ImageNotGiven, RequestEntityTooLarge
 
 
 def validate_image_field_given(
@@ -54,7 +54,7 @@ def validate_image_file_size(
         request_body: The body of the request.
 
     Raises:
-        requests.exceptions.ConnectionError: The image file size is too large.
+        RequestEntityTooLarge: The image file size is too large.
     """
     body_file = io.BytesIO(request_body)
 
@@ -74,7 +74,7 @@ def validate_image_file_size(
     # files.
     max_bytes = 2 * 1024 * 1024
     if len(image) > max_bytes:
-        raise requests.exceptions.ConnectionError
+        raise RequestEntityTooLarge
 
 
 def validate_image_dimensions(

tests/mock_vws/test_query.py

@@ -1307,7 +1307,7 @@ def test_jpeg(
         the maximum file size is "512 KiB for JPEG".
         However, this test shows that the maximum size for JPEG is 2 MiB.
 
-        Above this limit, a ``ConnectionError`` is raised.
+        Above this limit, a ``REQUEST_ENTITY_TOO_LARGE`` response is returned.
         We do not test exactly at this limit, but that may be beneficial in the
         future.
         """