Test the entity too large text

adamtheturtle · adamtheturtle · commit d6aa76aa9e78 · 2021-03-23T00:44:38.000Z
diff --git a/src/mock_vws/_query_validators/exceptions.py b/src/mock_vws/_query_validators/exceptions.py
@@ -636,8 +636,17 @@ def __init__(self) -> None:
         """
         super().__init__()
         self.status_code = HTTPStatus.REQUEST_ENTITY_TOO_LARGE
-        # TODO assert this!
-        self.response_text = ''
+        self.response_text = textwrap.dedent(
+            """\
+            <html>\r
+            <head><title>413 Request Entity Too Large</title></head>\r
+            <body bgcolor="white">\r
+            <center><h1>413 Request Entity Too Large</h1></center>\r
+            <hr><center>nginx</center>\r
+            </body>\r
+            </html>\r
+            """,
+        )
         date = email.utils.formatdate(None, localtime=False, usegmt=True)
         self.headers = {
             'Connection': 'Close',
diff --git a/src/mock_vws/_query_validators/image_validators.py b/src/mock_vws/_query_validators/image_validators.py
@@ -6,10 +6,13 @@
 import io
 from typing import Dict
 
-import requests
 from PIL import Image
 
-from mock_vws._query_validators.exceptions import BadImage, ImageNotGiven, RequestEntityTooLarge
+from mock_vws._query_validators.exceptions import (
+    BadImage,
+    ImageNotGiven,
+    RequestEntityTooLarge,
+)
 
 
 def validate_image_field_given(
diff --git a/tests/mock_vws/test_query.py b/tests/mock_vws/test_query.py
@@ -62,6 +62,17 @@
     """,  # noqa: E501
 )
 
+_NGINX_REQUEST_ENTITY_TOO_LARGE_ERROR = textwrap.dedent(
+    """\
+    <html>\r
+    <head><title>413 Request Entity Too Large</title></head>\r
+    <body bgcolor="white">\r
+    <center><h1>413 Request Entity Too Large</h1></center>\r
+    <hr><center>nginx</center>\r
+    </body>\r
+    </html>\r
+    """,
+)
 
 _JETTY_ERROR_DELETION_NOT_COMPLETE_START_PATH = (
     Path(__file__).parent / 'jetty_error_deletion_not_complete.html'
@@ -1296,6 +1307,7 @@ def test_png(
             www_authenticate=None,
             connection='Close',
         )
+        assert response.text == _NGINX_REQUEST_ENTITY_TOO_LARGE_ERROR
 
     def test_jpeg(
         self,
@@ -1370,6 +1382,8 @@ def test_jpeg(
             connection='Close',
         )
 
+        assert response.text == _NGINX_REQUEST_ENTITY_TOO_LARGE_ERROR
+
 
 @pytest.mark.usefixtures('verify_mock_vuforia')
 class TestMaximumImageDimensions:
