Проверка Django - проверка сообщений для представления, которое перенаправляет

Я пишу тесты для одного из своих приложений 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)

Теперь при тестировании этого представления client.get ответ не содержит словаря context, который содержит messages, так как в этом представлении используется перенаправление. Для представлений, отображающих шаблоны, мы можем получить доступ к списку сообщений с помощью messages = response.context.get('messages'). Как мы можем получить доступ messages для представления, которое перенаправляет?

Ответ 1

Используйте параметр 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)

Ответ 2

Вы можете использовать get_messages() с ответом .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 how you test the first message
    self.assertEqual(all_messages[0].tags, "success")
    self.assertEqual(all_messages[0].message, "you have done well")

Ответ 3

Если ваши представления перенаправляются и вы используете 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 - это то, что вы возвращаете с get или post на Client.

Это немного хрупкий, но, надеюсь, это сэкономит кому-то еще некоторое время.

Ответ 4

У меня была такая же проблема при использовании стороннего приложения.

Если вы хотите получать сообщения из представления, которое возвращает 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

Ответ 5

Альтернативный метод издевательских сообщений (не нужно переадресовывать):

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...