mirror of
https://github.com/django/django.git
synced 2025-10-23 21:59:11 +00:00
Convert much of the regression tests to use absolute imports. There's still work to be done though.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@16976 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
@@ -4,24 +4,25 @@ 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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user