class EditPollTesting(TestCase): def setUp(self): self.usr = "******" self.a_user = UserFactory(username=self.usr) self.a_user.set_password(DEFAULT_PASSWORD) self.a_user.save() self.client.login(username=self.usr, password=DEFAULT_PASSWORD) self.a_question = 'a question?' self.poll = self.a_user.poll_set.create(question=self.a_question) self.a_choice = 'a choice' self.choice = self.poll.choice_set.create(choice=self.a_choice) def tearDown(self): """Delete the created Poll instance.""" self.client.logout() User.objects.all().delete() # Borra su poll y la choice def test_edit_poll_not_authenticated_redirects_to_login(self): """Not authenticated users trying edit_poll, must be redirected to login page.""" self.client.logout() # Make sure it's not logged-in original = reverse('polls:edit_poll', kwargs={'poll_id':self.poll.id}) response = self.client.post(original) destiny = "%s?next=%s"%(reverse('polls:login'), original) self.assertRedirects(response, destiny) def test_edit_not_owned_poll_fails(self): """An authenticated user cannot edit polls owned by someone else.""" assert(self.a_user.is_authenticated()) another_user = UserFactory(username="******") another_user.set_password(DEFAULT_PASSWORD) another_user.save() alien_poll = another_user.poll_set.create(question="whatever") # The logged in user tries to edit the alien_poll request = request_factory.post( reverse('polls:edit_poll', kwargs={'poll_id':alien_poll.id}), ) request.user = self.a_user with self.assertRaises(PermissionDenied): response = views.edit_poll(request, poll_id=alien_poll.id) def test_edit_poll_empty_question_shows_specific_message(self): """Try to edit_poll with an empty question, renders a specific msg""" # No extra data (such as the 'question' value) invalid_data = aux_initial_management_form() request = request_factory.post( reverse('polls:edit_poll', kwargs={'poll_id':self.poll.id}), data = invalid_data, ) request.user = self.a_user response = views.edit_poll(request) response.render() self.assertContains(response, html.escape(forms.EMPTY_QUESTION_MSG)) def test_edit_poll_with_duplicated_choices_fail(self): """POST to /<id>/edit_poll with duplicated choices, invalids formset.""" # No extra data (such as the 'question' value) choices_values = ['a', 'a'] invalid_data = aux_initial_management_form(total_forms=2, extra={ 'question': self.a_question, "choice_set-0-ORDER" : 1, "choice_set-0-choice" : choices_values[0], "choice_set-1-ORDER" : 2, "choice_set-1-choice" : choices_values[1], }) response = self.client.post( reverse('polls:edit_poll', kwargs={'poll_id':self.poll.id}), data = invalid_data ) self.assertFalse(response.context['choices_formset'].is_valid()) def test_edit_poll_choices_strip(self): """Choices value is stripped (no starting spaces).""" # No extra data (such as the 'question' value) extra = {'question': self.a_question} choices_values = [ #(choice id, user given value) (1 , u' a '), (None, u'\t b \t'), (None, u'\n c \n') ] desired_values = map(unicode.strip, [c for (i,c) in choices_values]) valid_data = formset_management_form(choices_values, extra=extra) #print valid_data request = request_factory.post( reverse('polls:edit_poll', kwargs={'poll_id':self.poll.id}), data = valid_data, ) request.user = self.a_user response = views.edit_poll(request, poll_id=self.poll.id) for i, db_choice in enumerate(Choice.objects.all()): self.assertEqual(db_choice.choice, desired_values[i])