Merge pull request #47 from cuent/patch-1

Add must to query type
vrcmarcos · Dec 3, 2020 · e2e47f9 · e2e47f9
2 parents 15cd0c6 + affbc4c
commit e2e47f9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/elasticmock/fake_elasticsearch.py b/elasticmock/fake_elasticsearch.py
@@ -25,6 +25,7 @@ class QueryType:
     MATCH = 'MATCH'
     TERM = 'TERM'
     TERMS = 'TERMS'
+    MUST = 'MUST'
 
     @staticmethod
     def get_query_type(type_str):
@@ -38,6 +39,8 @@ def get_query_type(type_str):
             return QueryType.TERM
         elif type_str == 'terms':
             return QueryType.TERMS
+        elif type_str == 'must':
+            return QueryType.MUST
         else:
             raise NotImplementedError(f'type {type_str} is not implemented for QueryType')