def test_validate_file_ok(self):
self.assertEquals(
dict(filename=u'test.txt',
content_type=u'text/plain',
filedata='1' * 1024),
validate_file(
self.FileUpload(u'test.txt', u'text/plain', '1' * 1024), 1))
def test_validate_file_with_windows_path_in_filename(self):
self.assertEquals(
dict(filename=u'test.txt',
content_type=u'text/plain',
filedata='123456'),
validate_file(
self.FileUpload(r'c:\Documents and Settings\user\test.txt',
u'text/plain', '123456'), 1))
def test_validate_file_with_unix_path_in_filename(self):
self.assertEquals(
dict(filename=u'test.txt',
content_type=u'text/plain',
filedata='123456'),
validate_file(
self.FileUpload(u'/path/to/test.txt', u'text/plain', '123456'),
1))
def _validate_image(self, f):
file = validators.validate_file(f)
if file is None:
raise ValueError(_(u"Can't be empty"))
filename = file['filename']
data = file['filedata']
extension = os.path.splitext(filename)[1].lower()
prefix = uuid.uuid4().hex
filename = prefix + extension
# checks the image is not already in the gallery
if GalleryImageData.exist(data):
raise ValueError(_(u'Image already in gallery'))
return dict(filename=filename, data=data)
def _validate_image(self, f):
file = validators.validate_file(f)
if file is None:
raise ValueError(_(u"Can't be empty"))
filename = file['filename']
data = file['filedata']
extension = os.path.splitext(filename)[1].lower()
prefix = uuid.uuid4().hex
filename = prefix + extension
# checks the image is not already in the gallery
if GalleryImageData.exist(data):
raise ValueError(_(u'Image already in gallery'))
return dict(filename=filename, data=data)
def test_validate_file_with_windows_path_in_filename(self):
self.assertEquals(dict(filename=u'test.txt', content_type=u'text/plain', filedata='123456'),
validate_file(self.FileUpload(r'c:\Documents and Settings\user\test.txt', u'text/plain', '123456'), 1))
def test_validate_file_with_unix_path_in_filename(self):
self.assertEquals(dict(filename=u'test.txt', content_type=u'text/plain', filedata='123456'),
validate_file(self.FileUpload(u'/path/to/test.txt', u'text/plain', '123456'), 1))
def test_validate_file_too_big(self):
self.assertRaises(ValueError,
lambda: validate_file(self.FileUpload('test.txt', 'text/plain', '1' * 1025), 1))
def test_validate_file_ok(self):
self.assertEquals(dict(filename=u'test.txt', content_type=u'text/plain', filedata='1' * 1024),
validate_file(self.FileUpload(u'test.txt', u'text/plain', '1' * 1024), 1))
def validate_attachment(attachment):
return validate_file(attachment, attachments_max_size())
def test_validate_file_too_big(self):
self.assertRaises(
ValueError, lambda: validate_file(
self.FileUpload('test.txt', 'text/plain', '1' * 1025), 1))
def validate_attachment(attachment):
return validate_file(attachment, attachments_max_size())