summaryrefslogtreecommitdiff
path: root/django
diff options
context:
space:
mode:
authorFrançois Freitag <mail@franek.fr>2023-06-15 17:16:46 +0200
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2023-09-05 11:55:37 +0200
commitcafe7266ee69f7e017ddbc0d440084ace559b04b (patch)
treeed48113e9c2002d37c1249ad17931e4ca486448f /django
parent4f0c0e6fa1ce2e4da9407f122d0c1c27c7f8ad83 (diff)
Fixed #34730 -- Added django.contrib.messages.test.MessagesTestMixin.assertMessages().
Diffstat (limited to 'django')
-rw-r--r--django/contrib/messages/test.py8
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)