Testing: Add addPayment action to order testing backend

Author: tleguijt

CHANGES

@@ -6,6 +6,7 @@ x.x.x (unreleased)
 - Improved error information in CommercetoolsError object
 - Testing: Add setAttribute and addVariant actions to product testing backend
 - Testing: Add addAttributeDefinition action to testing backend
+- Testing: Add addPayment action to order testing backend
 - Testing: Check for unique values in testing backend
 - Testing: Fixed Attribute return object in ProductTypes testing backend
 

src/commercetools/testing/orders.py

@@ -75,6 +75,21 @@ def updater(self, obj, action):
     return updater
 
 
+def add_payment():
+    def updater(self, obj, action):
+        obj = copy.deepcopy(obj)
+        if "paymentInfo" in obj and obj["paymentInfo"]:
+            payments = obj["paymentInfo"]["payments"]
+        else:
+            payments = []
+
+        payments.append(schemas.PaymentReferenceSchema().dump(action.payment))
+        obj["paymentInfo"] = {"payments": payments}
+        return obj
+
+    return updater
+
+
 class OrdersBackend(ServiceBackend):
     service_path = "orders"
     model_class = OrdersModel
@@ -102,4 +117,5 @@ def urls(self):
         "setBillingAddress": update_attribute("billingAddress", "address"),
         "setCustomerEmail": update_attribute("customerEmail", "email"),
         "setCustomField": set_custom_field(),
+        "addPayment": add_payment(),
     }