Added delete and update methods to custom objects test

Author: Ardjan-Aalberts

src/commercetools/testing/abstract.py

@@ -118,6 +118,16 @@ def delete_by_key(self, key):
             if obj["key"] == key:
                 return self.objects.pop(obj_id)
 
+    def delete_by_container_key(self, container, key):
+        return next(
+            (
+                obj
+                for obj in self.objects.values()
+                if obj["container"] == container and obj["key"] == key
+            ),
+            None,
+        )
+
     def save(self, obj):
         assert obj["id"]
         obj["version"] += 1
@@ -277,6 +287,18 @@ def delete_by_key(self, request, key):
             return create_commercetools_response(request, json=obj)
         return create_commercetools_response(request, status_code=404)
 
+    def delete_by_container_key(self, request, container, key):
+        obj = self.model.get_by_container_key(container, key)
+        if obj:
+            if self._verify_version:
+                response = self._validate_resource_version(request, obj)
+                if response is not None:
+                    return response
+
+            obj = self.model.delete_by_container_key(key)
+            return create_commercetools_response(request, json=obj)
+        return create_commercetools_response(request, status_code=404)
+
     def _expand(self, request: _RequestObjectProxy, raw_obj):
         params = utils.parse_request_params(self._schema_query_params, request)
         if "expand" not in params:

src/commercetools/testing/custom_objects.py

@@ -81,12 +81,12 @@ def urls(self):
             (
                 "^(?P<container>[^/]+)/(?P<key>[^/]+)$",
                 "POST",
-                self.get_by_container_key,
+                self.update_by_container_key,
             ),
             (
                 "^(?P<container>[^/]+)/(?P<key>[^/]+)$",
                 "DELETE",
-                self.get_by_container_key,
+                self.delete_by_container_key,
             ),
             ("^(?P<container>[^/]+)$", "GET", self.query_by_container),
         ]
@@ -106,3 +106,19 @@ def get_by_container_key(self, request, container: str, key: str):
             return create_commercetools_response(request, json=item)
         else:
             return create_commercetools_response(request, status_code=404)
+
+    def update_by_container_key(self, request, container: str, key: str):
+        item = self.model._get_by_container_key(container, key)
+        return self._update(request, item)
+
+    def delete_by_container_key(self, request, container: str, key: str):
+        item = self.model._get_by_container_key(container, key)
+        if item:
+            if self._verify_version:
+                response = self._validate_resource_version(request, item)
+                if response is not None:
+                    return response
+
+            obj = self.model.delete_by_container_key(container, key)
+            return create_commercetools_response(request, json=obj)
+        return create_commercetools_response(request, status_code=404)