Django test FileField с использованием тестовых приспособлений

Django предоставляет отличный способ писать тесты для FileFields, не копаясь в реальной файловой системе - используйте SimpleUploadedFile.

from django.core.files.uploadedfile import SimpleUploadedFile

my_model.file_field = SimpleUploadedFile('best_file_eva.txt', b'these are the contents of the txt file')

Это одна из волшебных функций django - которые не появляются в документации :). Однако здесь имеется ссылка на .

Вы можете переопределить настройку MEDIA_ROOT для своих тестов с помощью @override_settings() декоратора как задокументировано:

from django.test import override_settings


@override_settings(MEDIA_ROOT='/tmp/django_test')
def test_post_solution_file(self):
  # your code here

Раньше я писал модульные тесты для всего приложения галереи, и что мне понравилось, так это использование модулей python tempfile и shutil для создания копий тестовых файлов во временных каталогах и последующего их удаления.

Следующий пример не работает / не завершен, но должен указать вам правильный путь:

import os, shutil, tempfile

PATH_TEMP = tempfile.mkdtemp(dir=os.path.join(MY_PATH, 'temp'))

def make_objects():
    filenames = os.listdir(TEST_FILES_DIR)

    if not os.access(PATH_TEMP, os.F_OK):
        os.makedirs(PATH_TEMP)

    for filename in filenames:
        name, extension = os.path.splitext(filename)
        new = os.path.join(PATH_TEMP, filename)
        shutil.copyfile(os.path.join(TEST_FILES_DIR, filename), new)

        #Do something with the files/FileField here

def remove_objects():
    shutil.rmtree(PATH_TEMP)

Я запускаю эти методы в методах setUp () и tearDown () своих модульных тестов, и они отлично работают! У вас есть чистая копия ваших файлов, чтобы протестировать ваше файловое поле, которое можно использовать повторно и предсказуемо.

с pytest и pytest-django я использую это в conftest.py файле:

import tempfile
import shutil
from pytest_django.lazy_django import skip_if_no_django
from pytest_django.fixtures import SettingsWrapper


@pytest.fixture(scope='session')
#@pytest.yield_fixture()
def settings():
    """A Django settings object which restores changes after the testrun"""
    skip_if_no_django()

    wrapper = SettingsWrapper()
    yield wrapper
    wrapper.finalize()


@pytest.fixture(autouse=True, scope='session')
def media_root(settings):
    tmp_dir = tempfile.mkdtemp()
    settings.MEDIA_ROOT = tmp_dir
    yield settings.MEDIA_ROOT
    shutil.rmtree(tmp_dir)


@pytest.fixture(scope='session')
def django_db_setup(media_root, django_db_setup):
    print('inject_after')

может быть полезно:

1. https://dev.funkwhale.audio/funkwhale/funkwhale/blob/de777764da0c0e9fe66d0bb76317679be964588b/api/tests/conftest.py
2. https://framagit.org/ideascube/ideascube/blob/master/conftest.py
3. https://stackoverflow.com/a/56177770/5305401

Это то, что я сделал для своего теста. После загрузки файла он должен оказаться в свойстве photo объекта модели моей организации:

    import tempfile
    filename = tempfile.mkstemp()[1]
    f = open(filename, 'w')
    f.write('These are the file contents')
    f.close()
    f = open(filename, 'r')
    post_data = {'file': f}
    response = self.client.post("/org/%d/photo" % new_org_data["id"], post_data)
    f.close()
    self.assertEqual(response.status_code, 200)

    ## Check the file
    ## org is where the file should end up
    org = models.Organization.objects.get(pk=new_org_data["id"])
    self.assertEqual("These are the file contents", org.photo.file.read())

    ## Remove the file
    import os
    os.remove(org.photo.path)