Question

我有一个我需要测试的函数，它将请求作为参数。它不会通过URL公开为视图，因此我无法使用测试客户端对其进行测试。

我需要将请求对象传递给它，并且请求对象需要启用消息中间件，因为函数中使用了消息传递中间件。

我使用RequestFactory来创建我的请求。 documentation说：

它不支持中间件。如果视图需要正常运行，则必须由测试本身提供会话和身份验证属性。

如何使用RequestFactory设置消息传递中间件？我想我还需要会话中间件来使消息传递中间件工作

这是我的测试当前使用vanilla 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)

Answer 1

此问题was raised，如上所述，您可以使用该代码修复单元测试：

from django.contrib.messages.storage.fallback import FallbackStorage
setattr(request, 'session', 'session')
messages = FallbackStorage(request)
setattr(request, '_messages', messages)

Answer 2

如果您不需要测试请求对象本身的行为，您可以使用mock库而不是RequestFactory来模拟请求，例如：

import mock

request = mock.MagicMock()

# Call your function using the mocked request
store_to_request(request)

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

2 个答案: