fix: command stats and response

pyle38/follower.py

@@ -97,7 +97,12 @@ async def server_extended(self) -> ServerStatsExtendedResponse:
         )
 
     async def stats(self, *keys: List[str]) -> StatsResponse:
-        return StatsResponse(**(await self.client.command(Command.STATS, *keys)))
+        response = await self.client.command(Command.STATS, *keys)
+
+        if response["stats"] == [None]:
+            response["stats"] = []
+
+        return StatsResponse(**response)
 
     async def quit(self) -> str:
 

pyle38/leader.py

@@ -17,8 +17,8 @@ async def delete(self, key: str, id: str) -> JSONResponse:
     async def drop(self, key: str) -> JSONResponse:
         return JSONResponse(**(await self.client.command(Command.DROP, [key])))
 
-    async def flushdb(self) -> dict:
-        return await self.client.command("FLUSHDB")
+    async def flushdb(self) -> JSONResponse:
+        return JSONResponse(**(await self.client.command("FLUSHDB")))
 
     async def expire(self, key: str, id: str, seconds: int) -> JSONResponse:
         return JSONResponse(

pyle38/responses.py

@@ -1,4 +1,3 @@
-from typing import Any
 from typing import Dict
 from typing import Generic
 from typing import List
@@ -32,6 +31,9 @@ class JSONResponse(BaseModel):
     elapsed: str
     err: Optional[str] = None
 
+    class Config:
+        exclude_unset = True
+
 
 class Object(GenericModel, Generic[T]):
     object: T
@@ -136,7 +138,7 @@ class Stats(BaseModel):
 
 
 class StatsResponse(JSONResponse):
-    stats: List[Union[Stats, Any]]
+    stats: Optional[List[Stats]] = []
 
 
 class ServerStatsLeader(BaseModel):

tests/test_command_stats.py

@@ -5,6 +5,10 @@
 async def test_command_stats(tile38_with_follower):
     tile38 = tile38_with_follower
 
+    response = await tile38.stats(["fleet"])
+    assert response.ok
+    assert response.stats == []
+
     await tile38.set("fleet", "truck1").point(1, 1).exec()
     await tile38.set("zones", "parking1").bounds(1, 1, 1, 1).exec()