implement count functionality

Signed-off-by: wiseaidev <[email protected]>

Author: wiseaidev

aredis_om/model/model.py

@@ -760,6 +760,9 @@ async def all(self, batch_size=10):
             return await query.execute()
         return await self.execute()
 
+    async def count(self, batch_size=10):
+        return len(await self.all(batch_size))
+
     def sort_by(self, *fields: str):
         if not fields:
             return self

tests/test_hash_model.py

@@ -95,6 +95,13 @@ async def members(m):
 
     yield member1, member2, member3
 
+@py_test_mark_asyncio
+async def test_count_query(members, m):
+
+    count = await m.Member.find(m.Member.first_name="Kim").count()
+    assert count == 1
+    count = await m.Member.find().count()
+    assert count == 3
 
 @py_test_mark_asyncio
 async def test_exact_match_queries(members, m):