15 lines
587 B
Python
15 lines
587 B
Python
|
from django.test.testcases 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)
|