def test_file_validation(fixture):
"""A FileField needs to check that the right number of files were submitted, depending on the
setting of allow_multiple and/or required.
"""
field = FileField()
obj = fixture.model_object
field.bind('file_value', obj)
files = [EmptyStub(), EmptyStub()]
# Single file only
with expected(SingleFileConstraint):
field.set_user_input(files)
with expected(NoException):
field.set_user_input(files[:1])
# Single file that is required
field = FileField(required=True)
field.bind('file_value', obj)
with expected(NoException):
field.set_user_input(files[:1])
with expected(RequiredConstraint):
field.set_user_input([])
# Multiple files
field = FileField(allow_multiple=True)
field.bind('file_value', obj)
with expected(NoException):
field.set_user_input(files)
def file_validation_mime_type(self, fixture):
"""A FileField can also limit the mimetype of files allowed to be uploaded.
"""
field = FileField(allow_multiple=True, accept=['text/*'])
obj = fixture.model_object
field.bind('file_value', obj)
files = [UploadedFile('file1', b'stuff 1', 'text/html'), UploadedFile('file2', b'stuff 2', 'text/xml')]
with expected(NoException):
field.set_user_input(files)
files = [UploadedFile('file1', b'stuff 3', 'text/html'), UploadedFile('file2', b'stuff 4', 'application/java')]
with expected(MimeTypeConstraint):
field.set_user_input(files)
def file_validation_max_files(self, fixture):
"""A maximum can be placed upon the number of files that may be uploaded.
"""
field = FileField(allow_multiple=True, max_files=1)
obj = fixture.model_object
field.bind('file_value', obj)
files = [UploadedFile('file1', b'stuff 1', '')]
with expected(NoException):
field.set_user_input(files)
files = [UploadedFile('file1', b'stuff 2', ''), UploadedFile('file2', b'stuff 3', '')]
with expected(MaxFilesConstraint):
field.set_user_input(files)
def file_validation_size(self, fixture):
"""A FileField can also limit the size if files uploaded.
"""
field = FileField(allow_multiple=True, max_size_bytes=100)
obj = fixture.model_object
field.bind('file_value', obj)
files = [UploadedFile('file1', b'.'*100, ''), UploadedFile('file2', b'.'*50, '')]
with expected(NoException):
field.set_user_input(files)
files = [UploadedFile('file1', b'.'*100, ''), UploadedFile('file2', b'.'*200, '')]
with expected(FileSizeConstraint):
field.set_user_input(files)