В моем приложении django у меня есть представление, которое выполняет загрузку файлов. Основной фрагмент выглядит следующим образом

...
if  (request.method == 'POST'):
    if request.FILES.has_key('file'):
        file = request.FILES['file']
        with open(settings.destfolder+'/%s' % file.name, 'wb+') as dest:
            for chunk in file.chunks():
                dest.write(chunk)

Я хотел бы провести модульное тестирование представления. Я планирую протестировать счастливый путь, а также путь сбоя ... т.е. случай, когда у request.FILESнего нет ключевого файла, случай, когда request.FILES['file']есть None..

Как мне настроить данные поста для счастливого пути? Может кто-нибудь подскажет?

Из документов Django Client.post:

Отправка файлов - особый случай. Чтобы отправить файл POST, вам нужно только указать имя поля файла в качестве ключа и дескриптор файла, который вы хотите загрузить, в качестве значения. Например:

c = Client()
with open('wishlist.doc') as fp:
  c.post('/customers/wishes/', {'name': 'fred', 'attachment': fp})

Раньше я делал то же самое, with open('some_file.txt') as fp:но потом мне понадобились изображения, видео и другие реальные файлы в репозитории, а также я тестировал часть основного компонента Django, который хорошо протестирован, поэтому в настоящее время я делаю вот что:

from django.core.files.uploadedfile import SimpleUploadedFile

def test_upload_video(self):
    video = SimpleUploadedFile("file.mp4", "file_content", content_type="video/mp4")
    self.client.post(reverse('app:some_view'), {'video': video})
    # some important assertions ...

В Python 3.5+ вам нужно использовать bytesобъект вместо str. Изменить "file_content"наb"file_content"

Он работает нормально, SimpleUploadedFileсоздает файл, InMemoryFileкоторый ведет себя как обычная загрузка, и вы можете выбрать имя, контент и тип контента.

Я рекомендую вам взглянуть на Django RequestFactory . Это лучший способ имитировать данные, указанные в запросе.

Сказал, что я нашел несколько недостатков в вашем коде.

• «Модульное» тестирование означает тестирование только одной «единицы» функциональности. Итак, если вы хотите протестировать это представление, вы должны тестировать представление и файловую систему, следовательно, не совсем модульный тест. Чтобы прояснить этот момент. Если вы запустите этот тест, и представление работает нормально, но у вас нет разрешений на сохранение этого файла, ваш тест не удастся из-за этого.
• Другой важный момент - тестовая скорость . Если вы делаете что-то вроде TDD, скорость выполнения ваших тестов действительно важна. Доступ к любому вводу-выводу - не лучшая идея .

Итак, я рекомендую вам провести рефакторинг вашего представления, чтобы использовать такую ​​функцию, как:

def upload_file_to_location(request, location=None): # Can use the default configured

И посмеемся над этим. Вы можете использовать Python Mock .

PS: Вы также можете использовать Django Test Client, но это будет означать, что вы добавляете еще одну вещь для тестирования, потому что этот клиент использует сеансы, промежуточное программное обеспечение и т. Д. Ничего похожего на модульное тестирование.

Я делаю что-то подобное для своего собственного приложения, связанного с событиями, но у вас должно быть более чем достаточно кода, чтобы продолжить свой собственный вариант использования.

import tempfile, csv, os

class UploadPaperTest(TestCase):

    def generate_file(self):
        try:
            myfile = open('test.csv', 'wb')
            wr = csv.writer(myfile)
            wr.writerow(('Paper ID','Paper Title', 'Authors'))
            wr.writerow(('1','Title1', 'Author1'))
            wr.writerow(('2','Title2', 'Author2'))
            wr.writerow(('3','Title3', 'Author3'))
        finally:
            myfile.close()

        return myfile

    def setUp(self):
        self.user = create_fuser()
        self.profile = ProfileFactory(user=self.user)
        self.event = EventFactory()
        self.client = Client()
        self.module = ModuleFactory()
        self.event_module = EventModule.objects.get_or_create(event=self.event,
                module=self.module)[0]
        add_to_admin(self.event, self.user)

    def test_paper_upload(self):
        response = self.client.login(username=self.user.email, password='foz')
        self.assertTrue(response)

        myfile = self.generate_file()
        file_path = myfile.name
        f = open(file_path, "r")

        url = reverse('registration_upload_papers', args=[self.event.slug])

        # post wrong data type
        post_data = {'uploaded_file': i}
        response = self.client.post(url, post_data)
        self.assertContains(response, 'File type is not supported.')

        post_data['uploaded_file'] = f
        response = self.client.post(url, post_data)

        import_file = SubmissionImportFile.objects.all()[0]
        self.assertEqual(SubmissionImportFile.objects.all().count(), 1)
        #self.assertEqual(import_file.uploaded_file.name, 'files/registration/{0}'.format(file_path))

        os.remove(myfile.name)
        file_path = import_file.uploaded_file.path
        os.remove(file_path)

Я сделал что-то подобное:

from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import TestCase
from django.core.urlresolvers import reverse
from django.core.files import File
from django.utils.six import BytesIO

from .forms import UploadImageForm

from PIL import Image
from io import StringIO


def create_image(storage, filename, size=(100, 100), image_mode='RGB', image_format='PNG'):
   """
   Generate a test image, returning the filename that it was saved as.

   If ``storage`` is ``None``, the BytesIO containing the image data
   will be passed instead.
   """
   data = BytesIO()
   Image.new(image_mode, size).save(data, image_format)
   data.seek(0)
   if not storage:
       return data
   image_file = ContentFile(data.read())
   return storage.save(filename, image_file)


class UploadImageTests(TestCase):
   def setUp(self):
       super(UploadImageTests, self).setUp()


   def test_valid_form(self):
       '''
       valid post data should redirect
       The expected behavior is to show the image
       '''
       url = reverse('image')
       avatar = create_image(None, 'avatar.png')
       avatar_file = SimpleUploadedFile('front.png', avatar.getvalue())
       data = {'image': avatar_file}
       response = self.client.post(url, data, follow=True)
       image_src = response.context.get('image_src')

       self.assertEquals(response.status_code, 200)
       self.assertTrue(image_src)
       self.assertTemplateUsed('content_upload/result_image.html')

Функция create_image создаст изображение, поэтому вам не нужно указывать статический путь изображения.

Примечание: вы можете обновить код в соответствии с вашим кодом. Этот код для Python 3.6.

В Django 1.7 есть проблема с TestCase, которая может быть решена с помощью open (filepath, 'rb'), но при использовании тестового клиента мы не контролируем его. Я думаю, что лучше всего убедиться, что file.read () всегда возвращает байты.

источник: https://code.djangoproject.com/ticket/23912 , автор KevinEtienne

Без параметра rb возникает ошибка TypeError:

TypeError: sequence item 4: expected bytes, bytearray, or an object with the buffer interface, str found

from rest_framework.test import force_authenticate
from rest_framework.test import APIRequestFactory

factory = APIRequestFactory()
user = User.objects.get(username='#####')
view = <your_view_name>.as_view()
with open('<file_name>.pdf', 'rb') as fp:
    request=factory.post('<url_path>',{'file_name':fp})
force_authenticate(request, user)
response = view(request)

Как упоминалось в официальной документации Django :

Отправка файлов - особый случай. Чтобы отправить файл POST, вам нужно только указать имя поля файла в качестве ключа и дескриптор файла, который вы хотите загрузить, в качестве значения. Например:

c = Client()
with open('wishlist.doc') as fp:
    c.post('/customers/wishes/', {'name': 'fred', 'attachment': fp})

Дополнительная информация: как проверить, передан ли файл в качестве аргумента какой-либо функции?

Во время тестирования иногда мы хотим убедиться, что файл передается в качестве аргумента какой-либо функции.

например

...
class AnyView(CreateView):
    ...
    def post(self, request, *args, **kwargs):
        attachment = request.FILES['attachment']
        # pass the file as an argument
        my_function(attachment)
        ...

В тестах используйте mock Python примерно так:

# Mock 'my_function' and then check the following:

response = do_a_post_request()

self.assertEqual(mock_my_function.call_count, 1)
self.assertEqual(
    mock_my_function.call_args,
    call(response.wsgi_request.FILES['attachment']),
)

from django.test import Client
from requests import Response

client = Client()
with open(template_path, 'rb') as f:
    file = SimpleUploadedFile('Name of the django file', f.read())
    response: Response = client.post(url, format='multipart', data={'file': file})

Надеюсь это поможет.

Я использую Python == 3.8.2, Django == 3.0.4, djangorestframework == 3.11.0

Я пробовал, self.client.postно получилResolver404 исключение.

Для меня сработало следующее:

import requests
upload_url='www.some.com/oaisjdoasjd' # your url to upload
with open('/home/xyz/video1.webm', 'rb') as video_file:
    # if it was a text file we would perhaps do
    # file = video_file.read()
    response_upload = requests.put(
        upload_url,
        data=video_file,
        headers={'content-type': 'video/webm'}
    )