Add setLineItemCustomField action for cart and order

Author: tleguijt

CHANGES.md

@@ -33,6 +33,7 @@
 - Testing: Add `datetime` and `list` updater utils
 - Testing: Add `setLocalizedDescription` action for shipping_method
 - Testing: Add `changeAttributeConstraint` action for product_type
+- Testing: Add `setLineItemCustomField` action for cart and order
 
 
 ## Notes on code generation

src/commercetools/testing/carts.py

@@ -198,21 +198,6 @@ def urls(self):
             ("^(?P<id>[^/]+)$", "DELETE", self.delete_by_id),
         ]
 
-    def set_custom_field(self, obj, action: models.CartSetCustomFieldAction):
-        if not obj["custom"]:
-            raise ValueError(
-                "This resource has no custom type set - please use "
-                "setCustomType first to set the type of the custom fields"
-            )
-
-        name = action.name
-        value = action.value
-
-        # real API always increments version, so always apply new value.
-        new = copy.deepcopy(obj)
-        new["custom"]["fields"][name] = value
-        return new
-
     def set_custom_type(self, obj, action: models.CartSetCustomTypeAction):
         custom_type = self.model._storage.get_by_resource_identifier(action.type)
 
@@ -226,6 +211,7 @@ def set_custom_type(self, obj, action: models.CartSetCustomTypeAction):
 
     _actions = {
         "addPayment": add_payment(),
-        "setCustomField": set_custom_field,
+        "setCustomField": utils.set_custom_field(),
+        "setLineItemCustomField": utils.set_line_item_custom_field(),
         "setCustomType": set_custom_type,
     }

src/commercetools/testing/orders.py

@@ -17,6 +17,7 @@
 from commercetools.testing.abstract import BaseModel, ServiceBackend
 from commercetools.testing.utils import (
     set_custom_field,
+    set_line_item_custom_field,
     update_attribute,
     update_enum_attribute,
 )
@@ -154,5 +155,6 @@ def urls(self):
         "setBillingAddress": update_attribute("billingAddress", "address"),
         "setCustomerEmail": update_attribute("customerEmail", "email"),
         "setCustomField": set_custom_field(),
+        "setLineItemCustomField": set_line_item_custom_field(),
         "addPayment": add_payment(),
     }

src/commercetools/testing/utils.py

@@ -17,6 +17,8 @@
 )
 from commercetools.platform.models._abstract import _BaseType
 from commercetools.platform.models._schemas.product import ProductSchema
+from commercetools.platform.models.cart import Cart, CartSetLineItemCustomFieldAction
+from commercetools.platform.models.order import Order, OrderSetLineItemCustomFieldAction
 
 
 class InternalUpdateError(ValueError):
@@ -223,6 +225,12 @@ def updater(
         obj,
         action: typing.Union[OrderSetCustomFieldAction, CartSetCustomFieldAction],
     ):
+        if not obj["custom"]:
+            raise ValueError(
+                "This resource has no custom type set - please use "
+                "setCustomType first to set the type of the custom fields"
+            )
+
         name = action.name
         value = action.value
 
@@ -236,6 +244,36 @@ def updater(
     return updater
 
 
+def set_line_item_custom_field():
+    """Set custom field of a line item.
+    Note it always sets the type now, instead of type checking the custom field type!
+    """
+
+    def updater(
+        self,
+        obj,
+        action: typing.Union[
+            OrderSetLineItemCustomFieldAction, CartSetLineItemCustomFieldAction
+        ],
+    ):
+        line_item_id = action.line_item_id
+        name = action.name
+        value = action.value
+
+        # real API always increments version, so always apply new value.
+        new = copy.deepcopy(obj)
+        for line in new["lineItems"]:
+            if line["id"] != line_item_id:
+                continue
+
+            if not line["custom"]:
+                line["custom"] = {"fields": {}}
+            line["custom"]["fields"][name] = value
+        return new
+
+    return updater
+
+
 def get_product_from_storage(
     storage: "commercetools.testing.Storage",
     product_id: uuid.UUID = None,