python - template - ¿Cómo puedo probar los mensajes de django por unidad?

django template variable

Question

En mi aplicación django, intento escribir una prueba unitaria que realiza una acción y luego verifico los mensajes en la respuesta.

Por lo que puedo decir, no hay una buena manera de hacer esto.

Estoy usando el método de almacenamiento CookieStorage, y me gustaría hacer algo similar a lo siguiente:

    response = self.client.post(''/do-something/'', follow=True)
    self.assertEquals(response.context[''messages''][0], "fail.")

El problema es que todo lo que recupero es una

print response.context[''messages'']
<django.contrib.messages.storage.cookie.CookieStorage object at 0x3c55250>

¿Cómo puedo convertir esto en algo útil, o lo estoy haciendo todo mal?

Gracias, Daniel

Answer 1

Actualizar

Mi respuesta original fue escrita cuando django todavía era 1.1 o más. Esta respuesta ya no es relevante. Ver la answer @ daveoncode para una mejor solución.

Respuesta original

Hice un experimento para probar esto. Cambié la configuración de MESSAGE_STORAGE en uno de mis proyectos a ''django.contrib.messages.storage.cookie.CookieStorage'' y ''django.contrib.messages.storage.cookie.CookieStorage'' una prueba que había escrito para verificar si había mensajes. Funcionó.

La diferencia clave de lo que estabas haciendo es la forma en que recuperé los mensajes. Vea abajo:

def test_message_sending(self):
    data = dict(...)
    response = self.client.post(reverse(''my_view''), data)
    messages = self.user.get_and_delete_messages()

    self.assertTrue(messages)
    self.assertEqual(''Hey there!'', messages[0])

Esto puede valer la pena intentarlo.

Answer 2

De la documentación de django :

Fuera de las plantillas, puedes usar get_messages ()

Entonces, podrías escribir algo como:

from django.contrib.messages import get_messages

[...]

messages = [m.message for m in get_messages(response.wsgi_request)]
self.assertIn(''My message'', messages)

Answer 3

Encontré un enfoque realmente fácil:

response = self.client.post(''/foo/'')
messages = list(response.context[''messages''])
self.assertEqual(len(messages), 1)
self.assertEqual(str(messages[0]), ''my message'')

Si necesita verificar los mensajes en una respuesta que no tiene contexto, puede usar lo siguiente:

from django.contrib.messages import get_messages
messages = list(get_messages(response.wsgi_request))
self.assertEqual(len(messages), 1)
self.assertEqual(str(messages[0]), ''my message'')

El almacenamiento alternativo no es compatible con la indexación, sin embargo, es iterable.

Answer 4

Esto funciona para mí (muestra todos los mensajes):

print [m.message for m in list(response.context[''messages''])]

También aquí hay un par de métodos de utilidad que tengo en una clase de prueba heredada de TestCase de Django. Si prefiere tenerlos como funciones, elimine los argumentos self.fail() y reemplace self.fail() con un raise .

def assert_message_count(self, response, expect_num):
    """
    Asserts that exactly the given number of messages have been sent.
    """

    actual_num = len(response.context[''messages''])
    if actual_num != expect_num:
        self.fail(''Message count was %d, expected %d'' %
            (actual_num, expect_num))

def assert_message_contains(self, response, text, level=None):
    """
    Asserts that there is exactly one message containing the given text.
    """

    messages = response.context[''messages'']

    matches = [m for m in messages if text in m.message]

    if len(matches) == 1:
        msg = matches[0]
        if level is not None and msg.level != level:
            self.fail(''There was one matching message but with different''
                ''level: %s != %s'' % (msg.level, level))

        return

    elif len(matches) == 0:
        messages_str = ", ".join(''"%s"'' % m for m in messages)
        self.fail(''No message contained text "%s", messages were: %s'' %
            (text, messages_str))
    else:
        self.fail(''Multiple messages contained text "%s": %s'' %
            (text, ", ".join((''"%s"'' % m) for m in matches)))

def assert_message_not_contains(self, response, text):
    """ Assert that no message contains the given text. """

    messages = response.context[''messages'']

    matches = [m for m in messages if text in m.message]

    if len(matches) > 0:
        self.fail(''Message(s) contained text "%s": %s'' %
            (text, ", ".join((''"%s"'' % m) for m in matches)))

Answer 5

Una versión más simple del punto muerto:

class TestCaseMessagesMixture(object):
    def assertMessageCount(self, response, expect_num):
        """
        Asserts that exactly the given number of messages have been sent.
        """

        actual_num = len(response.context[''messages''])
        if actual_num != expect_num:
            self.fail(''Message count was %d, expected %d'' %
                    (actual_num, expect_num)
                )

    def assertMessageEqual(self, response, text):
        """
        Asserts that the response includes the message text.
        """

        messages = [m.message for m in response.context[''messages'']]

        if text not in messages:
            self.fail(
                ''No message with text "%s", messages were: %s'' % 
                    (text, messages)
                )

    def assertMessageNotEqual(self, response, text):
        """
        Asserts that the response does not include the message text.
        """

        messages = [m.message for m in response.context[''messages'']]

        if text in messages:
            self.fail(
                ''Message with text "%s" found, messages were: %s'' % 
                    (text, messages)
                )