首页 > 解决方案 > 测试发布请求时的Django消息中间件问题

问题描述

我正在尝试测试将消息添加到重定向成功页面的 UpdateView。看来我的问题来自于messagespytest 返回:

django.contrib.messages.api.MessageFailure:不安装 django.contrib.messages.middleware.MessageMiddleware 就无法添加消息

我的测试代码是:

def test_authenticated_staff(self, rf):
    langues = LanguageCatalog.objects.create(
        lang_src='wz',
        lang_dest='en',
        percent='4'
    )
    req = rf.get(reverse("dashboard.staff:lang-update", kwargs={'pk': langues.pk}))
    data = {'lang_src': 'it',
            'lang_dest': 'en',
            'percent': '34'}
    req = rf.post(reverse(
        "dashboard.staff:lang-update", kwargs={'pk': langues.pk}), data=data)
    req.user = UserFactory()
    resp = views.LangUpdateView.as_view()(req, pk=langues.pk)

我确切地说它MessageMiddleware存在于MIDDLEWARE设置中。我用Django==2.0.13.

标签: pythondjangotestingpytest

解决方案


我找到了解决方案。为了测试这样的请求,您需要先用会话注释它,然后再用消息注释它。实际上这意味着添加这些行:

from django.contrib.messages.middleware import MessageMiddleware
from django.contrib.sessions.middleware import SessionMiddleware

# in your test method:
"""Annotate a request object with a session"""
middleware = SessionMiddleware()
middleware.process_request(req)
req.session.save()

"""Annotate a request object with a messages"""
middleware = MessageMiddleware()
middleware.process_request(req)
req.session.save()

# and then (in my case)
resp = views.LangUpdateView.as_view()(req, pk=langues.pk)

推荐阅读