单元测试：如何模拟 MEDIA_ROOT 但仍然访问存储在我的普通 MEDIA 文件夹中的文件

Question

我正在测试我的一个网页 POST 功能。if request.method == 'POST'它生成一个 pdf 并将其附加到用户。每次我运行测试时，我都会生成在我的 MEDIA 文件夹中构建的 pdf 文件，我必须手动删除它们。我决定寻找防止这种情况发生的方法，在我看来最好的方法是用临时文件覆盖我的 MEDIA_ROOT。

我收到以下错误消息

FileNotFoundError: [Errno 2] No such file or directory: 'C:\\Users\\Acer\\AppData\\Local\\Temp\\profile_pics\\default.jpg'

我确定这是因为我必须在运行测试之前创建并登录一个用户（我不想在设置中这样做，因为我必须为不同的测试生成不同类型的用户）。每次创建用户时，他们的个人资料图片都会设置为profile_pics\\default.jpg'。

我的理解是模拟的 MEDIA_ROOT 是空的，不包含默认的配置文件图片，所以会抛出错误。

我的问题是如何绕过这个（最好是模拟 default.jpg 图像）？除了模拟 MEDIA_ROOT 文件夹之外，我还找到了其他解决方案，但它们看起来很笨拙，我更喜欢正确地执行此操作，并希望从中学习。

测试.py

from django.test import override_settings
import tempfile

@override_settings(MEDIA_ROOT=tempfile.gettempdir())
def test_redirects_after_POST(self):
    user = User.objects.create_superuser('username')
    self.client.force_login(user)

    response = self.client.post(
        reverse('final_question'), data={
        'answer': 'this has me totally stumped'}
    )

    self.assertRedirects(response, reverse('page_2'))

谢谢你。

Answer 1

问题：

PDF 文件会在您的 MEDIA_ROOT 文件夹中堆积，因为没有任何程序会自动删除它们。

推荐解决方案：

事实上，几个月前我自己也遇到过这个问题，我找到的解决方案是django-cleanup工具。它非常容易安装并集成到您的应用程序中（只需将其添加到您的INSTALLED_APPSin中settings.py）。

注意：这是假设您的 PDF 文件位于FileField您的其中一个模型中。如果情况并非如此，那么此解决方案将不适合您。