diff options
Diffstat (limited to 'tests/messages_tests/test_mixins.py')
-rw-r--r-- | tests/messages_tests/test_mixins.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/messages_tests/test_mixins.py b/tests/messages_tests/test_mixins.py index e102627089..051ed82b6b 100644 --- a/tests/messages_tests/test_mixins.py +++ b/tests/messages_tests/test_mixins.py @@ -1,3 +1,4 @@ +from django.core.signing import b64_decode from django.test import SimpleTestCase, override_settings from django.urls import reverse @@ -11,4 +12,8 @@ class SuccessMessageMixinTests(SimpleTestCase): author = {'name': 'John Doe', 'slug': 'success-msg'} add_url = reverse('add_success_msg') req = self.client.post(add_url, author) - self.assertIn(ContactFormViewWithMsg.success_message % author, req.cookies['messages'].value) + # Uncompressed message is stored in the cookie. + value = b64_decode( + req.cookies['messages'].value.split(":")[0].encode(), + ).decode() + self.assertIn(ContactFormViewWithMsg.success_message % author, value) |