Implement exchange imports and frontend reporting APIs

tests/apps/user/factories.py

@@ -1,9 +1,10 @@
 """Фабрики для создания тестовых объектов с использованием factory_boy и faker"""
 
 import factory
-from apps.user.models import Profile, User
 from faker import Faker
 
+from apps.user.models import Profile, User
+
 fake = Faker("ru_RU")
 
 

tests/apps/user/test_serializers.py

@@ -1,5 +1,9 @@
 """Tests for user serializers"""
 
+from django.contrib.auth import get_user_model
+from django.test import TestCase
+from faker import Faker
+
 from apps.user.serializers import (
     LoginSerializer,
     PasswordChangeSerializer,
@@ -9,9 +13,6 @@ from apps.user.serializers import (
     UserSerializer,
     UserUpdateSerializer,
 )
-from django.contrib.auth import get_user_model
-from django.test import TestCase
-from faker import Faker
 
 from .factories import ProfileFactory, UserFactory
 
@@ -183,7 +184,14 @@ class ProfileUpdateSerializerTest(TestCase):
     def test_fields_allowed(self):
         """Test only allowed fields can be updated"""
         serializer = ProfileUpdateSerializer()
-        allowed_fields = ["first_name", "mid_name", "last_name", "bio", "avatar", "date_of_birth"]
+        allowed_fields = [
+            "first_name",
+            "mid_name",
+            "last_name",
+            "bio",
+            "avatar",
+            "date_of_birth",
+        ]
 
         self.assertEqual(set(serializer.Meta.fields), set(allowed_fields))
 

tests/apps/user/test_services.py

@@ -1,12 +1,13 @@
 """Tests for user services"""
 
-from apps.core.exceptions import NotFoundError
-from apps.user.services import ProfileService, UserService
 from django.contrib.auth import get_user_model
 from django.test import TestCase
 from faker import Faker
 from rest_framework_simplejwt.tokens import RefreshToken
 
+from apps.core.exceptions import NotFoundError
+from apps.user.services import ProfileService, UserService
+
 from .factories import ProfileFactory, UserFactory
 
 User = get_user_model()

tests/apps/user/test_views.py

@@ -1,13 +1,14 @@
 """Tests for user DRF views"""
 
-from apps.user.models import Profile
-from apps.user.services import UserService
 from django.contrib.auth import get_user_model
 from django.urls import reverse
 from faker import Faker
 from rest_framework import status
 from rest_framework.test import APITestCase
 
+from apps.user.models import Profile
+from apps.user.services import UserService
+
 from .factories import ProfileFactory, UserFactory
 
 User = get_user_model()
@@ -136,6 +137,18 @@ class CurrentUserViewTest(APITestCase):
         self.assertEqual(response.data["id"], self.user.id)
         self.assertEqual(response.data["email"], self.user.email)
         self.assertIn("profile", response.data)
+        self.assertEqual(response.data["role"], "user")
+        self.assertEqual(response.data["capabilities"]["can_access_admin_page"], False)
+
+    def test_get_current_user_returns_admin_role_for_staff_user(self):
+        self.user.is_staff = True
+        self.user.save(update_fields=["is_staff"])
+
+        response = self.client.get(self.current_user_url)
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data["role"], "staff")
+        self.assertEqual(response.data["capabilities"]["can_access_admin_page"], True)
 
     def test_get_current_user_unauthenticated(self):
         """Test getting current user when unauthenticated"""