示例#1
0
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])