50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from rest_framework.test import APITestCase
|
|
from rest_framework import status
|
|
from authorization.tests.tests_authorization import get_tokens_for_user
|
|
from django.urls import reverse
|
|
from account.models import User
|
|
|
|
|
|
class AccountResetPassWordTests(APITestCase):
|
|
|
|
def setUp(self):
|
|
self.data = get_tokens_for_user()
|
|
|
|
def test_reset_password(self):
|
|
url = reverse('web:account:password-reset')
|
|
data = {
|
|
"username_or_email": self.data["email"]
|
|
}
|
|
response = self.client.post(url, data=data)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
data = {
|
|
"username_or_email": self.data["username"]
|
|
}
|
|
response = self.client.post(url, data=data)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
|
|
class AccountResetPassWordTests(APITestCase):
|
|
|
|
def setUp(self):
|
|
self.data = get_tokens_for_user()
|
|
|
|
def test_reset_password_confirm(self):
|
|
data ={
|
|
"password": "newpasswordnewpassword"
|
|
}
|
|
user = User.objects.get(email=self.data["email"])
|
|
token = user.reset_password_token
|
|
uidb64 = user.get_user_uidb64
|
|
|
|
url = reverse('web:account:password-reset-confirm', kwargs={
|
|
'uidb64': uidb64,
|
|
'token': token
|
|
})
|
|
response = self.client.patch(url, data=data)
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
|
|
|
|
|