Django Test mocked ImageField在测试后阻止上传或清理

Question

我正在为我的项目开发一个TestCase，涉及一些Image和FileFields。一些对象被加载了灯具，一些对象在测试过程中被创建。我用InMemoryUploadedFile模拟了我的图像。

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

这些模拟图像实际上是在测试过程中本地存储的。有很多关于的信息如何模拟图像。但是我无法真正找到一些关于在测试之后清理这些上传的好消息。

问题：
，以防止有从实际被上传这些文件的可能性，或者这是不可能的/ DIS-鼓励？

或
如果这些文件有利于测试的上载（如果是这样，为什么？），和拆卸过程中事后清理？那么清理这些上传文件的适当方法是什么？

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

八分型号：

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 /目录会被自动删除。所有的测试都通过了。

老答案
我能够从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) 

嘲笑与default_storage一个mock.patch()存储不幸的是，我没能测试包含一个update()的save()方法中的典范：django-mocking-storage-on-model-with-update-throws-error