27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
from django.core.signing import b64_decode
|
|
from django.test import TestCase, override_settings
|
|
from django.urls import reverse
|
|
|
|
from .models import SomeObject
|
|
from .urls import ContactFormViewWithMsg, DeleteFormViewWithMsg
|
|
|
|
|
|
@override_settings(ROOT_URLCONF='messages_tests.urls')
|
|
class SuccessMessageMixinTests(TestCase):
|
|
|
|
def test_set_messages_success(self):
|
|
author = {'name': 'John Doe', 'slug': 'success-msg'}
|
|
add_url = reverse('add_success_msg')
|
|
req = self.client.post(add_url, author)
|
|
# 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)
|
|
|
|
def test_set_messages_success_on_delete(self):
|
|
object_to_delete = SomeObject.objects.create(name='MyObject')
|
|
delete_url = reverse('success_msg_on_delete', args=[object_to_delete.pk])
|
|
response = self.client.post(delete_url, follow=True)
|
|
self.assertContains(response, DeleteFormViewWithMsg.success_message)
|