Add missing cart_discount, discount_code and extension actions

Author: qdegraaf

src/commercetools/testing/cart_discounts.py

@@ -72,4 +72,6 @@ def urls(self):
         "setName": update_attribute("name", "name"),
         "setDescription": update_attribute("description", "description"),
         "setCartPredicate": update_attribute("cartPredicate", "cart_predicate"),
+        "setValidFrom": update_attribute("valid_from", "valid_from"),
+        "setValidUntil": update_attribute("valid_until", "valid_until")
     }

src/commercetools/testing/discount_codes.py

@@ -67,4 +67,6 @@ def urls(self):
         "setMaxApplicationsPerCustomer": update_attribute(
             "maxApplicationsPerCustomer", "max_applications_per_customer"
         ),
+        "setValidFrom": update_attribute("valid_from", "valid_from"),
+        "setValidUntil": update_attribute("valid_until", "valid_until")
     }

src/commercetools/testing/extensions.py

@@ -10,6 +10,7 @@
     ExtensionUpdateSchema,
 )
 from commercetools.testing.abstract import BaseModel, ServiceBackend
+from commercetools.testing.utils import update_attribute
 
 
 class ExtensionsModel(BaseModel):
@@ -49,3 +50,7 @@ def urls(self):
             ("^(?P<id>[^/]+)$", "DELETE", self.delete_by_id),
             ("^key=(?P<key>[^/]+)$", "DELETE", self.delete_by_key),
         ]
+
+    _actions = {
+        "changeTriggers": update_attribute("triggers", "triggers"),
+    }

tests/platform/test_service_cart_discounts.py

@@ -1,3 +1,5 @@
+from datetime import datetime
+
 import pytest
 from requests.exceptions import HTTPError
 
@@ -90,3 +92,49 @@ def test_cart_discount_update(old_client):
     )
 
     assert cart_discount.is_active is False
+
+
+@pytest.mark.freeze_time("2021-03-01 12:34:56")
+def test_cart_discount_set_valid_from(old_client):
+    cart_discount = old_client.cart_discounts.create(
+        models.CartDiscountDraft(
+            name=models.LocalizedString(en="en-cart_discount"),
+            value=models.CartDiscountValueRelative(permyriad=10),
+            is_active=True,
+            cart_predicate="",
+            sort_order="",
+            requires_discount_code=False,
+        )
+    )
+    assert cart_discount.id
+
+    cart_discount = old_client.cart_discounts.update_by_id(
+        id=cart_discount.id,
+        version=cart_discount.version,
+        actions=[models.CartDiscountSetValidFromAction(valid_from=datetime.now())],
+    )
+
+    assert cart_discount.valid_from == datetime.now()
+
+
+@pytest.mark.freeze_time("2021-03-01 12:34:56")
+def test_cart_discount_set_valid_until(old_client):
+    cart_discount = old_client.cart_discounts.create(
+        models.CartDiscountDraft(
+            name=models.LocalizedString(en="en-cart_discount"),
+            value=models.CartDiscountValueRelative(permyriad=10),
+            is_active=True,
+            cart_predicate="",
+            sort_order="",
+            requires_discount_code=False,
+        )
+    )
+    assert cart_discount.id
+
+    cart_discount = old_client.cart_discounts.update_by_id(
+        id=cart_discount.id,
+        version=cart_discount.version,
+        actions=[models.CartDiscountSetValidUntilAction(valid_until=datetime.now())],
+    )
+
+    assert cart_discount.valid_until == datetime.now()

tests/platform/test_service_discount_codes.py

@@ -1,3 +1,5 @@
+from datetime import datetime
+
 import pytest
 from requests.exceptions import HTTPError
 
@@ -69,3 +71,47 @@ def test_discount_code_update(old_client):
     )
 
     assert discount_code.is_active is False
+
+
+@pytest.mark.freeze_time("2021-03-01 12:34:56")
+def test_discount_code_set_valid_from(old_client):
+    discount_code = old_client.discount_codes.create(
+        models.DiscountCodeDraft(
+            name=models.LocalizedString(en="en-discount_code"),
+            code="1337",
+            is_active=True,
+            cart_discounts=[],
+        )
+    )
+    assert discount_code.id
+    assert discount_code.valid_from is None
+
+    discount_code = old_client.discount_codes.update_by_id(
+        id=discount_code.id,
+        version=discount_code.version,
+        actions=[models.DiscountCodeSetValidFromAction(valid_from=datetime.now())],
+    )
+
+    assert discount_code.valid_from == datetime.now()
+
+
+@pytest.mark.freeze_time("2021-03-01 12:34:56")
+def test_discount_code_set_valid_until(old_client):
+    discount_code = old_client.discount_codes.create(
+        models.DiscountCodeDraft(
+            name=models.LocalizedString(en="en-discount_code"),
+            code="1337",
+            is_active=True,
+            cart_discounts=[],
+        )
+    )
+    assert discount_code.id
+    assert discount_code.valid_until is None
+
+    discount_code = old_client.discount_codes.update_by_id(
+        id=discount_code.id,
+        version=discount_code.version,
+        actions=[models.DiscountCodeSetValidUntilAction(valid_until=datetime.now())],
+    )
+
+    assert discount_code.valid_until == datetime.now()

tests/platform/test_service_extensions.py

@@ -26,3 +26,31 @@ def test_extension_get_by_id(old_client):
 
     extension = old_client.extensions.get_by_id(extension.id)
     assert extension.id
+
+
+def test_extension_update_change_triggers(old_client):
+    extension = old_client.extensions.create(models.ExtensionDraft(
+            destination=models.ExtensionAWSLambdaDestination(
+                arn="arn:", access_key="access", access_secret="secret"
+            ),
+            triggers=[],
+        )
+    )
+    assert extension.id
+    assert extension.triggers == []
+
+    extension = old_client.extensions.update_by_id(
+        id=extension.id,
+        version=extension.version,
+        actions=[
+            models.ExtensionChangeTriggersAction(
+                triggers=[models.ExtensionTrigger(
+                    resource_type_id=models.ExtensionResourceTypeId.CART,
+                    actions=[models.ExtensionAction.CREATE, models.ExtensionAction.UPDATE]
+                )]
+        )
+    ])
+    assert extension.triggers == [models.ExtensionTrigger(
+                resource_type_id=models.ExtensionResourceTypeId.CART,
+                actions=[models.ExtensionAction.CREATE, models.ExtensionAction.UPDATE]
+            )]