Add missing API actions to testing backend

Author: qdegraaf

src/commercetools/testing/cart_discounts.py

@@ -1,3 +1,4 @@
+import copy
 import datetime
 import typing
 import uuid
@@ -11,7 +12,11 @@
 )
 from commercetools.testing import utils
 from commercetools.testing.abstract import BaseModel, ServiceBackend
-from commercetools.testing.utils import update_attribute, update_enum_attribute
+from commercetools.testing.utils import (
+    update_attribute,
+    update_datetime_attribute,
+    update_enum_attribute,
+)
 
 
 class CartDiscountsModel(BaseModel):
@@ -72,4 +77,6 @@ def urls(self):
         "setName": update_attribute("name", "name"),
         "setDescription": update_attribute("description", "description"),
         "setCartPredicate": update_attribute("cartPredicate", "cart_predicate"),
+        "setValidFrom": update_datetime_attribute("validFrom", "valid_from"),
+        "setValidUntil": update_datetime_attribute("validUntil", "valid_until"),
     }

src/commercetools/testing/discount_codes.py

@@ -1,3 +1,4 @@
+import copy
 import datetime
 import typing
 import uuid
@@ -11,7 +12,11 @@
 )
 from commercetools.testing import utils
 from commercetools.testing.abstract import BaseModel, ServiceBackend
-from commercetools.testing.utils import update_attribute
+from commercetools.testing.utils import (
+    update_attribute,
+    update_datetime_attribute,
+    update_nested_object_attribute,
+)
 
 
 class DiscountCodesModel(BaseModel):
@@ -67,4 +72,9 @@ def urls(self):
         "setMaxApplicationsPerCustomer": update_attribute(
             "maxApplicationsPerCustomer", "max_applications_per_customer"
         ),
+        "setValidFrom": update_datetime_attribute("validFrom", "valid_from"),
+        "setValidUntil": update_datetime_attribute("validUntil", "valid_until"),
+        "changeCartDiscounts": update_nested_object_attribute(
+            "cartDiscounts", "cart_discounts"
+        ),
     }

src/commercetools/testing/extensions.py

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

src/commercetools/testing/utils.py

@@ -3,6 +3,7 @@
 import json
 import typing
 import uuid
+from datetime import datetime
 
 from marshmallow import Schema
 from requests_mock import create_response
@@ -113,6 +114,40 @@ def updater(self, obj, action):
     return updater
 
 
+def update_datetime_attribute(dst: str, src: str):
+    def updater(self, obj, action):
+        value = getattr(action, src)
+
+        if not isinstance(value, datetime):
+            raise TypeError(f"Unsupported datetime object type: f{type(value)}")
+
+        if obj.get(dst) != value.isoformat():
+            new = copy.deepcopy(obj)
+            new[dst] = value.isoformat()
+            return new
+        return obj
+
+    return updater
+
+
+def update_nested_object_attribute(dst: str, src: str):
+    def updater(self, obj, action):
+        values = getattr(action, src)
+
+        if not isinstance(values, list):
+            raise TypeError(f"Unsupported nested object type: f{type(values)}")
+
+        items = [item.serialize() for item in values]
+        if items != obj.get(dst):
+            new = copy.deepcopy(obj)
+            new[dst] = items
+            return new
+
+        return obj
+
+    return updater
+
+
 def update_enum_attribute(dst: str, src: str):
     def updater(self, obj, action):
         value = getattr(action, src).value

tests/platform/test_service_cart_discounts.py

@@ -1,4 +1,7 @@
+from datetime import datetime
+
 import pytest
+from freezegun import freeze_time
 from requests.exceptions import HTTPError
 
 from commercetools.platform import models
@@ -90,3 +93,51 @@ def test_cart_discount_update(old_client):
     )
 
     assert cart_discount.is_active is False
+
+
+@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
+    assert cart_discount.valid_from is None
+
+    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()
+
+
+@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
+    assert cart_discount.valid_until is None
+
+    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,4 +1,7 @@
+from datetime import datetime
+
 import pytest
+from freezegun import freeze_time
 from requests.exceptions import HTTPError
 
 from commercetools.platform import models
@@ -69,3 +72,89 @@ def test_discount_code_update(old_client):
     )
 
     assert discount_code.is_active is False
+
+
+@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()
+
+
+@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.version == 2
+    assert discount_code.valid_until == datetime.now()
+
+
+def test_discount_code_change_cart_discounts(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.cart_discounts == []
+
+    cart_discount = old_client.cart_discounts.create(
+        models.CartDiscountDraft(
+            name=models.LocalizedString(en="cart-discount-test"),
+            value=models.CartDiscountValueDraft(type="absolute"),
+            cart_predicate="sku",
+            sort_order="1",
+            requires_discount_code=True,
+        )
+    )
+    assert cart_discount.id
+
+    discount_code = old_client.discount_codes.update_by_id(
+        id=discount_code.id,
+        version=discount_code.version,
+        actions=[
+            models.DiscountCodeChangeCartDiscountsAction(
+                cart_discounts=[
+                    models.CartDiscountResourceIdentifier(id=cart_discount.id)
+                ]
+            )
+        ],
+    )
+
+    assert discount_code.version == 2
+    assert discount_code.cart_discounts == [
+        models.CartDiscountReference(id=cart_discount.id)
+    ]

tests/platform/test_service_extensions.py

@@ -26,3 +26,40 @@ 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],
+        )
+    ]