File tree Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Expand file tree Collapse file tree 2 files changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -760,6 +760,9 @@ async def all(self, batch_size=10):
760
760
return await query .execute ()
761
761
return await self .execute ()
762
762
763
+ async def count (self , batch_size = 10 ):
764
+ return len (await self .all (batch_size ))
765
+
763
766
def sort_by (self , * fields : str ):
764
767
if not fields :
765
768
return self
Original file line number Diff line number Diff line change @@ -95,6 +95,15 @@ async def members(m):
95
95
96
96
yield member1 , member2 , member3
97
97
98
+ @py_test_mark_asyncio
99
+ async def test_count_query (members , m ):
100
+
101
+ count = await m .Member .find ((m .Member .first_name == "Brookins" ) & (m .Member .last_name == "Brookins" )).count ()
102
+ assert count == 1
103
+ count = await m .Member .find (m .Member .first_name = "Kim" ).count ()
104
+ assert count == 1
105
+ count = await m .Member .find ().count ()
106
+ assert count == 3
98
107
99
108
@py_test_mark_asyncio
100
109
async def test_exact_match_queries (members , m ):
You can’t perform that action at this time.
0 commit comments