Django REST UnitTest没有提交文件

时间:2017-09-26 12:34:28

标签: python django unit-testing django-rest-framework

我成功实施了小案例。然后我开始使用更大的结构。我得到了错误。
错误:
No file was submitted.

import tempfile
from unittest import skip

from django.conf import settings
from django.contrib.auth.models import User
from django.core.files import File
from django.core.files.uploadedfile import SimpleUploadedFile
from model_mommy import mommy
from rest_framework import status
from rest_framework.reverse import reverse
from rest_framework.test import APITestCase, APIClient


class CustomerFromExcelViewsetTest(APITestCase):
    def setUp(self):
        self.client = APIClient()
        self.soken_staff = mommy.make(User, username='spearhead')
        self.user = mommy.make(User, username='Justin')
        settings.MEDIA_ROOT = tempfile.mkdtemp()

    def test_upload_file(self):
        """Expect created_user, and updated_user correct set"""
        file = File(open('./soken_web/apps/uploaded_files/complete-customer.xlsx', 'rb'))
        uploaded_file = SimpleUploadedFile('new_excel.xlsx', file.read(), content_type='multipart/form-data')
        data = {
            file: uploaded_file,
        }
        self.client.force_authenticate(user=self.user)
        response = self.client.post(reverse('api:customer_from_excel-list'), data, format='multipart')
        response.render()

        self.assertEqual(status.HTTP_201_CREATED, response.status_code)

这里是modelsserializersviewsets

models.py
https://gist.github.com/elcolie/52daf2bd144af82b348f7353656be434

serializers.py
https://gist.github.com/elcolie/7f097642c4a752e76044c6938c49e097

viewsets.py
https://gist.github.com/elcolie/34fa66632209f14624899d997919d3fb

过了一天,我无法弄清楚那个虫子在哪里。

参考文献:
DRF APITestCase not use `multipart` with other param

1 个答案:

答案 0 :(得分:1)

看起来你在data字典中错过了引号。它应该是:

data = {
    'file': uploaded_file,
}