django/tests/messages_tests/test_mixins.py

15 lines
535 B
Python
Raw Normal View History

from django.test import SimpleTestCase, override_settings
from django.urls import reverse
from .urls import ContactFormViewWithMsg
2013-11-03 04:12:09 +08:00
@override_settings(ROOT_URLCONF='messages_tests.urls')
class SuccessMessageMixinTests(SimpleTestCase):
def test_set_messages_success(self):
2017-01-05 22:23:39 +08:00
author = {'name': 'John Doe', 'slug': 'success-msg'}
add_url = reverse('add_success_msg')
req = self.client.post(add_url, author)
2017-01-05 22:23:39 +08:00
self.assertIn(ContactFormViewWithMsg.success_message % author, req.cookies['messages'].value)