Refs #30422 -- Added test for removing temporary files in MultiPartParser when StopUpload is raised.

1 files changed, 9 insertions, 1 deletions

diff --git a/tests/file_uploads/uploadhandler.py b/tests/file_uploads/uploadhandler.py
index 7c6199fd16..9ad335301f 100644
--- a/tests/file_uploads/uploadhandler.py
+++ b/tests/file_uploads/uploadhandler.py
@@ -2,7 +2,9 @@
 Upload handlers to test the upload API.
 """
 
-from django.core.files.uploadhandler import FileUploadHandler, StopUpload
+from django.core.files.uploadhandler import (
+    FileUploadHandler, StopUpload, TemporaryFileUploadHandler,
+)
 
 
 class QuotaUploadHandler(FileUploadHandler):
@@ -27,6 +29,12 @@ class QuotaUploadHandler(FileUploadHandler):
         return None
 
 
+class StopUploadTemporaryFileHandler(TemporaryFileUploadHandler):
+    """A handler that raises a StopUpload exception."""
+    def receive_data_chunk(self, raw_data, start):
+        raise StopUpload()
+
+
 class CustomUploadError(Exception):
     pass