chore: testing (#8)

drf_passwordless_jwt/tests.py

@@ -0,0 +1,66 @@
+import os
+from unittest.mock import patch
+
+from django.core import mail
+from django.test import override_settings
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+
+class TaskTest(APITestCase):
+    @override_settings(EMAIL_BACKEND="django.core.mail.backends.locmem.EmailBackend")
+    def test_token_email(self):
+        response = self.client.post(
+            reverse("auth_email_token"),
+            {"email": "[email protected]"},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(
+            response.json(),
+            {"detail": "A login token has been sent to your email."},
+        )
+        self.assertEqual(len(mail.outbox), 1)
+        msg = mail.outbox[0]
+        self.assertEqual(msg.subject, 'Your Login Token')
+        self.assertTrue(msg.body.startswith('Enter this token to sign in:'))
+        token = msg.body.split()[-1]
+        self.assertEqual(len(token), 6)
+        self.assertTrue(token.isdigit())
+        self.assertEqual(msg.from_email, '[email protected]')
+        self.assertEqual(msg.to, ['[email protected]'])
+
+    @patch.dict(os.environ, {"EMAIL_TEST_ACCOUNT_a_at_a_com": "123456"})
+    def test_auth_jwt_token(self):
+        response = self.client.post(
+            reverse("auth_jwt_token"),
+            {"email": "[email protected]", "token": "123456"},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        json = response.json()
+        self.assertEqual(list(json.keys()), ['email', 'token'])
+        self.assertEqual(json['email'], '[email protected]')
+
+    @patch.dict(os.environ, {"EMAIL_TEST_ACCOUNT_a_at_a_com": "123456"})
+    def test_verify_jwt_token(self):
+        response = self.client.post(
+            reverse("auth_jwt_token"),
+            {"email": "[email protected]", "token": "123456"},
+            format="json",
+        )
+        token = response.json()['token']
+
+        response = self.client.post(
+            reverse("verify_jwt_token"),
+            {"token": token},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        json = response.json()
+        self.assertEqual(list(json.keys()), ['email', 'exp'])
+        self.assertEqual(json['email'], '[email protected]')

drf_passwordless_jwt/views.py

@@ -55,8 +55,8 @@ class VerifyJWTView(APIView):
     serializer_class = JWTSerializer
 
     def post(self, request, *args, **kwargs):
-        email = request.data['email']
-        if exists_test_account(email):
+        email = request.data.get('email')
+        if email and exists_test_account(email):
             return Response({
                 'email': email,
                 'exp': '9999-12-31T23:59:59',