summaryrefslogtreecommitdiff
path: root/tests/messages_tests/test_mixins.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/messages_tests/test_mixins.py')
-rw-r--r--tests/messages_tests/test_mixins.py7
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)