Add testing backend for CustomObject query by container

Author: davidweterings

src/commercetools/testing/custom_objects.py

@@ -46,8 +46,18 @@ def urls(self):
             ("^$", "GET", self.query),
             ("^$", "POST", self.create),
             ("^(?P<container>[^/]+)/(?P<key>[^/]+)$", "GET", self.get_by_container_key),
+            ("^(?P<container>[^/]+)$", "GET", self.query_by_container),
         ]
 
+    def query_by_container(self, request, container: str):
+        # container is not a valid predicate filter, but we can abuse it internally.
+        if "where" not in request.qs:
+            request.qs["where"] = f'container = "{container}"'
+        else:
+            request.qs["where"] = f'container = "{container}" AND {request.qs["where"]}'
+
+        return self.query(request)
+
     def get_by_container_key(self, request, container: str, key: str):
         item = next(
             (

tests/test_service_custom_objects.py

@@ -59,3 +59,21 @@ def test_custom_object_update(client):
         types.CustomObjectDraft(container="unittest", key="test-object-1", value=2345)
     )
     assert custom_object.key == "test-object-1"
+
+
+def test_custom_object_query_by_container(client):
+    """Test filtering by container."""
+    client.custom_objects.create_or_update(
+        types.CustomObjectDraft(container="unittest", key="test-object-1", value=1234)
+    )
+    client.custom_objects.create_or_update(
+        types.CustomObjectDraft(container="unittest", key="test-object-2", value=1234)
+    )
+    client.custom_objects.create_or_update(
+        types.CustomObjectDraft(container="unittest2", key="test-object-1", value=1234)
+    )
+
+    result = client.custom_objects.query_by_container("unittest")
+    assert len(result.results) == 2
+    assert result.total == 2
+