web-dev-qa-db-ja.com

Djangoメッセージを単体テストするにはどうすればよいですか？

Django=アプリケーションでは、アクションを実行してから応答のメッセージを確認する単体テストを作成しようとしています。

私の知る限り、これを行うための良い方法はありません。

私はCookieStorageストレージメソッドを使用しており、次のようなことをしたいと思います。

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

問題は、私が返すのは

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

これを便利なものにするにはどうすればよいですか、それともすべて間違っていますか？

ありがとう、ダニエル

pythondjangounit-testingdjango-testing
70
dvydra

私は本当に簡単なアプローチを見つけました：

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

コンテキストのない応答でメッセージを確認する必要がある場合は、以下を使用できます。

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')

フォールバックストレージはインデックス作成をサポートしていませんが、反復可能です。

64
daveoncode

これは私のために働きます（すべてのメッセージを表示します）：

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

また、DjangoのTestCaseから継承したテストクラスにあるユーティリティメソッドがいくつかあります。それらを関数として使用したい場合は、self引数を削除し、self.fail()を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)))
_
17
anttikoo

Djangoのドキュメント から：

テンプレート以外では、get_messages（）を使用できます

したがって、次のように書くことができます。

from Django.contrib.messages import get_messages

[...]

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

更新

私の元の回答はDjangoがまだ1.1程度だったときに書かれました。この回答はもはや関係ありません。より良い解決策については@daveoncodeの answer を参照してください。

元の回答

私はこれをテストする実験をしました。プロジェクトのMESSAGE_STORAGE設定を'Django.contrib.messages.storage.cookie.CookieStorage'に変更し、メッセージをチェックするために作成したテストを実行しました。出来た。

あなたがしていたこととの主な違いは、私がメッセージを取得する方法です。下記参照：

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])

これは試してみる価値があるかもしれません。

3
Manoj Govindan

行き詰まったもののより単純なバージョン：

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)
                )
0
Marco Fucci