Add middleware/permssion/cache helper test codes

Author: teamhide

.pre-commit-config.yaml

@@ -28,6 +28,7 @@ repos:
         exclude: migrations/
         additional_dependencies:
             [
+                "redis==4.5.1",
                 "types-redis==4.6.0",
                 "types-ujson==5.9.0",
             ]

core/fastapi/dependencies/permission.py

@@ -21,7 +21,7 @@ class BasePermission(ABC):
 
     @abstractmethod
     async def has_permission(self, request: Request) -> bool:
-        pass
+        """has permssion"""
 
 
 class IsAuthenticated(BasePermission):

core/helpers/cache/base/backend.py

@@ -5,12 +5,12 @@
 class BaseBackend(ABC):
     @abstractmethod
     async def get(self, *, key: str) -> Any:
-        ...
+        """Get"""
 
     @abstractmethod
     async def set(self, *, response: Any, key: str, ttl: int = 60) -> None:
-        ...
+        """Set"""
 
     @abstractmethod
     async def delete_startswith(self, *, value: str) -> None:
-        ...
+        """Delete starts with"""

core/helpers/cache/base/key_maker.py

@@ -5,4 +5,4 @@
 class BaseKeyMaker(ABC):
     @abstractmethod
     async def make(self, *, function: Callable, prefix: str) -> str:
-        ...
+        """Base key maker"""

core/helpers/cache/redis_backend.py

@@ -4,28 +4,25 @@
 import ujson
 
 from core.helpers.cache.base import BaseBackend
-from core.helpers.redis import redis
+from core.helpers.redis import redis_client
 
 
 class RedisBackend(BaseBackend):
     async def get(self, *, key: str) -> Any:
-        result = await redis.get(key)
+        result = await redis_client.get(key)
         if not result:
             return
 
-        try:
-            return ujson.loads(result.decode("utf8"))
-        except UnicodeDecodeError:
-            return pickle.loads(result)
+        return ujson.loads(result)
 
     async def set(self, *, response: Any, key: str, ttl: int = 60) -> None:
         if isinstance(response, dict):
             response = ujson.dumps(response)
-        elif isinstance(response, object):
+        else:
             response = pickle.dumps(response)
 
-        await redis.set(name=key, value=response, ex=ttl)
+        await redis_client.set(name=key, value=response, ex=ttl)
 
     async def delete_startswith(self, *, value: str) -> None:
-        async for key in redis.scan_iter(f"{value}::*"):
-            await redis.delete(key)
+        async for key in redis_client.scan_iter(f"{value}*"):
+            await redis_client.delete(key)

core/helpers/redis.py

@@ -1,5 +1,5 @@
-import redis.asyncio as aioredis
+from redis import asyncio as redis
 
 from core.config import config
 
-redis = aioredis.from_url(url=f"redis://{config.REDIS_HOST}")
+redis_client = redis.from_url(url=f"redis://{config.REDIS_HOST}", decode_responses=True)

poetry.lock

@@ -29,6 +29,7 @@ httpx = "^0.26.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.4"
+types-redis = "^4.6.0.20240106"
 
 [tool.coverage.run]
 omit=["tests/*", "test_*.py", "migrations/*"]