如何在 Django 测试中发送带有图像字段查询参数的 POST 请求

发布于05月07日

我有test.py个


from rest_framework.test import APITestCase
from django.core.files.uploadedfile import SimpleUploadedFile
from django.urls import reverse

class PostImageTest(APITestCase):
    def setUp(self) -> None:
        self.product = Product.objects.create(...)
        self.sampleImage = SimpleUploadedFile("test_image.jpg", b"binary data", content_type="image/jpeg")

    def test_with_valid_data(self) -> None:
        data = { "image":self.sampleImage, "is_featured":true }
        response = client.post(reverse('images'), data, format='multipart')

我想将像100这样的查询参数传递给client.post()方法.另外，我不知道如何对图像进行编码并发送POST请求.

当我试着这样做的时候

response = client.post(reverse('images'), {'product' : self.product.id}, data=data, format='multipart')

它给了我这个错误

TypeError: Client.post() got multiple values for argument 'data'个

推荐答案

就这样解决了

from io import BytesIO
from PIL import Image as img

def get_url(self, paramKey=None, paramValue=None):
    url = reverse('images') + f"?{paramKey}={paramValue}"
    return url

def create_valid_image(self):
    image_file = BytesIO()
    image = img.new('RGBA', size=(50, 50), color=(155, 0, 0))
    image.save(image_file, 'png')
    image_file.name = 'test_image.png'
    image_file.seek(0)
    return image_file

def test_with_valid_data(self) -> None:
    self.myImage = self.create_valid_image()
    expectedStatusCode = status.HTTP_201_CREATED

    url = self.get_url('product', self.product.id)
    data = { 'image' : self.myImage, 'is_featured' : True }
    response = client.post(url, data)
    receivedStatusCode = response.status_code

    self.assertEqual(receivedStatusCode, expectedStatusCode)

因此，看起来如果我们在client.post()中传递Query_Params，它会将其视为请求数据.因此，我们必须将查询参数连接到原始的url.

可以进一步通过这document