mirror of
https://github.com/HamzaSha1/zod-backend.git
synced 2026-10-23 23:37:56 +00:00
100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
"""
|
|
web admin test auth file
|
|
"""
|
|
from datetime import datetime
|
|
from django.utils import timezone
|
|
from django.urls import reverse
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from rest_framework.test import APITestCase, APIClient
|
|
from rest_framework import status
|
|
|
|
from account.models import UserEmailOtp
|
|
from base.constants import USER_TYPE
|
|
from guardian.tasks import generate_otp
|
|
from web_admin.tests.test_set_up import BaseSetUp
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class AdminOTPTestCase(BaseSetUp):
|
|
"""
|
|
|
|
"""
|
|
|
|
def setUp(self):
|
|
"""
|
|
inherit data here
|
|
:return:
|
|
"""
|
|
super(AdminOTPTestCase, self).setUp()
|
|
|
|
def test_admin_otp_for_valid_email(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
url = reverse('web_admin:admin-otp')
|
|
data = {
|
|
'email': '[email protected]'
|
|
}
|
|
response = self.client.post(url, data, format='json')
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(UserEmailOtp.objects.count(), 1)
|
|
|
|
def test_admin_otp_for_invalid_email(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
url = reverse('web_admin:admin-otp')
|
|
data = {
|
|
'email': '[email protected]'
|
|
}
|
|
response = self.client.post(url, data, format='json')
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class AdminVerifyOTPTestCase(BaseSetUp):
|
|
"""
|
|
|
|
"""
|
|
|
|
def setUp(self):
|
|
"""
|
|
inherit data here
|
|
:return:
|
|
"""
|
|
super(AdminVerifyOTPTestCase, self).setUp()
|
|
self.verification_code = generate_otp()
|
|
expiry = timezone.now() + timezone.timedelta(days=1)
|
|
self.user_email_otp = UserEmailOtp.objects.create(email='[email protected]',
|
|
otp=self.verification_code,
|
|
expired_at=expiry,
|
|
user_type=dict(USER_TYPE).get('3'),
|
|
)
|
|
|
|
def test_admin_verify_otp_with_valid_otp(self):
|
|
url = reverse('web_admin:admin-verify-otp')
|
|
data = {
|
|
'email': '[email protected]',
|
|
"otp": self.verification_code
|
|
}
|
|
|
|
response = self.client.post(url, data)
|
|
self.user_email_otp.refresh_from_db()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(self.user_email_otp.is_verified, True)
|
|
|
|
def test_admin_verify_otp_with_invalid_otp(self):
|
|
url = reverse('web_admin:admin-verify-otp')
|
|
data = {
|
|
'email': '[email protected]',
|
|
"otp": generate_otp()
|
|
}
|
|
|
|
response = self.client.post(url, data)
|
|
self.user_email_otp.refresh_from_db()
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
self.assertEqual(self.user_email_otp.is_verified, False)
|