blob: 9f3e1bf0bcb0568b9633e95d4532b0cf34ab7c0e (
plain)
1
2
3
4
5
6
7
8
9
10
11
|
from .api import get_messages
# Make unittest ignore frames in this module when reporting failures.
__unittest = True
class MessagesTestMixin:
def assertMessages(self, response, expected_messages, *, ordered=True):
request_messages = list(get_messages(response.wsgi_request))
assertion = self.assertEqual if ordered else self.assertCountEqual
assertion(request_messages, expected_messages)
|
