Expose tantivy's AllQuery (#230)

quickwit-oss · Mar 31, 2024 · 255eb6e · 255eb6e
1 parent e249241
commit 255eb6e
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/src/query.rs b/src/query.rs
@@ -43,4 +43,13 @@ impl Query {
             inner: Box::new(inner),
         })
     }
+
+    /// Construct a Tantivy's AllQuery
+    #[staticmethod]
+    pub(crate) fn all_query() -> PyResult<Query> {
+        let inner = tv::query::AllQuery {};
+        Ok(Query {
+            inner: Box::new(inner),
+        })
+    }
 }
diff --git a/tantivy/tantivy.pyi b/tantivy/tantivy.pyi
@@ -193,6 +193,10 @@ class Query:
     def term_query(schema: Schema, field_name: str, field_value: Any, index_option: str = "position") -> Query:
         pass
 
+    @staticmethod
+    def all_query() -> Query:
+        pass
+
 
 class Order(Enum):
     Asc = 1

diff --git a/tests/tantivy_test.py b/tests/tantivy_test.py
@@ -764,3 +764,10 @@ def test_term_query(self, ram_index):
         _, doc_address = result.hits[0]
         searched_doc = index.searcher().doc(doc_address)
         assert searched_doc["title"] == ["The Old Man and the Sea"]
+
+    def test_all_query(self, ram_index):
+        index = ram_index
+        query = Query.all_query()
+
+        result = index.searcher().search(query, 10)
+        assert len(result.hits) == 3