Как я могу протестировать загрузку двоичного файла с помощью тестового клиента django-rest-framework?

Ответ 1

При тестировании загрузки файлов вы должны передавать в запрос объект потока, а не данные.

Это было указано в комментариях @arocks

Вместо этого передайте {'image': file}

Но это не полностью объяснило, почему это было необходимо (и также не соответствовало вопросу). Для этого конкретного вопроса, вы должны делать

from PIL import Image

class TestFileUpload(APITestCase):

    def test_file_is_accepted(self):
        self.client.force_authenticate(self.user)

        image = Image.new('RGB', (100, 100))

        tmp_file = tempfile.NamedTemporaryFile(suffix='.jpg')
        image.save(tmp_file)
        tmp_file.seek(0)

        response = self.client.post('my_url', {'image': tmp_file}, format='multipart')

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

Это будет соответствовать стандартному запросу Django, где файл передается как потоковый объект, и Django REST Framework обрабатывает его. Когда вы просто передаете данные файла, Django и Django REST Framework интерпретируют его как строку, что вызывает проблемы, поскольку он ожидает поток.

А для тех, кто приезжает сюда и ищет другую распространенную ошибку, почему загрузка файлов просто не будет работать, а обычные данные формы: обязательно установите format="multipart" при создании запроса.

Это также создает аналогичную проблему, о которой @RobinElvin указал в комментариях.

Это потому, что мне не хватало format = 'multipart'

Ответ 2

Пользователи Python 3: убедитесь, что вы open файл в mode='rb' (read, binary). В противном случае, когда Django вызывает read в файле, кодек utf-8 немедленно начнет задыхаться. Файл должен быть декодирован как двоичный, а не utf-8, ascii или любая другая кодировка.

# This won't work in Python 3
with open(tmp_file.name) as fp:
        response = self.client.post('my_url', 
                                   {'image': fp}, 
                                   format='multipart')

# Set the mode to binary and read so it can be decoded as binary
with open(tmp_file.name, 'rb') as fp:
        response = self.client.post('my_url', 
                                   {'image': fp}, 
                                   format='multipart')

Ответ 3

Не так просто понять, как это сделать, если вы хотите использовать метод PATCH, но я нашел решение в этом вопросе.

from django.test.client import BOUNDARY, MULTIPART_CONTENT, encode_multipart

with open(tmp_file.name, 'rb') as fp:
    response = self.client.patch(
        'my_url', 
        encode_multipart(BOUNDARY, {'image': fp}), 
        content_type=MULTIPART_CONTENT
    )

Ответ 4

Для тех, кто в Windows, ответ немного отличается. Я должен был сделать следующее:

resp = None
with tempfile.NamedTemporaryFile(suffix='.jpg', delete=False) as tmp_file:
    image = Image.new('RGB', (100, 100), "#ddd")
    image.save(tmp_file, format="JPEG")
    tmp_file.close()

# create status update
with open(tmp_file.name, 'rb') as photo:
    resp = self.client.post('/api/articles/', {'title': 'title',
                                               'content': 'content',
                                               'photo': photo,
                                               }, format='multipart')
os.remove(tmp_file.name)

Разница, указанная в этом ответе (fooobar.com/info/165774/...), не может быть использована после закрытия в Windows. Под Linux ответ @Meistro должен работать.