def test_choice_short_name_single_choice(self): # make choices choice1 = Choice(title='yes', short_name='y') # make a question question = Question( title='is this a test', allow_multiple_choice=True, show_results=False) self.polls_index.add_child(instance=question) question.add_child(instance=choice1) question.save_revision().publish() # make a vote client = Client() client.login(username='******', password='******') client.post(reverse('molo.polls:vote', kwargs={'question_id': question.id}), {'choice': choice1.id}) # should automatically create the poll vote # test poll vote response = download_as_csv(QuestionAdmin(Question, self.site), None, Question.objects.all()) date = str(datetime.datetime.now().date()) expected_output = ('Content-Type: text/csv\r\nContent-Disposition:' ' attachment;filename=questions-' + date + '.csv\r\n\r\n' 'title,date_submitted,user,answer' '\r\nis this a test,' + date + ',superuser,' 'y\r\n') self.assertEquals(str(response), expected_output)
def test_download_csv_free_text_question_short_name(self): question = FreeTextQuestion( title='is this a test', short_name='short') self.polls_index.add_child(instance=question) question.save_revision().publish() client = Client() client.login(username='******', password='******') response = client.get('/') self.assertContains(response, 'is this a test') client.post(reverse('molo.polls:free_text_vote', kwargs={'question_id': question.id}), {'answer': 'this is an answer'}) response = download_as_csv(QuestionAdmin(Question, self.site), None, Question.objects.all()) date = str(datetime.datetime.now().date()) expected_output = ('Content-Type: text/csv\r\nContent-Disposition:' ' attachment;filename=questions-' + date + '.csv\r\n\r\n' 'title,date_submitted,user,answer' '\r\nshort,' + date + ',superuser,' 'this is an answer\r\n') self.assertEquals(str(response), expected_output)