def test_redirects_to_project_if_done_message_is_not_set(self):
form = mommy.make(Form, project=self.project,
slug='form-1', published=True)
self.request._messages._store([self.misc_message])
response = response_done(self.request, form.project, form.slug)
self.assertEqual(response.status_code, 302)
self.assertEqual(response.url, settings.PROJECTS[form.project]['root'])
def test_template_is_rendered_if_done_message_is_set(self):
form = mommy.make(Form, project=self.project,
slug='form-1', published=True)
self.request._messages._store([self.done_message])
response = response_done(self.request, form.project, form.slug)
self.assertEqual(response.status_code, 200)
self.assertContains(response, 'success')
self.assertEqual(len(self.request._messages._get()[0]), 0)
