test.py 332 B

12345678
  1. from .api import get_messages
  2. class MessagesTestMixin:
  3. def assertMessages(self, response, expected_messages, *, ordered=True):
  4. request_messages = list(get_messages(response.wsgi_request))
  5. assertion = self.assertEqual if ordered else self.assertCountEqual
  6. assertion(request_messages, expected_messages)