python – 如何在单元测试期间在Django RequestFactory中设置消息传递和会话中间件

前端之家收集整理的这篇文章主要介绍了python – 如何在单元测试期间在Django RequestFactory中设置消息传递和会话中间件前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我有一个我需要测试的函数,它需要一个请求作为它的参数.它不是通过URL显示为视图,因此我无法使用测试客户端进行测试.

我需要传递请求对象,并且请求对象需要启用消息传递中间件,因为在该功能中使用消息传递中间件.

我正在使用RequestFactory来创建我的请求. documentation说:

It does not support middleware. Session and authentication attributes must be supplied by the test itself if required for the view to function properly.

如何使用RequestFactory设置邮件中间件?我想我也需要会话中间件来使消息中间件工作

这是我当前使用香草RequestFactory时测试的错误.

MessageFailure: You cannot add messages without installing django.contrib.messages.middleware.MessageMiddleware

这是我正在测试的功能,以防它有助于理解这个问题:

from django.contrib import messages as django_messages

def store_to_request(self,request):
        """
        Place all the messages stored in this class into message storage in
        the request object supplied.
        :param request: The request object to which we should store all the messages
        :return: Does not return anything
        """
        for message in self._messages:
            django_messages.add_message(request,message.level,message.message,message.extra_tags,message.fail_silently)

解决方法

这个问题 was raised,就这样说,你可以使用该代码修复你的单元测试:
from django.contrib.messages.storage.fallback import FallbackStorage
setattr(request,'session','session')
messages = FallbackStorage(request)
setattr(request,'_messages',messages)

猜你在找的Python相关文章