28

我一直在为我的一个 django 应用程序编写测试，并且一直在寻找解决这个问题已经有一段时间了。我有一个视图可以发送django.contrib.messages用于不同情况的消息。该视图如下所示。

from django.contrib import messages
from django.shortcuts import redirect

import custom_messages

def some_view(request):
    """ This is a sample view for testing purposes.
    """

    some_condition = models.SomeModel.objects.get_or_none(
        condition=some_condition)
    if some_condition:
        messages.success(request, custom_message.SUCCESS)
    else:
        messages.error(request, custom_message.ERROR)
    redirect(some_other_view)

现在，在测试这个 viewclient.get的响应时不包含包含 的context字典，messages因为这个 view 使用了重定向。对于呈现模板的视图，我们可以使用messages = response.context.get('messages'). 我们如何获得messages重定向视图的访问权限？

4

5 回答 5

46

在调用中使用该follow=True选项，client.get()客户端将遵循重定向。然后，您可以测试该消息是否在您重定向到的视图的上下文中。

def test_some_view(self):
    # use follow=True to follow redirect
    response = self.client.get('/some-url/', follow=True)

    # don't really need to check status code because assertRedirects will check it
    self.assertEqual(response.status_code, 200)
    self.assertRedirects(response, '/some-other-url/')

    # get message from context and check that expected text is there
    message = list(response.context.get('messages'))[0]
    self.assertEqual(message.tags, "success")
    self.assertTrue("success text" in message.message)
于 2013-04-22T09:38:15.177 回答
8

您可以像这样将 get_messages() 与 response.wsgi_request 一起使用（在 Django 1.10 中测试）：

from django.contrib.messages import get_messages  
...
def test_view(self):
    response = self.client.get('/some-url/') # you don't need follow=True
    self.assertRedirects(response, '/some-other-url/')
    # each element is an instance of  django.contrib.messages.storage.base.Message
    all_messages = [msg for msg in get_messages(response.wsgi_request)]

    # here's how you test the first message
    self.assertEqual(all_messages[0].tags, "success")
    self.assertEqual(all_messages[0].message, "you have done well")
于 2017-02-15T14:34:51.307 回答
1

如果您的视图正在重定向并且您follow=true在对测试客户端的请求中使用上述内容将不起作用。我最终编写了一个辅助函数来获取与响应一起发送的第一条（在我的情况下是唯一的）消息。

@classmethod
def getmessage(cls, response):
    """Helper method to return message from response """
    for c in response.context:
        message = [m for m in c.get('messages')][0]
        if message:
            return message

您将其包含在您的测试类中并像这样使用它：

message = self.getmessage(response)

response你从 aget或post到 a得到的东西在哪里Client。

这有点脆弱，但希望它可以节省其他人一些时间。

于 2013-09-12T00:59:32.337 回答
1

我在使用 3rd 方应用程序时遇到了同样的问题。

如果您想从另一个视图中返回 HttpResponseRedict （您无法从中访问上下文）的视图中获取消息，您可以使用get_messages(request) 

from django.contrib.messages import get_messages  

storage = get_messages(request)  
for message in storage:  
    do_something_with_the_message(message)

不过，这会清除消息存储，因此如果您想稍后从模板访问消息，请添加：

storage.used = False
于 2014-06-24T07:41:58.900 回答
1

替代方法模拟消息（不需要遵循重定向）：

from mock import ANY, patch
from django.contrib import messages

@patch('myapp.views.messages.add_message')
def test_some_view(self, mock_add_message):
    r = self.client.get('/some-url/')
    mock_add_message.assert_called_once_with(ANY, messages.ERROR, 'Expected message.')  # or assert_called_with, assert_has_calls...
于 2017-05-23T07:26:59.900 回答