如何在Django测试中提供InMemoryStorage媒体文件服务

核心方案是自定义存储感知的媒体文件服务视图，替代默认的django.views.static.serve，让它能从当前配置的存储后端（包括InMemoryStorage）读取文件并返回给浏览器。

1. 自定义存储感知视图 #

在项目中创建一个视图（比如views.py或专门的test_utils/views.py），实现如下逻辑：

from django.http import FileResponse, Http404
from django.core.files.storage import default_storage

def serve_storage_media(request, path):
    storage = default_storage
    # 检查文件是否存在于当前存储后端
    if not storage.exists(path):
        raise Http404("媒体文件不存在")
    # 从存储后端打开文件并返回响应
    file_handle = storage.open(path)
    # 自动识别文件内容类型，也可根据需求手动指定
    response = FileResponse(file_handle)
    return response

这个视图完全基于Django的标准存储API，不管你用的是InMemoryStorage还是其他后端（如S3、本地文件系统），只要符合Django存储接口规范，就能正常工作。

2. 添加测试环境专属路由 #

在项目的urls.py中，针对测试环境添加该视图的路由（可通过自定义的测试环境标识判断）：

from django.urls import re_path
from django.conf import settings
from .test_utils.views import serve_storage_media  # 替换为实际的视图导入路径

# 仅在测试环境启用该路由
if getattr(settings, "TESTING", False):
    urlpatterns += [
        re_path(
            r"^%s(?P<path>.*)$" % re.escape(settings.MEDIA_URL.lstrip("/")),
            serve_storage_media,
        ),
    ]

其中settings.TESTING是你在测试配置中自定义的标识（比如在settings/test.py里设置TESTING = True），确保只在测试阶段启用这个路由。

3. 配置测试环境的媒体存储 #

在测试配置文件（如settings/test.py）中，明确指定使用InMemoryStorage：

from django.core.files.storage import InMemoryStorage

DEFAULT_FILE_STORAGE = "django.core.files.storage.InMemoryStorage"
MEDIA_URL = "/media/"
# InMemoryStorage不需要实际物理路径，但保留MEDIA_ROOT占位以适配Django逻辑
MEDIA_ROOT = "/tmp/test_media_temp/"

方案原理说明 #

默认的django.views.static.serve直接操作本地文件系统，只能读取document_root下的物理文件；而自定义视图通过Django统一的default_storage接口，完全适配InMemoryStorage的内存文件存储逻辑，能正确读取测试中上传的媒体文件，解决Selenium请求返回404的问题。

内容的提问来源于stack exchange，提问作者mgalgs