diff options
Diffstat (limited to 'django')
| -rw-r--r-- | django/contrib/messages/test.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/django/contrib/messages/test.py b/django/contrib/messages/test.py new file mode 100644 index 0000000000..3a69f54585 --- /dev/null +++ b/django/contrib/messages/test.py @@ -0,0 +1,8 @@ +from .api import get_messages + + +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) |