沃梦达 / 编程问答 / Python问题 / 正文

使用 ImageField 的 Django 测试模型

沃梦达教程 Python问题

2022-01-01

Django testing model with ImageField(使用 ImageField 的 Django 测试模型)

本文介绍了使用 ImageField 的 Django 测试模型的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

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

I need to test the Photo model of my Django application. How can I mock the ImageField with a test image file?

tests.py

class PhotoTestCase(TestCase):

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

推荐答案

对于未来的用户，我已经解决了问题.您可以模拟 ImageField 带有 SimpleUploadedFile 实例.


For future users, I've solved the problem.
You can mock an ImageField with a SimpleUploadedFile instance.
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')


                        这篇关于使用 ImageField 的 Django 测试模型的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！

沃梦达教程

本文标题为：使用 ImageField 的 Django 测试模型

上一篇：断言未使用 Mock 调用函数/方法

下一篇：如何通过 unittest setUp 在 python 中正确使用 mock