python - 使用 ImageField 的 Django 测试模型

Question

我需要测试我的 Django 应用程序的照片模型。如何使用测试图像文件模拟 ImageField？

tests.py

class PhotoTestCase(TestCase):

    def test_add_photo(self):
        newPhoto = Photo()
        newPhoto.image = # ??????
        newPhoto.save()
        self.assertEqual(Photo.objects.count(), 1)

Answer 1

对于 future 的用户，我已经解决了这个问题。您可以模拟 ImageField与 SimpleUploadedFile实例。

test.py

from django.core.files.uploadedfile import SimpleUploadedFile

newPhoto.image = SimpleUploadedFile(name='test_image.jpg', content=open(image_path, 'rb').read(), content_type='image/jpeg')