integration tests

src/database/auth/refresh_tokens.py

@@ -2,8 +2,10 @@
 from uuid import UUID
 
 from sqlalchemy import and_, not_, select
+from sqlalchemy.exc import NoResultFound
 from sqlalchemy.orm import sessionmaker
 
+from src.errors.http_errors.errors import Errors
 from src.models.database_models.model import RefreshTokens, engine
 from src.models.pydantic_models.model import RefreshTokensOut
 
@@ -11,6 +13,7 @@ from src.models.pydantic_models.model import RefreshTokensOut
 class JwtCrudActions:
     def __init__(self) -> None:
         self.Session=sessionmaker(bind=engine)
+        self.error=Errors()
     
     def get_token_by_user_id(self, user_id:UUID)->RefreshTokensOut|None:
         with self.Session() as session:  # noqa: SIM117
@@ -45,6 +48,23 @@ class JwtCrudActions:
                 response.is_revoked=True
                 response.replaced_by=new_jti
                 return True
+            
+    def create_and_update_token(self,data:dict, old_jti:UUID, new_jti:UUID)->bool:
+        with self.Session() as session: #noqa:SIM117
+            with session.begin():
+                new_token=RefreshTokens(**data)
+                session.add(new_token)
+                query=select(RefreshTokens).where(RefreshTokens.id==old_jti)
+                
+                try:
+                    response = session.scalars(query).one()
+                except NoResultFound as e:
+                    raise self.error.not_found_error(detail="Token not found") from e
+                
+                response.is_revoked=True
+                response.replaced_by=new_jti
+                return True
+                
     
     def revoke_all(self, user_id:UUID)->bool:
         with self.Session() as session: # noqa: SIM117