feat: add tests command

src/infrastructure/database/unit_of_work.py

@@ -1,10 +1,10 @@
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
 from src.application.abstractions import IUnitOfWork
-from src.application.abstractions.repositories import IOrderRepository,IPaymentRepository
+from src.application.abstractions.repositories import IOrderRepository,IPaymentRepository,IUserRepository
 from src.application.contracts import ILogger
 from src.infrastructure.database.repositories.order_repository import OrderRepository
 from src.infrastructure.database.repositories.payment_repository import PaymentRepository
-# from src.infrastructure.database.repositories import UserRepository, SessionRepository
+from src.infrastructure.database.repositories.user_repository import UserRepository
 
 
 
@@ -14,14 +14,14 @@ class UnitOfWork(IUnitOfWork):
         self._session: AsyncSession = None
         self._order_repository: IOrderRepository | None = None
         self._payment_repository: IPaymentRepository | None = None
-        # self._user_repository: IUserRepository = None
-        # self._session_repository: ISessionRepository = None
+        self._user_repository: IUserRepository | None = None
         self._logger: ILogger = logger
 
     async def __aenter__(self):
         self._session = self.session_factory()
         self._order_repository = None
         self._payment_repository = None
+        self._user_repository = None
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -48,14 +48,9 @@ class UnitOfWork(IUnitOfWork):
             self._payment_repository = PaymentRepository(session=self._session, logger=self._logger)
         return self._payment_repository
 
-    # @property
-    # def user_repository(self) -> IUserRepository:
-    #     if self._user_repository is None:
-    #         self._user_repository = UserRepository(session=self._session, logger=self._logger)
-    #     return self._user_repository
-    #
-    # @property
-    # def session_repository(self) -> ISessionRepository:
-    #     if self._session_repository is None:
-    #         self._session_repository = SessionRepository(session=self._session, logger=self._logger)
-    #     return self._session_repository
+
+    @property
+    def user_repository(self) -> IUserRepository:
+        if self._user_repository is None:
+            self._user_repository = UserRepository(session=self._session, logger=self._logger)
+        return self._user_repository