12 lines
421 B
Python
12 lines
421 B
Python
|
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)
|