diff options
Diffstat (limited to 'parts/django/tests/regressiontests/file_uploads/uploadhandler.py')
-rw-r--r-- | parts/django/tests/regressiontests/file_uploads/uploadhandler.py | 34 |
1 files changed, 0 insertions, 34 deletions
diff --git a/parts/django/tests/regressiontests/file_uploads/uploadhandler.py b/parts/django/tests/regressiontests/file_uploads/uploadhandler.py deleted file mode 100644 index 9f3960a..0000000 --- a/parts/django/tests/regressiontests/file_uploads/uploadhandler.py +++ /dev/null @@ -1,34 +0,0 @@ -""" -Upload handlers to test the upload API. -""" - -from django.core.files.uploadhandler import FileUploadHandler, StopUpload - -class QuotaUploadHandler(FileUploadHandler): - """ - This test upload handler terminates the connection if more than a quota - (5MB) is uploaded. - """ - - QUOTA = 5 * 2**20 # 5 MB - - def __init__(self, request=None): - super(QuotaUploadHandler, self).__init__(request) - self.total_upload = 0 - - def receive_data_chunk(self, raw_data, start): - self.total_upload += len(raw_data) - if self.total_upload >= self.QUOTA: - raise StopUpload(connection_reset=True) - return raw_data - - def file_complete(self, file_size): - return None - -class CustomUploadError(Exception): - pass - -class ErroringUploadHandler(FileUploadHandler): - """A handler that raises an exception.""" - def receive_data_chunk(self, raw_data, start): - raise CustomUploadError("Oops!") |