def test_provides_csv_download_of_responses(self):
self._create_and_login_user(is_staff=True)
form = mommy.make(Form, number=1, version=2)
field = mommy.make(FormField, form=form, kind='short-answer')
mommy.make(FormFieldResponse,
form_response__form=form,
form_field=field,
details={'answer': 'ok'})
response = form_with_responses_csv(self.request, 1, 2)
self.assertIn('text/csv', response['Content-Type'])
self.assertIn('attachment; filename=', response['Content-Disposition'])
rows = list(csv.reader(StringIO(response.content)))
self.assertEqual(len(rows), 2)
self.assertEqual(len(rows[0]), 3)
self.assertEqual(len(rows[1]), 3)
self.assertEqual(rows[1][2], 'ok')
def test_returns_404_error_if_form_does_not_exist(self):
self._create_and_login_user(is_staff=True)
response = form_with_responses_csv(self.request, 1, 2)
self.assertEqual(response.status_code, 404)
self.assertIn('error', json.loads(response.content))
