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))