Support 'in' syntax for test query predicates (#105)

Co-authored-by: David Weterings <d.weterings@labdigital.nl>

Author: davidweterings

src/commercetools/testing/predicates.py

@@ -9,6 +9,7 @@
 import operator
 import re
 import typing
+from functools import partial
 
 import marshmallow
 
@@ -419,6 +420,7 @@ class PredicateFilter:
         "!=": operator.ne,
         "<>": operator.ne,
         "is not": operator.is_not,
+        "in": operator.contains,
     }
 
     def __init__(self, predicate, schema):
@@ -481,6 +483,8 @@ def filter_field(
                     for child_doc in obj
                 )
 
+        is_sequence_value = False
+
         if isinstance(schema_field, marshmallow.fields.Dict):
             obj = schema_field._deserialize(obj, None, None)
             assert len(path) == 1
@@ -489,15 +493,27 @@ def filter_field(
             if obj is not None:
                 obj = schema_field._deserialize(obj, None, None)
             if value is not None:
-                value = schema_field._deserialize(value, None, None)
+                is_sequence_value = isinstance(value, tuple) or isinstance(value, list)
+                if is_sequence_value:
+                    deserialize = partial(
+                        schema_field._deserialize, attr=None, data=None
+                    )
+                    value = list(map(deserialize, value))
+                else:
+                    value = schema_field._deserialize(value, None, None)
 
-        # Case insensitve comparison for strings
+        op = self.operators[operator_value]
+
+        # Case insensitive comparison for strings
         if isinstance(obj, str):
             obj = obj.lower()
         if isinstance(value, str):
             value = value.lower()
+        elif is_sequence_value:
+            value = [x.lower() if isinstance(x, str) else x for x in value]
+            if operator_value == "in":
+                return op(value, obj)
 
-        op = self.operators[operator_value]
         return op(obj, value)
 
     def case_insensitive_get(sef, dict, key, default=None):

tests/test_testing_predicates.py

@@ -1,6 +1,3 @@
-import ast
-import re
-
 import pytest
 
 from commercetools._schemas._product import ProductDataSchema
@@ -58,3 +55,27 @@ def mock_filter_field(obj, path, operator, value):
     pf.filter_field = mock_filter_field
     pf.match({"slug": {"nl-BE": "test-categorie"}})
     assert paths == found_paths
+
+
+def test_in_filter():
+    skus = '("1337", "1338")'
+    predicate = f"masterVariant(sku in {skus}) or variants(sku in {skus})"
+    pf = predicates.PredicateFilter(predicate, schema=ProductDataSchema)
+
+    product_data = {
+        "masterVariant": {"sku": "1337"},
+        "variants": [{"sku": "1338"}, {"sku": "1339"}],
+    }
+    assert pf.match(product_data) is True
+
+    product_data = {
+        "masterVariant": {"sku": "1444"},
+        "variants": [{"sku": "1338"}, {"sku": "1339"}],
+    }
+    assert pf.match(product_data) is True
+
+    product_data = {
+        "masterVariant": {"sku": "1444"},
+        "variants": [{"sku": "1438"}, {"sku": "1139"}],
+    }
+    assert pf.match(product_data) is False