[Answered ]-DRF test multiple image upload

by

1👍

The problem here is that you’re sending the same Image object to the database to be saved twice. Once the first Image object has been decoded to a file to be saved in the database, the Image object will become unreadable, and since its the same object as the second, an error will be thrown.

So if you send the same Image object as two different items, only the first one will be readable. To avoid this you’ll have to send two different image objects.

Your ImageFactory can be refactored to:

def get_image(count=1):
    images = []
    for i in range(count):
        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)
        images.append(file)
    return images[0] if count == 1 else images

and your Test:

def test_upload_multiple_images(self):
    self.images = get_image(2)
    payload = {
        "image": self.images
    }
    response = self.client.post(
        reverse("gallery-list", args=[self.item.pk]),
        data=payload,
        format="multipart"
    )
👤The Digerati

Leave a comment