feat: support for find queries in count

src/quorum/model.py

@@ -593,6 +593,10 @@ def find(cls, *args, **kwargs):
     @classmethod
     def count(cls, *args, **kwargs):
         cls._clean_attrs(kwargs)
+
+        cls._find_s(kwargs)
+        cls._find_d(kwargs)
+
         collection = cls._collection()
         if kwargs:
             if hasattr(collection, "count_documents"):

src/quorum/test/model.py

@@ -118,6 +118,27 @@ def test_count(self):
         result = mock.Person.count()
         self.assertEqual(result, 1)
 
+    @quorum.secured
+    def test_count_find(self):
+        adapter = quorum.get_adapter()
+        if not adapter.name in ("mongo",):
+            if not hasattr(self, "skipTest"): return
+            self.skipTest("Adapter is not supported")
+
+        result = mock.Person.count()
+        self.assertEqual(result, 0)
+
+        person = mock.Person()
+        person.age = 1
+        person.name = "Name"
+        person.save()
+
+        result = mock.Person.count(**dict(find_d = ["name:eq:Name"]))
+        self.assertEqual(result, 1)
+
+        result = mock.Person.count(**dict(find_d = ["name:eq:OtherName"]))
+        self.assertEqual(result, 0)
+
     @quorum.secured
     def test_delete(self):
         result = mock.Person.count()