Django使用pytest单元测试时模板未渲染Messages消息问题求助

问题根因 #

你当前遇到的问题是测试流程跳过了Django上下文处理器的执行步骤：正常请求流程中，模板用到的messages变量由django.contrib.messages.context_processors.messages上下文处理器自动注入模板上下文，你使用RequestFactory直接构造请求调用视图的方式，没有触发上下文处理器的执行逻辑，模板读取不到messages变量，所以{% if messages %}判断为假，不会渲染对应的内容。
因为你已经确认正常业务视图中消息可以正常渲染，说明你的项目TEMPLATES配置已经正确添加了对应的上下文处理器，不需要额外修改配置。

两种修复方案 #

方案1：使用TestClient模拟请求（更推荐） #

TestClient会走完整的Django请求处理链路，自动执行中间件、上下文处理器，不需要你手动初始化中间件，代码更简洁：

import pytest
from django.contrib import messages
from django.views.generic import TemplateView

pytestmark = pytest.mark.django_db

class TestBaseTemplate:
    def test_messages_middleware(self, client):
        # 构造临时测试视图，添加消息后渲染base模板
        def test_view(request):
            messages.info(request, "Hello, world!")
            return TemplateView.as_view(template_name="base.html")(request)
        
        # 临时注册路由
        from django.urls import path
        from django.conf import settings
        settings.ROOT_URLCONF = [path("", test_view)]
        
        response = client.get("/")
        assert "Hello, world!" in response.content.decode("utf-8")

方案2：保留RequestFactory，手动注入messages变量 #

如果你需要保留现有RequestFactory的实现逻辑，只需要在渲染前手动将messages注入模板上下文即可：

import pytest
from django.contrib import messages
from django.contrib.messages import get_messages
from django.contrib.messages.middleware import MessageMiddleware
from django.contrib.sessions.middleware import SessionMiddleware
from django.test import RequestFactory
from django.views.generic import TemplateView

pytestmark = pytest.mark.django_db

class TestBaseTemplate:
    def test_messages_middleware(self):
        request = RequestFactory().get("/")
        view = TemplateView.as_view(template_name="base.html")

        middleware1 = SessionMiddleware()
        middleware1.process_request(request)
        request.session.save()

        middleware2 = MessageMiddleware()
        middleware2.process_request(request)

        foobar_msg = "Hello, world!"
        messages.info(request, foobar_msg)

        request.session.save()
        # 调用视图时手动传入messages到上下文
        response = view(request, extra_context={"messages": get_messages(request)})

        response.render()
        content = response.content.decode("utf-8")

        assert foobar_msg in content 

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