在 Django REST Framework 中測試圖片上傳

Question

我正在嘗試向 Django 項目中的一個應用程序添加一些測試。 此應用程序包含Image model：

from django.db import models
from django.utils.translation import gettext_lazy as _

from cecommerce.image_mappings import ImageAgnosticMapping
from cecoresizer.fields import ResizableImageField


class Image(models.Model):
    file = ResizableImageField(
        _("Image"),
        max_length=300,
        image_config=ImageAgnosticMapping,
    )

    def __str__(self):
        return str(self.file)

然后使用以下序列化程序對其進行序列化：

from django.http import Http404

from rest_framework import serializers

from cecotec_apps.landings.models import ProductLanding
from cecotec_apps.partner.models import Home

from images.models import Image


class ImageSerializer(serializers.ModelSerializer):
    file = serializers.ImageField()

    def to_representation(self, instance):
        return {"file": str(instance)}

    class Meta:
        model = Image
        exclude = ("id",)

我正在嘗試通過此測試來測試Image實例的創建（在使用 Imsonia 請求 API 時有效）：

import tempfile

from PIL import Image as ImageFile

from django.test import tag

from model_bakery import baker

from rest_framework.test import APITestCase
from rest_framework.authtoken.models import Token
from rest_framework.reverse import reverse


from cecotec_apps.landings.models import ProductLanding
from cecotec_apps.partner.models import Home

from images.api.serializers import ImageSerializer
from images.models import Image

from tests.oscar.decorator.decorator_all_methods import global_test_decorator

from user.models import User


@tag("e2e", "image")
@global_test_decorator()
class ImageTestCase(APITestCase):
    API_VERSION = "v1"
    IMAGES_QUANTITY = 20
    HTTP_HOST = "localhost:8000"

    @classmethod
    def _populate(cls):
        cls.token = baker.make(Token)
        baker.make_recipe("images.image_recipe", _quantity=cls.IMAGES_QUANTITY)

    @classmethod
    def _generate_image_file(cls):
        with tempfile.NamedTemporaryFile(suffix="jpg") as tmp_file:
            image = ImageFile.new("RGB", size=(100, 100))
            image.save(tmp_file, "jpeg")
            tmp_file.seek(0)

            return tmp_file

    @classmethod
    def _depopulate(cls):
        Token.objects.all().delete()
        User.objects.all().delete()
        ProductLanding.objects.all().delete()
        Home.objects.all().delete()
        Image.objects.all().delete()

    def setUp(self, *args):
        self._depopulate()
        self._populate()

    def tearDown(self, *args):
        self._depopulate()

    @tag("create", "authenticated")
    def test_create_authenticated(self, *args):
        self.client.force_authenticate(user=self.token.user, token=self.token)

        data = self._prepare_create_data(**{"file": self._generate_image_file()})

        response = self._call_create(data)

        self._assert_create(response)
        self.assertRegex(response.data.get("file"), "^images\/.+$")

    def _call_create(self, data):
        return self.client.post(
            path=reverse(f"image-list"),
            data=data,
            # content_type=MULTIPART_CONTENT,
            format="multipart",
            HTTP_HOST=self.HTTP_HOST,
        )

    def _assert_create(self, response, expected_status_code=201):
        self.assertEqual(response.status_code, expected_status_code)

        if expected_status_code == 201:
            response_data = response.data

            self.assertEqual(Image.objects.count(), self.IMAGES_QUANTITY + 1)
            self._assert_response_body(response_data)
        else:
            self.assertEqual(Image.objects.count(), self.IMAGES_QUANTITY)

    def _assert_response_body(self, response_body):
        self.assertIn("file", response_body)

但是，運行它時返回的響應代碼是400 Bad Request而不是201 Created 。 響應體是這樣的：

{
    "file": [
        ErrorDetail(string="The submitted data was not a file. Check the encoding type on the form.",
        code="invalid")
    ]
}

我也試過使用已經創建的文件進行測試，但錯誤是一樣的。 我在網上看到很多帖子，但沒有一個能幫助我解決這個問題。

我會很感激一些幫助。 提前致謝。

Answer 1

原因

我認為問題來自_generate_image_file方法。

@classmethod
def _generate_image_file(cls):
    with tempfile.NamedTemporaryFile(suffix="jpg") as tmp_file:
        image = ImageFile.new("RGB", size=(100, 100))
        image.save(tmp_file, "jpeg")
        tmp_file.seek(0)

        return tmp_file

由於tempfile.NamedTemporaryFile是使用with語句創建的，因此當您返回它並在另一種方法中使用它時，該文件已經關閉並刪除。

解決方案

您應該通過使用yied裝飾方法將_generate_image_file更改為上下文管理器，並使用contextlib.contextmanager語句而不是 return。

from contextlib import contextmanager

@contextmanager
@classmethod
def _generate_image_file(cls):
    with tempfile.NamedTemporaryFile(suffix="jpg") as tmp_file:
        image = ImageFile.new("RGB", size=(100, 100))
        image.save(tmp_file, "jpeg")
        tmp_file.seek(0)

        yied tmp_file

並在test_create_authenticated方法上，將其更改為以下內容

@tag("create", "authenticated")
def test_create_authenticated(self, *args):
    ...
    with self._generate_image_file() as image_file:
        data = self._prepare_create_data(**{"file": image_file})
        response = self._call_create(data)
    ...

現在圖像文件將在with塊退出后被刪除。

Answer 2

我建議使用這個：

from django.core.files.uploadedfile import SimpleUploadedFile

with open("/some/path/to/image.jpg", "rb") as image:
        image = SimpleUploadedFile("image.jpg", image.read(), content_type="image/jpg")
        data = self._prepare_create_data(**{"file": image})
...

發生的情況是您的序列化程序正在驗證正在發送的文件無效。 確保在您的視圖中有parser_classes = (parsers.MultiPartParser,) ，在您的models.ImageField中也有 models.ImageField。