如何在 Django Rest Framework 上测试图像上传

Question

我在挣扎。问题在于单元测试（“test.py”），我想出了如何使用 tempfile 和 PIL 上传图像，但这些临时图像永远不会被删除。我考虑创建一个临时目录，然后使用 os.remove 删除该 temp_dir，但图像上传到不同的媒体目录取决于模型，所以我真的不知道如何发布 temp_images然后删除它们。

这是我的 models.py

class Noticia(models.Model):
  ...
  img = models.ImageField(upload_to="noticias", storage=OverwriteStorage(), default="noticias/tanque_arma3.jpg")
  ...

test.py

def temporary_image():
    import tempfile
    from PIL import Image

    image = Image.new('RGB', (100, 100))
    tmp_file = tempfile.NamedTemporaryFile(suffix='.jpg', prefix="test_img_")
    image.save(tmp_file, 'jpeg')
    tmp_file.seek(0)
    return tmp_file

class NoticiaTest(APITestCase):
    def setUp(self):
        ...
        url = reverse('api:noticia-create')
        data = {'usuario': usuario.pk, "titulo":"test", "subtitulo":"test", "descripcion":"test", "img": temporary_image()}
        response = client.post(url, data,format="multipart")

        ...

所以，总而言之，问题是，¿我如何从不同目录中删除临时文件，考虑到这些文件必须严格上传到这些目录上？

Answer 1

您可以使用包dj-inmemorystorage 进行测试，Django 不会保存到磁盘。序列化程序和模型仍将按预期工作，如果需要，您可以将数据读回。

在您的设置中，当您处于测试模式时，覆盖默认文件存储。您还可以在此处放置任何其他“测试模式”设置，只要确保它在您的其他设置之后运行。

if 'test' in sys.argv :
    # store files in memory, no cleanup after tests are finished
    DEFAULT_FILE_STORAGE = 'inmemorystorage.InMemoryStorage'
    # much faster password hashing, default one is super slow (on purpose)
    PASSWORD_HASHERS = ['django.contrib.auth.hashers.MD5PasswordHasher']

当您上传文件时，您可以使用SimpleUploadFile，它纯粹是在内存中的。这负责“客户端”端，而 dj-inmemorystorage 包负责 Django 的存储。

def temporary_image():
    bts = BytesIO()
    img = Image.new("RGB", (100, 100))
    img.save(bts, 'jpeg')
    return SimpleUploadedFile("test.jpg", bts.getvalue())