summaryrefslogtreecommitdiff
path: root/tests/requests/test_data_upload_settings.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/requests/test_data_upload_settings.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/requests/test_data_upload_settings.py')
-rw-r--r--tests/requests/test_data_upload_settings.py187
1 files changed, 108 insertions, 79 deletions
diff --git a/tests/requests/test_data_upload_settings.py b/tests/requests/test_data_upload_settings.py
index 6f44dbd10d..363e9a966f 100644
--- a/tests/requests/test_data_upload_settings.py
+++ b/tests/requests/test_data_upload_settings.py
@@ -5,19 +5,23 @@ from django.core.handlers.wsgi import WSGIRequest
from django.test import SimpleTestCase
from django.test.client import FakePayload
-TOO_MANY_FIELDS_MSG = 'The number of GET/POST parameters exceeded settings.DATA_UPLOAD_MAX_NUMBER_FIELDS.'
-TOO_MUCH_DATA_MSG = 'Request body exceeded settings.DATA_UPLOAD_MAX_MEMORY_SIZE.'
+TOO_MANY_FIELDS_MSG = (
+ "The number of GET/POST parameters exceeded settings.DATA_UPLOAD_MAX_NUMBER_FIELDS."
+)
+TOO_MUCH_DATA_MSG = "Request body exceeded settings.DATA_UPLOAD_MAX_MEMORY_SIZE."
class DataUploadMaxMemorySizeFormPostTests(SimpleTestCase):
def setUp(self):
- payload = FakePayload('a=1&a=2&a=3\r\n')
- self.request = WSGIRequest({
- 'REQUEST_METHOD': 'POST',
- 'CONTENT_TYPE': 'application/x-www-form-urlencoded',
- 'CONTENT_LENGTH': len(payload),
- 'wsgi.input': payload,
- })
+ payload = FakePayload("a=1&a=2&a=3\r\n")
+ self.request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "POST",
+ "CONTENT_TYPE": "application/x-www-form-urlencoded",
+ "CONTENT_LENGTH": len(payload),
+ "wsgi.input": payload,
+ }
+ )
def test_size_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_MEMORY_SIZE=12):
@@ -35,19 +39,25 @@ class DataUploadMaxMemorySizeFormPostTests(SimpleTestCase):
class DataUploadMaxMemorySizeMultipartPostTests(SimpleTestCase):
def setUp(self):
- payload = FakePayload("\r\n".join([
- '--boundary',
- 'Content-Disposition: form-data; name="name"',
- '',
- 'value',
- '--boundary--'
- ]))
- self.request = WSGIRequest({
- 'REQUEST_METHOD': 'POST',
- 'CONTENT_TYPE': 'multipart/form-data; boundary=boundary',
- 'CONTENT_LENGTH': len(payload),
- 'wsgi.input': payload,
- })
+ payload = FakePayload(
+ "\r\n".join(
+ [
+ "--boundary",
+ 'Content-Disposition: form-data; name="name"',
+ "",
+ "value",
+ "--boundary--",
+ ]
+ )
+ )
+ self.request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "POST",
+ "CONTENT_TYPE": "multipart/form-data; boundary=boundary",
+ "CONTENT_LENGTH": len(payload),
+ "wsgi.input": payload,
+ }
+ )
def test_size_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_MEMORY_SIZE=10):
@@ -63,31 +73,39 @@ class DataUploadMaxMemorySizeMultipartPostTests(SimpleTestCase):
self.request._load_post_and_files()
def test_file_passes(self):
- payload = FakePayload("\r\n".join([
- '--boundary',
- 'Content-Disposition: form-data; name="file1"; filename="test.file"',
- '',
- 'value',
- '--boundary--'
- ]))
- request = WSGIRequest({
- 'REQUEST_METHOD': 'POST',
- 'CONTENT_TYPE': 'multipart/form-data; boundary=boundary',
- 'CONTENT_LENGTH': len(payload),
- 'wsgi.input': payload,
- })
+ payload = FakePayload(
+ "\r\n".join(
+ [
+ "--boundary",
+ 'Content-Disposition: form-data; name="file1"; filename="test.file"',
+ "",
+ "value",
+ "--boundary--",
+ ]
+ )
+ )
+ request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "POST",
+ "CONTENT_TYPE": "multipart/form-data; boundary=boundary",
+ "CONTENT_LENGTH": len(payload),
+ "wsgi.input": payload,
+ }
+ )
with self.settings(DATA_UPLOAD_MAX_MEMORY_SIZE=1):
request._load_post_and_files()
- self.assertIn('file1', request.FILES, "Upload file not present")
+ self.assertIn("file1", request.FILES, "Upload file not present")
class DataUploadMaxMemorySizeGetTests(SimpleTestCase):
def setUp(self):
- self.request = WSGIRequest({
- 'REQUEST_METHOD': 'GET',
- 'wsgi.input': BytesIO(b''),
- 'CONTENT_LENGTH': 3,
- })
+ self.request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "GET",
+ "wsgi.input": BytesIO(b""),
+ "CONTENT_LENGTH": 3,
+ }
+ )
def test_data_upload_max_memory_size_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_MEMORY_SIZE=2):
@@ -103,51 +121,60 @@ class DataUploadMaxMemorySizeGetTests(SimpleTestCase):
self.request.body
def test_empty_content_length(self):
- self.request.environ['CONTENT_LENGTH'] = ''
+ self.request.environ["CONTENT_LENGTH"] = ""
self.request.body
class DataUploadMaxNumberOfFieldsGet(SimpleTestCase):
-
def test_get_max_fields_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_NUMBER_FIELDS=1):
with self.assertRaisesMessage(TooManyFieldsSent, TOO_MANY_FIELDS_MSG):
- request = WSGIRequest({
- 'REQUEST_METHOD': 'GET',
- 'wsgi.input': BytesIO(b''),
- 'QUERY_STRING': 'a=1&a=2&a=3',
- })
- request.GET['a']
+ request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "GET",
+ "wsgi.input": BytesIO(b""),
+ "QUERY_STRING": "a=1&a=2&a=3",
+ }
+ )
+ request.GET["a"]
def test_get_max_fields_not_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_NUMBER_FIELDS=3):
- request = WSGIRequest({
- 'REQUEST_METHOD': 'GET',
- 'wsgi.input': BytesIO(b''),
- 'QUERY_STRING': 'a=1&a=2&a=3',
- })
- request.GET['a']
+ request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "GET",
+ "wsgi.input": BytesIO(b""),
+ "QUERY_STRING": "a=1&a=2&a=3",
+ }
+ )
+ request.GET["a"]
class DataUploadMaxNumberOfFieldsMultipartPost(SimpleTestCase):
def setUp(self):
- payload = FakePayload("\r\n".join([
- '--boundary',
- 'Content-Disposition: form-data; name="name1"',
- '',
- 'value1',
- '--boundary',
- 'Content-Disposition: form-data; name="name2"',
- '',
- 'value2',
- '--boundary--'
- ]))
- self.request = WSGIRequest({
- 'REQUEST_METHOD': 'POST',
- 'CONTENT_TYPE': 'multipart/form-data; boundary=boundary',
- 'CONTENT_LENGTH': len(payload),
- 'wsgi.input': payload,
- })
+ payload = FakePayload(
+ "\r\n".join(
+ [
+ "--boundary",
+ 'Content-Disposition: form-data; name="name1"',
+ "",
+ "value1",
+ "--boundary",
+ 'Content-Disposition: form-data; name="name2"',
+ "",
+ "value2",
+ "--boundary--",
+ ]
+ )
+ )
+ self.request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "POST",
+ "CONTENT_TYPE": "multipart/form-data; boundary=boundary",
+ "CONTENT_LENGTH": len(payload),
+ "wsgi.input": payload,
+ }
+ )
def test_number_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_NUMBER_FIELDS=1):
@@ -165,13 +192,15 @@ class DataUploadMaxNumberOfFieldsMultipartPost(SimpleTestCase):
class DataUploadMaxNumberOfFieldsFormPost(SimpleTestCase):
def setUp(self):
- payload = FakePayload("\r\n".join(['a=1&a=2&a=3', '']))
- self.request = WSGIRequest({
- 'REQUEST_METHOD': 'POST',
- 'CONTENT_TYPE': 'application/x-www-form-urlencoded',
- 'CONTENT_LENGTH': len(payload),
- 'wsgi.input': payload,
- })
+ payload = FakePayload("\r\n".join(["a=1&a=2&a=3", ""]))
+ self.request = WSGIRequest(
+ {
+ "REQUEST_METHOD": "POST",
+ "CONTENT_TYPE": "application/x-www-form-urlencoded",
+ "CONTENT_LENGTH": len(payload),
+ "wsgi.input": payload,
+ }
+ )
def test_number_exceeded(self):
with self.settings(DATA_UPLOAD_MAX_NUMBER_FIELDS=2):