Fix 302 redirect in test cause of login redirect

This commit is contained in:
sosokker 2023-09-12 14:01:07 +07:00
parent 953e6736ce
commit c2ab9ee760
2 changed files with 8 additions and 2 deletions

View File

@ -3,6 +3,7 @@ import datetime
from django.test import TestCase from django.test import TestCase
from django.utils import timezone from django.utils import timezone
from django.urls import reverse from django.urls import reverse
from django.contrib.auth.models import User
from .models import Question from .models import Question
@ -199,6 +200,9 @@ class QuestionDetailViewTests(TestCase):
future_question.pub_date = timezone.now() + timezone.timedelta(days=5) future_question.pub_date = timezone.now() + timezone.timedelta(days=5)
future_question.save() future_question.save()
user = User.objects.create_user(username="testcase", password="123test123")
self.client.login(username="testcase", password="123test123")
url = reverse("polls:detail", args=(future_question.id,)) url = reverse("polls:detail", args=(future_question.id,))
response = self.client.get(url) response = self.client.get(url)
self.assertEqual(response.status_code, 404) self.assertEqual(response.status_code, 404)
@ -212,6 +216,9 @@ class QuestionDetailViewTests(TestCase):
past_question.pub_date = timezone.now() - timezone.timedelta(days=5) past_question.pub_date = timezone.now() - timezone.timedelta(days=5)
past_question.save() past_question.save()
user = User.objects.create_user(username="testcase", password="123test123")
self.client.login(username="testcase", password="123test123")
url = reverse("polls:detail", args=(past_question.id,)) url = reverse("polls:detail", args=(past_question.id,))
response = self.client.get(url) response = self.client.get(url)
self.assertContains(response, past_question.question_text) self.assertContains(response, past_question.question_text)

View File

@ -121,4 +121,3 @@ def vote(request, question_id):
else: else:
messages.error(request, "You can not vote on this question.") messages.error(request, "You can not vote on this question.")
return HttpResponseRedirect(reverse("polls:index")) return HttpResponseRedirect(reverse("polls:index"))
return HttpResponseRedirect(reverse("polls:results", args=(question.id,)))