Merge pull request #54 from jankislinger/match-all-query

allow 'match_all' queries in FakeSearch
vrcmarcos · Dec 31, 2020 · 0b54228 · 0b54228
2 parents 90d5135 + 23530b5
commit 0b54228
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/elasticmock/fake_elasticsearch.py b/elasticmock/fake_elasticsearch.py
@@ -62,6 +62,8 @@ def evaluate(self, document):
     def _evaluate_for_query_type(self, document):
         if self.type == QueryType.MATCH:
             return self._evaluate_for_match_query_type(document)
+        elif self.type == QueryType.MATCH_ALL:
+            return True
         elif self.type == QueryType.TERM:
             return self._evaluate_for_term_query_type(document)
         elif self.type == QueryType.TERMS:

diff --git a/tests/fake_elasticsearch/test_search.py b/tests/fake_elasticsearch/test_search.py
@@ -28,6 +28,14 @@ def test_should_return_all_documents(self):
         search = self.es.search()
         self.assertEqual(index_quantity, search.get('hits').get('total'))
 
+    def test_should_return_all_documents_match_all(self):
+        index_quantity = 10
+        for i in range(0, index_quantity):
+            self.es.index(index='index_{0}'.format(i), doc_type=DOC_TYPE, body={'data': 'test_{0}'.format(i)})
+
+        search = self.es.search(body={'query': {'match_all': {}}})
+        self.assertEqual(index_quantity, search.get('hits').get('total'))
+
     def test_should_return_only_indexed_documents_on_index(self):
         index_quantity = 2
         for i in range(0, index_quantity):