- test: add test for edit user

Author: koladev32

api/routers.py

@@ -4,7 +4,7 @@
 
 router = routers.SimpleRouter(trailing_slash=False)
 
-router.register(r'edit', UserViewSet, basename='users')
+router.register(r'edit', UserViewSet, basename='user-edit')
 
 router.register(r'register', RegisterViewSet, basename='register')
 

api/user/tests.py

@@ -1,7 +1,40 @@
+from django.urls import reverse
 from rest_framework.test import APITestCase
+from rest_framework import status
 
 
 class UserViewSetTest(APITestCase):
+    base_edit_url = reverse('api:user-edit-list')
+    base_url_login = reverse("api:login-list")
+
+
+    data_login = {
+        "password": "12345678",
+        "email": "[email protected]"
+    }
 
     def test_edit(self):
-        pass
+        
+        # Login to retrieve token
+
+        response = self.client.post(f"{self.base_url_login}", data=self.data_login)
+        response_data = response.json()
+
+        token = response_data['token']
+        user_id = response_data['user']["_id"]
+
+        self.client.credentials(HTTP_AUTHORIZATION=token)
+
+        # Edit user
+        
+        data = {
+            "email": "[email protected]",
+            "userID": user_id,
+        }
+        
+        response = self.client.post(f"{self.base_edit_url}", data=data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+                
+        response_data = response.json()
+        
+        self.assertEqual(response_data["success"], True)

api/user/viewsets.py

@@ -34,8 +34,9 @@ def create(self, request, *args, **kwargs):
         if not user_id:
             raise ValidationError(self.error_message)
 
-        if self.request.user.pk != user_id and not self.request.user.is_superuser:
+        if self.request.user.pk != int(user_id) and not self.request.user.is_superuser:
             raise ValidationError(self.error_message)
+        
         self.update(request)
 
         return Response({