123456789101112131415 |
- from django.test import TestCase
- from django.contrib.messages.tests.urls import ContactFormViewWithMsg
- from django.core.urlresolvers import reverse
- class SuccessMessageMixinTests(TestCase):
- urls = 'django.contrib.messages.tests.urls'
- 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)
- self.assertIn(ContactFormViewWithMsg.success_message % author,
- req.cookies['messages'].value)
|