在DRF中測試ImageField

Question

我在models.py中使用Django Imagekit：

from imagekit.models import ProcessedImageField    
class AltroUser(models.Model):
        first_name = models.CharField(_('first name'), max_length=30)
        image = ProcessedImageField(upload_to='media/path',
                                    default='user_default.jpg',
                                    processors=[ResizeToFill(640, 640)],
                                    format='JPEG',
                                    options={'quality': 60})

serializers.py：

class UserRegistrationSerializer(Serializer):
    first_name = serializers.CharField()
    image = serializers.ImageField()

我正在嘗試測試圖像字段。 我嘗試了以下方法：

def get_test_image():
    try:
        image = DjangoFile(open(os.path.join(django_settings.MEDIA_ROOT, 'user_default.jpg'),
                                mode='rb'))
        return image
    except (OSError, IOError) as e:
        return None


def get_test_image1():
    file = io.BytesIO()
    image = Image.new('RGBA', size=(100, 100), color=(155, 0, 0))
    image.save(file, 'png')
    file.name = 'test.png'
    file.seek(0)
    return SimpleUploadedFile('abc.jpg', file.read())


def get_test_image2():
    path = os.path.join(django_settings.MEDIA_ROOT, 'user_default.jpg')
    file = File(open(path, 'r+b'))
    return SimpleUploadedFile('abc.jpg', file.read())

我嘗試調用上述三種方法來設置圖像鍵的值，但是它們都不起作用。 對於get_test_image() ，我得到一個響應“提交的文件為空。”

對於get_test_image1()和get_test_image2() ，我得到一個響應"The submitted file is empty." 但之前在數據映像字段'_io.BytesIO' object has no attribute 'encoding' 。

我不明白我在想什么。 請幫忙。

Answer 1

這是我用來生成測試圖像的示例：

from StringIO import StringIO
from PIL import Image
from django.core.files import File

def get_image_file(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)):
    file_obj = StringIO()
    image = Image.new("RGBA", size=size, color=color)
    image.save(file_obj, ext)
    file_obj.seek(0)
    return File(file_obj, name=name)