Django - 如何为DRF ImageField编写测试

Question

我有以下序列化器：

from rest_framework.serializers import Serializer, ImageField

class MySerializer(Serializer):
    avatar = ImageField()

我怎么能为它编写单元测试？ 我使用了Django TestCase，但它引发了一个错误。

from django.test import TestCase

class MySerializerTest(TestCase):

    def setUp(self):
        self.data = {}
        ...

    def test_image(self):
        import tempfile
        self.data['avatar'] = tempfile.NamedTemporaryFile(suffix=".jpg").file
        r_data = json.dumps(self.data)
        j_data = json.loads(r_data)
        serializer = MySerializer(data=j_data)
        if not serializer.is_valid():
            import pprint
            pprint.pprint(serializer.errors)
        self.assertEqual(serializer.is_valid(), True)

但它引发了以下错误：

TypeError：'BufferedRandom'类型的对象不是JSON可序列化的

我的错是什么？ 如何为图像区域编写测试？

Answer 1

通常当您上传文件时，您将使用multipart请求格式，并且视图会将图像转换为InMemoryUploadedFile对象，并将其传递到您的序列化程序

所以为了修复你的测试，我建议你尝试：

from PIL import Image
from tempfile import NamedTemporaryFile
from django.conf.files.uploadedfile import InMemoryUploadedFile

...

def test_image(self):
    image = Image.new("RGB", (100, 100))
    with NamedTemporaryFile(suffix=".png", mode="w+b") as tmp_file:
        image.save(tmp_file, format="png")
        tmp_file.seek(0)
        byio = BytesIO(temp_file.read())
        inm_file = InMemoryUploadedFile(
            file=byio,
            field_name="avatar",
            name="testImage.png",
            content_type="image/png",
            size=byio.getbuffer().nbytes,
            charset=None,
        )

        self.data['avatar'] = inm_file
        serializer = MySerializer(data=self.data)
        if not serializer.is_valid():
            import pprint
            pprint.pprint(serializer.errors)
        self.assertEqual(serializer.is_valid(), True)

这是做什么的：

1. 使用PIL.Image在内存中创建图像
2. 创建NamedTemporaryFile以存储Image数据
3. 获取NamedTemporaryFile并读入InMemoryUploadedFile
4. 将此InMemoryUploadedFile传递给序列化程序

Answer 2

我建议使用django中的SimpleUploadedFile类，并使用Pillow包创建和映像。 请参阅下面的示例。

from PIL import Image

from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import TestCase
from django.utils.six import BytesIO


class MySerializerTest(TestCase):
    ...

    def test_image(self):
        image = BytesIO()
        Image.new('RGB', (100, 100)).save(image, 'JPEG')
        image.seek(0)

        self.data['avatar'] = SimpleUploadedFile('image.jpg', image.getvalue())
        serializer = MySerializer(data=self.data)
        self.assertEqual(serializer.is_valid(), True)