4

我正在使用 Django 1.7 和 Python 3.4 编写单元测试。当file_data元素被注释掉时,下面的表单验证良好。包含file_data它不会验证并且测试失败。

from django.core.files.uploadedfile import SimpleUploadedFile
...

data = {
    'phone_number': '123456',
    'job': Job.objects.latest('id').pk,
}
file_data = {
    'portrait': SimpleUploadedFile(
        content=b'',
        content_type='image/jpeg',
        name='test.jpg',
    )
}
form = PersonForm(data, file_data)
self.assertTrue(form.is_valid())

类似的代码适用FileField于在我网站的其他地方上传测试。在 shell 中运行代码我得到以下信息form.errors'Upload a valid image. The file you uploaded was either not an image or a corrupted image.'因此我认为问题出在contentorcontent_type字段中。我已尝试在此答案中使用图像作为字符串,但无济于事。[编辑:图像作为字符串的方法实际上是答案,但我一定实施得很糟糕。接受的答案被验证为有效。]我在SimpleUloadedFile 源代码中找不到任何线索。

该表格在现实中工作正常,但我想确保它包含在未来维护的工作测试中。理想情况下,我希望避免必须存在实际test.jpg文件,因为图像在我的.gitignore文件中,我不想开始破解目前非常顺利的自动化部署。

我应该提供什么输入才能SimpleUploadedFile正确验证?

4

1 回答 1

5

Image您的测试套件和您的工作代码之间运行 lib 有什么区别吗?lib能否Image正确处理GIF等文件?您可能还需要检查PIL/pillow安装。

SimpleUploadedFile(name='foo.gif', 
                   content=b'GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00')

为我工作。

表单在内部依赖于forms.ImageField测试其to_python()中的图像,方式如下:

import io
from django.utils.image import Image
Image.open(io.BytesIO(image_content)).verify()

该行的任何异常verify()都会导致错误'Upload a valid image...'。您可以通过将其分配给来检查图像字节image_content

image_content = b'GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00'

对我也有用。

此外,您可以在线生成图像字节:

import io    
from django.utils.image import Image
fp = io.BytesIO()
Image.new('P', (1,1)).save(fp, 'png')
fp.seek(0)
portrait = SimpleUploadedFile(name=''foo, content=fp.read())
于 2014-10-01T14:02:01.460 回答