From 859d26d07364d9fbbd233f44601d14580894cac7 Mon Sep 17 00:00:00 2001
From: abutalib-kiwi <abu.talib@kiwitech.com>
Date: Fri, 8 Sep 2023 19:07:06 +0530
Subject: [PATCH] test cases for web admin auth

---
 celerybeat-schedule            | Bin 20480 -> 20480 bytes
 web_admin/tests/test_auth.py   |  99 +++++++++++++++++++++++++++++++++
 web_admin/tests/test_set_up.py |  22 ++++++--
 3 files changed, 116 insertions(+), 5 deletions(-)
 create mode 100644 web_admin/tests/test_auth.py

diff --git a/celerybeat-schedule b/celerybeat-schedule
index be892edea3778b38a0afe5c729adddc781b59706..7638e5252440a76df3a3df6b009f2805ccced487 100644
GIT binary patch
delta 74
zcmZozz}T>Wal>2(b`B0XMov@S$qUQ_8QCWr*l13ccT5)mi+0sa$q;Ls5;Vn|ck(pH
VVgU{=1_s9DYM9`H&5};fc>xRT7ApV%

delta 74
zcmZozz}T>Wal>2(c2*ufMZVII$qUQ_8QCTq*l13ccT5)mi_SeaB}1%jO3)N<p2^c3
Siy?xGdtid|HcL7^=LG;tI~U6U

diff --git a/web_admin/tests/test_auth.py b/web_admin/tests/test_auth.py
new file mode 100644
index 0000000..6033af1
--- /dev/null
+++ b/web_admin/tests/test_auth.py
@@ -0,0 +1,99 @@
+"""
+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': 'admin@example.com'
+        }
+        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': 'notadmin@example.com'
+        }
+        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='admin@example.com',
+                                                          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': 'admin@example.com',
+            "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': 'admin@example.com',
+            "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)
diff --git a/web_admin/tests/test_set_up.py b/web_admin/tests/test_set_up.py
index aa6a664..f5cf1f6 100644
--- a/web_admin/tests/test_set_up.py
+++ b/web_admin/tests/test_set_up.py
@@ -103,7 +103,22 @@ base64_image = ("data:image/jpeg;base64,/9j/4AAQSkZJRgABAQAAAQABAAD/2wCEAAoHCBIS
                 "GYdEC/arYms/HtkfAS7huhuDXXJmPEkn5whN4xrZ0NkYdPRpIPdLS0kG5QPdCEskKlYcIWHCEJUO3KFuUIQD3QhCB//Z")
 
 
-class ArticleTestSetUp(APITestCase):
+class BaseSetUp(APITestCase):
+    """
+
+    """
+    def setUp(self) -> None:
+        """
+
+        :return:
+        """
+        self.client = APIClient()
+        self.user = User.objects.create_user(username='user@example.com', password='user@1234')
+        self.admin_user = User.objects.create_user(username='admin@example.com', email='admin@example.com',
+                                                   password='admin@1234', is_staff=True, is_superuser=True)
+
+
+class ArticleTestSetUp(BaseSetUp):
     """
     test cases data set up for article create, update
     """
@@ -112,10 +127,7 @@ class ArticleTestSetUp(APITestCase):
         set up data for test
         :return:
         """
-        self.client = APIClient()
-        self.user = User.objects.create_user(username='user@example.com', password='user@1234')
-        self.admin_user = User.objects.create_user(username='admin@example.com', email='admin@example.com',
-                                                   password='admin@1234', is_staff=True, is_superuser=True)
+        super(ArticleTestSetUp, self).setUp()
 
         self.article = Article.objects.create(title="Existing Article", description="Existing Description",
                                               is_published=True)