Django 测试模拟 ImageField 在测试后阻止上传或清理

Question

我正在为我的项目开发一个测试用例，涉及一些图像和文件字段。有些对象加载了夹具，有些对象是在测试期间创建的。我用 InMemoryUploadedFile 模拟了我的图像。

创建新的测试对象时，正在上传模拟图像
到 MEDIA_ROOT 目录，在 upload_to 方法中定义的正确位置。

这些模拟图像实际上是在测试期间存储在本地的。有很多关于如何模拟图像的信息。但是我真的找不到一些关于在测试后清理这些上传的好信息。

问题 :
是否有可能阻止这些文件实际上传，或者这是不可能的/不鼓励的？

或
是否应该上传这些文件以支持测试(如果是，为什么？)，然后在拆卸期间进行清理？清理这些上传的文件的合适方法是什么？

我正在处理的当前情况的一些代码，裁剪到适合问题的大小。我不是在寻找一个完整的样本，而只是朝着正确的方向努力。提前感谢您的时间。

裁剪 模型:

def project_image_upload_location(instance, filename):
    return 'uploads/projects/images/%s' % \
        services.unique_filename(filename)


class ProjectImage(models.Model):
    project = models.ForeignKey(
        'projects.Project')
    name = models.CharField(
        _("Image name"),
        max_length=35)
    image = models.ImageField(
        _("Project image"),
        upload_to=project_image_upload_location,
        width_field='image_width',
        height_field='image_height')
    image_width = models.IntegerField(
        default=0)
    image_height = models.IntegerField(
        default=0)

裁剪 测试用例:

from django.core.files.uploadedfile import InMemoryUploadedFile
from django.test import TestCase
from PIL import Image
import StringIO

from projects.models import ProjectImage
from projects import services


class ProjectsTest(TestCase):
    fixtures = ['projects']

    def _create_project_image(self, project, name):
        return ProjectImage.objects.create(
                project=project,
                name=name,
                description="Description",
                image=self._create_mock_image(name="Mocked Image"),
                is_primary=True,
                is_published=True)

        def _get_project_image(self, name):
            return ProjectImage.objects.get(name=name)

    def _create_mock_image(self, name):
        name = name + '.jpg'
        io = StringIO.StringIO()
        size = (200,200)
        color = (255,0,0)
        image = Image.new("RGB", size, color)
        image.save(io, format='JPEG')
        image_file = InMemoryUploadedFile(io, None, name, 'jpeg', io.len, None)
        image_file.seek(0)
        return image_file

    def test_update_project_set_updated_datetime(self):
        project = self._get_project("Project B")
        self.assertEqual(project.updated_datetime, None) 

        project.save()
        self.assertTrue(isinstance(project.updated_datetime, datetime))

Answer 1

我已经用 testing-a-model-that-have-an-imagefield 改变了方法.图像将上传到临时目录。该指南建议删除 tearDown 中的目录。 .它可能会被更新，因为 temp/目录会被自动删除。所有测试都正确通过。

旧答案
我可以用 mock.patch() 模拟存储在 default_storage来自 joeray.me: mocking-files-and-file-storage-for-testing-django-models

<..crop..>
import mock

current_storage = 'django.core.files.storage.default_storage._wrapped'
def _mock_storage():
    return mock.MagicMock(spec=Storage, name="StorageMock")

class ProjectsTest(TestCase):
    fixtures = ['projects']

    @mock.patch(current_storage, _mock_storage())
    def _create_project(self, name):
    return Project.objects.create(
            name=name,
            short_description="Short description A",
            full_description="Full description A",
            url="http://test-project-url.com/",
            is_published=True)

不幸的是，我无法测试包含 update() 的模型。内save()方法:django-mocking-storage-on-model-with-update-throws-error