Users

backend/users/.env.prod

@@ -32,24 +32,4 @@ DJANGO_SUPERUSER_PASSWORD=password
 
 # ===== REDIS =====
 REDIS_LOCAL_HOST=redis
-REDIS_LOCAL_PORT=6379
-REDIS_LOCAL_PASSWORD=Qwe123!@#
-
-REDIS_SHARED_HOST=redis
-REDIS_SHARED_PORT=6379
-REDIS_SHARED_PASSWORD=Qwe123!@#
-
-REDIS_ACCESS_DB=0
-REDIS_REFRESH_DB=1
-REDIS_TOTP_DB=2
-REDIS_ACCESS_PREFIX=access
-REDIS_REFRESH_PREFIX=refresh
-REDIS_TOTP_PREFIX=totp
-
-# ===== TOTP =====
-EMAIL_EXCHANGE_NAME=email_exchange
-EMAIL_ROUTING_KEY=email_queue
-NOTIFY_EXCHANGE_NAME=notify_exchange
-NOTIFY_ROUTING_KEY=notify_queue
-TOTP_EXCHANGE_NAME=totp_exchange
-TOTP_ROUTING_KEY=totp_queue
+REDIS_LOCAL_PORT=6379

backend/users/backend/administrator/tests/test_get_jwt.py

@@ -1,6 +1,8 @@
 import os
 import time
 
+from social_core.pipeline import user
+
 from administrator.models import Admin
 from base.base_tests.tests import BaseViewTestCase
 from common.services.jwt.token import Token
@@ -21,7 +23,7 @@ def setUpTestData(cls):
         )
         cls.url = reverse("token_refresh")
         cls.token = cls.get_tokens(cls.user)
-        cls.data = {"refresh_token": cls.token.get("refresh")}
+        cls.data = {"refresh_token": cls.get_refresh_token(cls.user)}
 
     def client_post(self, data):
         return self.client.post(self.url, data=data, format="json")

backend/users/backend/base/base_tests/tests.py

@@ -1,10 +1,15 @@
 import pika
+
+from administrator.models import Admin
 from base.models import BaseAbstractUser
 from common.services.jwt.token import Token
 from django.conf import settings
 from django.test import TestCase
 from faker import Faker
 from rest_framework.test import APIClient
+
+from customer.models import CustomerUser
+from developer.models import CompanyUser
 from utils.broker.rabbitmq import RabbitMQ
 
 
@@ -30,6 +35,37 @@ def tearDown(self) -> None:
     def get_access_token(user: BaseAbstractUser) -> str:
         return Token().generate_access_token_for_user(user)
 
+    @staticmethod
+    def get_refresh_token(user: BaseAbstractUser) -> str:
+        return Token().generate_refresh_token_for_user(user)
+
     @staticmethod
     def get_tokens(user: BaseAbstractUser) -> dict:
         return Token().generate_tokens_for_user(user)
+
+    def get_user(self, user: BaseAbstractUser):
+
+        data_user = self.__generate_data_for_users(user)
+
+        if isinstance(user, CustomerUser):
+            customer_user = CustomerUser.objects.create_user(**data_user)
+            return customer_user
+        elif isinstance(user, Admin):
+            customer_user = Admin.objects.create_user(**data_user)
+            return customer_user
+        elif isinstance(user, CompanyUser):
+            customer_user = CompanyUser.objects.create_user(**data_user)
+            return customer_user
+        else:
+            raise ValueError(f"Type user {user.__class__.__name__} not found")
+
+    def __generate_data_for_users(self, type_user: BaseAbstractUser) -> dict:
+        data = {
+            "username": self.faker.word(),
+            "password": self.faker.word(),
+            "email": self.faker.email(),
+            "phone": self.faker.random_number(digits=10, fix_len=True),
+        }
+        if isinstance(type_user, CustomerUser):
+            data["birthday"] = self.faker.date_object()
+        return data

backend/users/backend/common/tests/test_logout.py

@@ -29,8 +29,8 @@ def create_user(cls, model: type[BaseAbstractUser]) -> BaseAbstractUser:
         return model.objects.create_user(**data)
 
     def create_refresh_token(self, user) -> dict:
-        refresh_token = self.get_tokens(user)
-        return {"refresh_token": refresh_token.get('refresh')}
+        refresh_token = self.get_refresh_token(user)
+        return {"refresh_token": refresh_token}
 
     def test_01_logout_developer(self):
         refresh_token = self.create_refresh_token(self.developer)

backend/users/backend/customer/tests/test_get_jwt.py

@@ -21,8 +21,8 @@ def setUpTestData(cls):
             is_active=True,
         )
         cls.url = reverse("token_refresh")
-        cls.token = cls.get_tokens(cls.user)
-        cls.data = {"refresh_token": cls.token.get("refresh")}
+        cls.token = cls.get_refresh_token(cls.user)
+        cls.data = {"refresh_token": cls.token}
 
     def client_post(self, data):
         return self.client.post(self.url, data=data, format="json")

backend/users/backend/developer/tests/test_get_jwt.py

@@ -21,7 +21,7 @@ def setUpTestData(cls):
         )
         cls.url = reverse("token_refresh")
         cls.token = cls.get_tokens(cls.user)
-        cls.data = {"refresh_token": cls.token.get("refresh")}
+        cls.data = {"refresh_token": cls.get_refresh_token(cls.user)}
 
     def client_post(self, data):
         return self.client.post(self.url, data=data, format="json")