add users testing

Author: RustamovAkrom

apps/users/models.py

@@ -4,7 +4,7 @@
 from django.db import models
 
 
-class User(AbstractUser, TimestempedAbstractModel):
+class User(TimestempedAbstractModel, AbstractUser):
     email = models.EmailField(_("email address"), unique=True)
 
     @property

apps/users/tests.py

@@ -1,15 +1,23 @@
 from django.test import TestCase
+from django.urls import reverse
 
 from .models import User, UserProfile
 
 
 class TestUsers(TestCase):
-    def setUp(self) -> None:
-        User.objects.create(
-            username="User",
-            email="user@example.com",
-            password="password"
-        )
-        return super().setUp()
-    
-    
+
+    def test_user_register_page(self):
+        response = self.client.get(reverse("users:register"))
+
+        self.assertEqual(response.status_code, 200)
+        self.assertTemplateUsed(response, "auth/register.html")
+        self.assertContains(response, "Register")
+        # self.assertRedirects(response, reverse("users:login"))
+
+    def test_user_login_page(self):
+        response = self.client.get(reverse("users:login"))
+
+        self.assertEqual(response.status_code, 200)
+        self.assertTemplateUsed(response, "auth/login.html")
+        self.assertContains(response, "Login")
+        # self.assertRedirects(response, reverse("blog:home"))

apps/users/views.py

@@ -1,5 +1,5 @@
 from django.shortcuts import render, redirect, get_object_or_404
-from django.urls import reverse
+from django.urls import reverse, reverse_lazy
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.contrib import messages
 from django.views import View
@@ -31,11 +31,11 @@ def post(self, request):
         if form.is_valid():
             form.save()
             messages.success(request, "User succesfully registered")
-            return redirect(reverse("users:login"))
+            return redirect(reverse_lazy("users:login"))
 
         messages.warning(request, "Invalid registration fields!")
-        return render(request, "auth/register.html", {"form": form})
-
+        return redirect(reverse_lazy("auth:register"))
+    
 
 class LoginPageView(CustomHtmxMixin, View):
     template_name = "auth/login.html"
@@ -59,19 +59,18 @@ def post(self, request):
 
             if user is not None:
 
-                response = redirect(reverse("blog:home"))
+                response = redirect(reverse_lazy("blog:home"))
 
                 # Login for jwt
                 response = get_jwt_login_response(response, user)
 
                 messages.success(request, f"You are logged in as { username }")
                 return response
 
-            else:
-                messages.error(request, "Invalid username or password.")
-                return redirect(reverse("users:login"))
+            messages.error(request, "Invalid username or password.")
 
-        return render(request, "auth/login.html", {"form": form})
+        return redirect(reverse_lazy("users:login"))
+        
 
 
 class LogoutPageView(CustomHtmxMixin, LoginRequiredMixin, View):