Implement publish and set prices action for products (#87)

* Implement publish and set prices action for products

* Always set hasStagedChanges to False when publishing

* Added tests for product update actions

Author: davidweterings

src/commercetools/testing/products.py

@@ -5,7 +5,7 @@
 
 from commercetools import schemas, types
 from commercetools.testing.abstract import BaseModel, ServiceBackend
-from commercetools.testing.utils import custom_fields_from_draft
+from commercetools.testing.utils import custom_fields_from_draft, get_product_variants
 
 
 class ProductsModel(BaseModel):
@@ -149,8 +149,8 @@ def _create_price_from_draft(
             )
 
 
-def _get_target_obj(obj: dict, action: types.ProductUpdateAction):
-    staged = getattr(action, "staged", False)
+def _get_target_obj(obj: dict, action: types.ProductUpdateAction, default_staged=False):
+    staged = getattr(action, "staged", default_staged)
     if not staged and obj["masterData"]["current"]:
         return obj["masterData"]["current"]
     return obj["masterData"]["staged"]
@@ -227,6 +227,54 @@ def updater(self, obj: dict, action: types.ProductAddVariantAction):
     return updater
 
 
+def _publish_product_action():
+    def updater(self, obj: dict, action: types.ProductPublishAction):
+        new = copy.deepcopy(obj)
+        # not implemented scopes right now.
+        if "staged" in new["masterData"]:
+            new["masterData"]["current"] = new["masterData"]["staged"]
+        new["masterData"]["hasStagedChanges"] = False
+        new["masterData"]["published"] = True
+        return new
+
+    return updater
+
+
+def _set_product_prices():
+    def updater(self, obj: dict, action: types.ProductSetPricesAction):
+        new = copy.deepcopy(obj)
+        target_obj = _get_target_obj(new, action, default_staged=True)
+        prices = []
+        for price_draft in action.prices:
+            price = types.Price(
+                id=str(uuid.uuid4()),
+                country=price_draft.country,
+                channel=price_draft.channel,
+                value=types.TypedMoney(
+                    fraction_digits=2,
+                    cent_amount=price_draft.value.cent_amount,
+                    currency_code=price_draft.value.currency_code,
+                    type=types.MoneyType.CENT_PRECISION,
+                ),
+                valid_from=price_draft.valid_from,
+                valid_until=price_draft.valid_until,
+                discounted=price_draft.discounted,
+                custom=price_draft.custom,
+                tiers=price_draft.tiers,
+            )
+            prices.append(price)
+
+        schema = schemas.PriceSchema()
+        for variant in get_product_variants(target_obj):
+            if variant["sku"] == action.sku:
+                variant["prices"] = schema.dump(prices, many=True)
+        if action.staged:
+            new["masterData"]["hasStagedChanges"] = True
+        return new
+
+    return updater
+
+
 class ProductsBackend(ServiceBackend):
     service_path = "products"
     model_class = ProductsModel
@@ -251,4 +299,6 @@ def urls(self):
         "changeSlug": _update_productdata_attr("slug", "slug"),
         "setAttribute": _set_attribute_action(),
         "addVariant": _add_variant_action(),
+        "setPrices": _set_product_prices(),
+        "publish": _publish_product_action(),
     }

src/commercetools/testing/utils.py

@@ -32,6 +32,15 @@ def flatten_multivaluedict(data: dict) -> dict:
     return result
 
 
+def get_product_variants(product_catalog_data: dict) -> typing.List[dict]:
+    variants = []
+    if product_catalog_data.get("masterVariant"):
+        variants.append(product_catalog_data["masterVariant"])
+    if product_catalog_data.get("variants"):
+        variants += product_catalog_data["variants"]
+    return variants
+
+
 def create_from_draft(draft):
     """Utility method to create normal objects out of draft objects.
 

tests/test_service_products.py

@@ -145,7 +145,12 @@ def test_product_update(client):
     It doesn't test the actual update itself.
     TODO: See if this is worth testing since we're using a mocking backend
     """
-    product = client.products.create(types.ProductDraft(key="test-product"))
+    product = client.products.create(
+        types.ProductDraft(
+            key="test-product",
+            master_variant=types.ProductVariantDraft(sku="1", key="1"),
+        )
+    )
 
     assert uuid.UUID(product.id)
     assert product.key == "test-product"
@@ -158,6 +163,31 @@ def test_product_update(client):
         ],
     )
     assert product.key == "test-product"
+    assert product.master_data.published is False
+
+    product = client.products.update_by_id(
+        id=product.id, version=product.version, actions=[types.ProductPublishAction()]
+    )
+    assert product.master_data.published is True
+
+    assert not product.master_data.current.master_variant.prices
+    product = client.products.update_by_id(
+        id=product.id,
+        version=product.version,
+        actions=[
+            types.ProductSetPricesAction(
+                sku="1",
+                prices=[
+                    types.PriceDraft(
+                        value=types.Money(cent_amount=1000, currency_code="GBP")
+                    )
+                ],
+                staged=False,
+            )
+        ],
+    )
+
+    assert len(product.master_data.current.master_variant.prices) == 1
 
     product = client.products.update_by_key(
         key="test-product",