add test count with limit

BeanieODM · Oct 7, 2024 · e747cc1 · e747cc1
1 parent 37844f0
commit e747cc1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/odm/documents/test_count.py b/tests/odm/documents/test_count.py
@@ -13,3 +13,13 @@ async def test_count_with_filter_query(documents):
     await documents(1, "cuatro", True)
     c = await DocumentTestModel.find_many({"test_str": "dos"}).count()
     assert c == 2
+
+
+async def test_count_with_limit(documents):
+    await documents(4, "uno", True)
+    await documents(2, "dos", True)
+    await documents(1, "cuatro", True)
+    c = await DocumentTestModel.find_many().limit(1).count()
+    assert c == 1
+    d = await DocumentTestModel.find_many().count()
+    assert d == 3