fix: change for tests

Author: RustamovAkrom

apps/users/api_endpoints/__init__.py

@@ -4,5 +4,5 @@
 
 
 router = routers.DefaultRouter()
-router.register("user", UserViewSet)
-router.register("user_profile", UserProfileViewSet)
+router.register("user", UserViewSet, basename="user")
+router.register("user_profile", UserProfileViewSet, basename="user_profile")

apps/users/api_endpoints/users/User/serializers.py

@@ -31,7 +31,6 @@ class Meta:
             "is_active", 
             "is_superuser",
             "is_staff",
-            "post_count",
             "profiles",
             "created_at",
             "updated_at",

apps/users/api_endpoints/users/User/tests.py

@@ -0,0 +1,70 @@
+from rest_framework.test import APITestCase
+from rest_framework_simplejwt.tokens import RefreshToken, AccessToken
+
+from django.urls import reverse
+
+from apps.users.models import User
+
+# Authorizations
+# self.client.credentials(HTTP_AUTHORIZATION=f"Bearer {self.token}")
+
+class UsersApiTestCase(APITestCase):
+    def setUp(self) -> None:
+        self.username = "Admin"
+        self.email = "admin@example.com"
+        self.password = "password"
+        
+        user = User.objects.create(
+            username=self.username,
+            email=self.email,
+        )
+        user.set_password(self.password)
+        user.save()
+        self.user = user
+        refresh = RefreshToken.for_user(self.user)
+        self.token = str(refresh.access_token)
+        return super().setUp()
+    
+    def test_api_tokens(self):
+        data = {
+            "username": "Admin",
+            "password": "password",
+        }
+        response = self.client.post(reverse("token_obtain_pair"), data=data)
+        tokens = dict(response.json())
+        refresh = tokens.get("refresh")
+        access = tokens.get("access")
+
+        # Test jwt access token
+        jwt_access = AccessToken(access)
+        self.assertEqual(self.user.id, jwt_access["user_id"])
+
+        # Test jwt refresh token
+        jwt_refresh = RefreshToken(refresh)
+        self.assertEqual(self.user.id, jwt_refresh.for_user(self.user).access_token["user_id"])
+
+        self.assertEqual(response.status_code, 200)
+
+    # def test_api_user_create(self):
+    #     url = "/api/v1/users/user/"
+    #     def check_user():
+    #         data = {
+    #             "username": "Admin2",
+    #             "email": "admin2@example.com",
+    #             "password": "password",
+    #             "password_confirm": "password"
+    #         }
+
+    #     invalid_data = {
+    #         "username": self.username,
+    #         "email": self.email,
+    #         "password": self.password,
+    #         "password_confirm": self.password
+    #     }
+        
+    #     response = self.client.post(url, data=valid_data)
+        
+
+    #     print(response.status_code)
+    #     print(response.json())
+

apps/users/api_endpoints/users/User/views.py

@@ -3,7 +3,7 @@
 
 from apps.users.models import User
 from .serializers import UserSerializer
-from .permissions import IsOwnerPermission
+from apps.users.permissions import IsOwnerPermission
 
 
 class UserViewSet(viewsets.ModelViewSet):

apps/users/api_endpoints/users/UserProfile/views.py

@@ -1,11 +1,16 @@
 from rest_framework import viewsets
+from rest_framework.permissions import IsAuthenticated, IsAdminUser
 
 from apps.users.models import UserProfile
+from apps.users.permissions import IsOwnerPermission
 from .serializers import UserProfileSerializer
 
 
 class UserProfileViewSet(viewsets.ModelViewSet):
     queryset = UserProfile.objects.all()
     serializer_class = UserProfileSerializer
 
+    def get_permissions(self):
+        return [IsAuthenticated(), IsAdminUser(), IsOwnerPermission()]
+
 __all__ = ("UserProfileViewSet", )

…/api_endpoints/users/User/permissions.py apps/users/permissions.pyapps/users/api_endpoints/users/User/permissions.py renamed to apps/users/permissions.py apps/users/api_endpoints/users/User/permissions.py renamed to apps/users/permissions.py

@@ -1,13 +1,7 @@
-from django.urls import path, include
+from django.urls import path
 
-from .api_endpoints import router
 from . import views
 
-from rest_framework_simplejwt.views import (
-    TokenObtainPairView,
-    TokenRefreshView
-)
-
 
 app_name = "users"