Question

我想为我的DRF应用编写测试，该应用使用 multipart 发布json和文件。

这是我到目前为止所尝试的内容，但collection_items（在创建方法中）为空。我是否需要修改我的视图以使其正常工作，或者我在下面的测试用例中做错了什么？

我的测试：

    image = Image.new('RGB', (100, 100))
    tmp_file = tempfile.NamedTemporaryFile(suffix='.jpg')
    image.save(tmp_file)

    files = {"collection_items": [{"image": tmp_file}]}
    payload = json.dumps({
        "title": "Test Collection",
    })

    self.api_factory.credentials(Authorization='Bearer ' + self.token)
    response = self.api_factory.post(url, data=payload, files=files, format='multipart')

这是型号：

class Collection(models.Model):

    title = models.CharField(max_length=60)
    collection_items = models.ManyToManyField('collection.Item')


class Item(models.Model):
    image = models.ImageField(upload_to="/",null=True, blank=True)

串行器：

class ItemCollectionDetailSerializer(serializers.ModelSerializer):
    class Meta:
        model = Item
        fields = ('id', 'image')
        read_only_fields = ('image',)


class CollectionListSerializer(serializers.ModelSerializer):

    url = serializers.HyperlinkedIdentityField(view_name='col_detail')
    collection_items = ItemCollectionDetailSerializer(many=True, required=True)

    class Meta:
        model = Collection
        fields = ('url', 'id', 'collection_items')

    def create(self, validated_data):

        item_data = validated_data.pop('collection_items')

        print(item_data)  # <----- **EMPTY HERE???**

        etc ....edited for brevity

所以print(item_data)为空[]，为什么？我该如何解决这个问题？

这是我的全部观点：下面，我是否需要在此处做点什么？

class CollectionListView(generics.ListCreateAPIView):

    queryset = Collection.objects.all()
    serializer_class = CollectionListSerializer

我正在使用Django Rest Framework 3.x，Django 1.8.x和Python 3.4.x。

更新

我在下面试过但仍然没有快乐！我的collection_items中create为空。这或者与它是一个嵌套对象或者我的视图中必须发生的事情有关。

    stream = BytesIO()
    image = Image.new('RGB', (100, 100))
    image.save(stream, format='jpeg')
    uploaded_file = SimpleUploadedFile("temp.jpeg", stream.getvalue())

    payload = {
        "title": "Test Collection",
        "collection_items": [{"image": uploaded_file}],
    }

    self.api_factory.credentials(Authorization='Bearer ' + self.test_access.token)
    response = self.api_factory.post(url, data=payload, format='multipart')

更新2

如果我将有效负载更改为使用json.dumps，它似乎现在看到该文件，但当然这不起作用！

payload = json.dumps({
            "title": "Test Collection",
            "collection_items": [{"image": uploaded_file}],
        })

错误

<SimpleUploadedFile: temp.jpeg (text/plain)> is not JSON serializable

PS

我知道文件正在上传，因为如果我在序列化程序中执行以下操作...

print(self.context.get("request").data['collection_items'])

我得到了

{'image': <SimpleUploadedFile: temp.jpeg (text/plain)>}

Answer 1

使用多部分解析器，您只需在post参数中传递文件处理程序（请参阅this）。在您的代码中，您提交了一个json编码的部分作为数据有效负载和files参数中的文件部分，我不认为它可以这样工作。

试试这段代码：

from PIL import Image
from io import BytesIO
from django.core.files.uploadedfile import SimpleUploadedFile

stream = BytesIO()
image = Image.new('RGB', (100, 100))
image.save(stream, format='jpeg')

uploaded_file = SimpleUploadedFile("file.jpg", stream.getvalue(), content_type="image/jpg")
payload = {
    "title": "Test collection",
    "collection_items": [{"image": uf}],
}
self.api_factory.credentials(Authorization='Bearer ' + self.token)
self.api_factory.post(url, data=payload, format='multipart')
...

我不完全确定嵌套序列化是否有效，但至少文件上传应该有效。

在DRF中支持JSON和文件分段上传的测试

1 个答案: