Я сейчас пытаюсь написать свой тест для успешного сброса пароля.
TLDR: Я не могу понять, как обработать ввод данных формы в запрос на публикацию в представлении set-password.
Для контекста, позвольте мне сначала дать вам мой код, поскольку строка кода говорит более 1000 слов.
счета / апи / views.py:
class PasswordResetView(generics.GenericAPIView):
"""(POST) Expects email. Sends a password reset mail to email."""
permission_classes = (permissions.AllowAny,)
serializer_class = PasswordResetSerializer
def post(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
if serializer.is_valid(raise_exception=True):
user = User.objects.get(email=serializer.data['email'])
if user.is_active and user.has_usable_password():
send_password_reset_email(user, request)
return Response(
{"detail": "A password reset email has been send."},
status=status.HTTP_204_NO_CONTENT
)
else:
return Response(
{"detail": "This users password can't be reset."},
status=status.HTTP_406_NOT_ACCEPTABLE
)
счета / апи / urls.py:
url(r'^password_reset/$', views.PasswordResetView.as_view(),
name="password_reset"),
счета / апи / serializers.py:
class PasswordResetSerializer(serializers.Serializer):
email = serializers.EmailField()
def validate_email(self, value):
if User.objects.filter(email__iexact=value).exists():
return value
else:
raise serializers.ValidationError("Email not found.")
счета / utils.py:
from django.contrib.auth.tokens import PasswordResetTokenGenerator
from django.contrib.sites.shortcuts import get_current_site
from django.utils import six
from django.utils.encoding import force_bytes
from django.utils.http import urlsafe_base64_encode
from templated_email import send_templated_mail
password_reset_token = PasswordResetTokenGenerator()
def build_email_context(user_obj, request):
current_site = get_current_site(request)
context = {
"name": "",
"domain": current_site.domain,
"uid": urlsafe_base64_encode(force_bytes(user_obj.pk)),
'token': password_reset_token.make_token(user_obj)
}
if user_obj.name != "":
context["name"]: " " + user_obj.name
return context
def send_password_reset_email(user_obj, request):
context = build_email_context(user_obj, request)
send_templated_mail(
template_name="password_reset",
from_email="noreply@ordersome.com",
recipient_list=[user_obj.email],
context=context
)
Представления, связанные с маршрутом в сообщении электронной почты, представляют собой стандартные CBV PasswordResetConfirmView и PasswordResetDoneView из django.contrib.auth.
И вот тест, который я написал для этой последовательности сброса пароля:
апи / тесты / test_views.py:
from django.contrib.auth import get_user, get_user_model
from django.core import mail
from rest_framework import status
from rest_framework.authtoken.models import Token
from rest_framework.reverse import reverse as api_reverse
from rest_framework.test import APITestCase
User = get_user_model()
class UserAPIViewsTestCase(APITestCase):
def setUp(self):
User.objects.create_user(
email="testuser@test.com", password="test1234")
self.login_url = api_reverse("api:auth:login")
self.register_url = api_reverse("api:auth:register")
self.password_reset_url = api_reverse("api:auth:password_reset")
self.password_change_url = api_reverse("api:auth:password_change")
self.user_delete_url = api_reverse("api:auth:user_delete")
def test_api_auth_password_reset_success(self):
"""Test if a password reset successfully resets password."""
pre_user = User.objects.get(email="testuser@test.com")
self.assertEqual(pre_user.check_password("test1234"), True)
email_data = {
"email": "testuser@test.com",
}
email_response = self.client.post(
self.password_reset_url, email_data, format="json")
self.assertEqual(email_response.status_code,
status.HTTP_204_NO_CONTENT)
password_reset_data = {
"new_password1": "newtest1234",
"new_password2": "newtest1234"
}
password_reset_response = self.client.get(
mail.outbox[0].body[-94:-37], format="json", follow=True)
path = password_reset_response.request["PATH_INFO"]
done_url = "http://testserver" + path
password_reset_done_response = self.client.post(
done_url, password_reset_data, format="json")
post_user = User.objects.get(email="testuser@test.com")
self.assertEqual(post_user.check_password("newtest1234"), True)
Последний self.assertEqual терпит неудачу. Но когда я проверяю представление вручную, оно работает, поэтому код должен быть правильным. Как мне изменить тест?
Когда я выхожу из системы done_url, это дает мне http://testserver/auth/reset/MQ/set-password/, который должен быть правильным путем.
И когда я выхожу из системы password_reset_done_response, он говорит, что это TemplateResponse и status_code равен 200 (что должно означать успешное, но это не так). Кроме того, этот ответ, похоже, все еще содержит путь /auth/reset/MQ/set-password/, который больше не должен иметь место.