我需要测试我的 Django 应用程序的照片模型。如何使用测试图像文件模拟 ImageField？
tests.py
class PhotoTestCase(TestCase):
def test_add_photo(self):
newPhoto = Photo()
newPhoto.image = # ??????
newPhoto.save()
self.assertEqual(Photo.objects.count(), 1)
【问题讨论】：
标签： python django testing mocking
对于未来的用户，我已经解决了这个问题。
您可以使用 SimpleUploadedFile 实例模拟 ImageField。
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')
【讨论】：
name 是您要用于读取的图像的文件名，image_path 是图像的路径。例如/path/to/image.jpg。
您可以使用临时文件，使用tempfile。所以你不需要一个真实的文件来做你的测试。
import tempfile
image = tempfile.NamedTemporaryFile(suffix=".jpg").name
如果您更喜欢手动清理，请改用tempfile.mkstemp()。
【讨论】：
告诉 mock 库根据 Django 的 File 类创建一个 mock 对象
import mock
from django.core.files import File
file_mock = mock.MagicMock(spec=File, name='FileMock')
然后在你的测试中使用
newPhoto.image = file_mock
【讨论】：
newPhoto.image = open(/path/to/foo.jpg)
TypeError: 'MagicMock' does not support the buffer interface
如果你不想在文件系统中创建一个实际的文件，你可以使用这个 37 字节的 GIF，小到足以在你的代码中成为字节文字：
from django.core.files.uploadedfile import SimpleUploadedFile
small_gif = (
b'\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x00\x00\x00\x21\xf9\x04'
b'\x01\x0a\x00\x01\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02'
b'\x02\x4c\x01\x00\x3b'
)
uploaded = SimpleUploadedFile('small.gif', small_gif, content_type='image/gif')
【讨论】：
b'\x47\x49\x46\x38\x39\x61\x01\x00\x01\x00\x80\x00\x00\x05\x04\x04\x00\x00\x00\x2c\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02\x44\x01\x00\x3b'
解决方案：
from StringIO import StringIO
# in python 3: from io import StringIO
from PIL import Image
from django.core.files.base import File
并在您的 TestCase 类中创建一个静态方法：
@staticmethod
def get_image_file(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)):
file_obj = StringIO()
image = Image.new("RGB", size=size, color=color)
image.save(file_obj, ext)
file_obj.seek(0)
return File(file_obj, name=name)
例子：
instance = YourModel(name=value, image=self.get_image_file())
【讨论】：
from io import StringIO
from io import BytesIO 而不是 StringIO
您可以做一些额外的事情来 (1) 避免保留专用的测试映像，以及 (2) 确保在测试期间创建的所有测试文件在之后立即被删除：
import shutil
import tempfile
from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import TestCase, override_settings
MEDIA_ROOT = tempfile.mkdtemp()
@override_settings(MEDIA_ROOT=MEDIA_ROOT)
class MyTest(TestCase):
@classmethod
def tearDownClass(cls):
shutil.rmtree(MEDIA_ROOT, ignore_errors=True) # delete the temp dir
super().tearDownClass()
def test(self):
img = SimpleUploadedFile('test.jpg', b'whatevercontentsyouwant')
# ^-- this will be saved in MEDIA_ROOT
# do whatever ...
【讨论】：
我的方法是如何在不传递任何有用数据的情况下测试模型：
from django.core.files import File
SomeModel.objects.create(image=File(file=b""))
【讨论】：
有人尝试使用 python 3.xx 进行上传图像测试
我很少用 Maxim Panfilov 的出色回答来制作更多具有独立名称的虚拟图像。
from io import BytesIO
from PIL import Image
from django.core.files.base import File
#in your TestCase class:
class TestClass(TestCase):
@staticmethod
def get_image_file(name, ext='png', size=(50, 50), color=(256, 0, 0)):
file_obj = BytesIO()
image = Image.new("RGBA", size=size, color=color)
image.save(file_obj, ext)
file_obj.seek(0)
return File(file_obj, name=name)
def test_upload_image(self):
c= APIClient()
image1 = self.get_image('image.png')
image2 = self.get_image('image2.png')
data =
{
"image1": iamge1,
"image2": image2,
}
response = c.post('/api_address/', data )
self.assertEqual(response.status_code, 201)
【讨论】：
如果您使用Factory Boy 生成测试数据，该库会使用ImageField factory 处理这种情况。
这是一个完整的例子。我假设所有这些文件都在同一个 Django 应用程序中。
models.py 示例：
from django.db import models
class YourModel(models.Model):
image = models.ImageField(upload_to="files/")
factories.py 示例：
import factory
from . import models
class YourModelFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.YourModel
image = factory.Django.ImageField()
tests.py 示例：
from django import test
from . import factories
class YourModelTests(test.TestCase):
def test_image_model(self):
yourmodel = factories.YourModelFactory()
self.assertIsNotNone(yourmodel.image)
【讨论】：