Feature/add product testing methods (#104)

* Add missing schema imports, fix incorrect type usage/type signatures

* Testing: fix product variants and add changeMasterVariant action

Co-authored-by: David Weterings <d.weterings@labdigital.nl>

Author: davidweterings

src/commercetools/testing/custom_objects.py

@@ -69,8 +69,7 @@ def get_by_container_key(self, request, container: str, key: str):
                     message=f"The CustomObject with ID '({container},{key})'",
                     errors=[
                         types.InvalidSubjectError(
-                            code="InvalidSubject",
-                            message=f"The CustomObject with ID '({container},{key}' was not found.",  # noqa E501
+                            message=f"The CustomObject with ID '({container},{key}' was not found."  # noqa E501
                         )
                     ],
                 )

src/commercetools/testing/customers.py

@@ -22,12 +22,13 @@ def _create_from_draft(
         self, draft: types.CustomerDraft, id: Optional[str] = None
     ) -> types.Customer:
         object_id = uuid.UUID(id) if id is not None else uuid.uuid4()
+        now = datetime.datetime.now(datetime.timezone.utc)
 
         return types.Customer(
             id=str(object_id),
             version=1,
-            created_at=datetime.datetime.now(datetime.timezone.utc),
-            last_modified_at=None,
+            created_at=now,
+            last_modified_at=now,
             customer_number=draft.customer_number,
             email=draft.email,
             password=draft.password,

src/commercetools/testing/orders.py

@@ -4,7 +4,7 @@
 import uuid
 
 from commercetools import types
-from commercetools._schemas._cart import CartSchema
+from commercetools._schemas._cart import CartSchema, ShippingInfoSchema
 from commercetools._schemas._order import (
     DeliverySchema,
     OrderFromCartDraftSchema,

src/commercetools/testing/payments.py

@@ -14,7 +14,6 @@
 from commercetools._schemas._type import CustomFieldsSchema
 from commercetools.testing import utils
 from commercetools.testing.abstract import BaseModel, ServiceBackend
-from commercetools.testing.utils import update_attribute_add_item
 
 
 class PaymentsModel(BaseModel):

src/commercetools/testing/product_projections.py

@@ -111,7 +111,7 @@ def get_by_key(self, request, key):
 
     def _convert_product_projection(
         self, product: types.Product, staged: bool = False
-    ) -> typing.Optional[typing.Dict]:
+    ) -> typing.Optional[types.ProductProjection]:
         """Convert a Product object to a ProductProjection object"""
         if product.master_data is None:
             return None

src/commercetools/testing/products.py

@@ -1,18 +1,20 @@
 import copy
 import datetime
 import uuid
-from typing import List, Optional, Union
+from typing import Dict, List, Optional, Union
 
 from marshmallow import Schema
 from marshmallow import fields as schema_fields
 
 from commercetools import types
 from commercetools._schemas._common import ImageSchema, PriceSchema
 from commercetools._schemas._product import (
+    AttributeSchema,
     ProductDraftSchema,
     ProductPagedQueryResponseSchema,
     ProductSchema,
     ProductUpdateSchema,
+    ProductVariantSchema,
 )
 from commercetools.services.products import _ProductQuerySchema
 from commercetools.testing import utils
@@ -45,7 +47,9 @@ def _create_from_draft(
             category_order_hints=draft.category_order_hints,
             description=draft.description,
             master_variant=master_variant,
-            variants=[master_variant] if master_variant else [],
+            variants=[self._create_variant_from_draft(vd) for vd in draft.variants]
+            if draft.variants
+            else [],
             slug=draft.slug or types.LocalizedString(),
             search_keywords=types.SearchKeywords(),
         )
@@ -375,6 +379,33 @@ def updater(self, obj: dict, action: types.ProductAddPriceAction):
     return updater
 
 
+def _change_master_variant():
+    def updater(self, obj: Dict, action: types.ProductChangeMasterVariantAction):
+        new = copy.deepcopy(obj)
+        staged = action.staged
+        if staged is None:
+            staged = True
+        target_obj = _get_target_obj(new, staged)
+        master_variant = target_obj["masterVariant"]
+        if master_variant and master_variant["sku"] == action.sku:
+            return obj
+        found_sku = False
+        for variant in get_product_variants(target_obj):
+            if variant["sku"] == action.sku:
+                target_obj["variants"].append(target_obj["masterVariant"])
+                target_obj["masterVariant"] = variant
+                target_obj["variants"].remove(variant)
+                found_sku = True
+                break
+        if not found_sku:
+            raise ValueError("Could not find sku %s" % action.sku)
+        if staged:
+            new["masterData"]["hasStagedChanges"] = True
+        return new
+
+    return updater
+
+
 class UploadImageQuerySchema(Schema):
     staged = schema_fields.Bool()
     filename = schema_fields.Field()
@@ -410,6 +441,7 @@ def urls(self):
         "setPrices": _set_product_prices(),
         "changePrice": _change_price(),
         "addPrice": _add_price(),
+        "changeMasterVariant": _change_master_variant(),
         "publish": _publish_product_action(),
     }
 

src/commercetools/testing/server.py

@@ -8,7 +8,6 @@
 from requests_mock.request import _RequestObjectProxy
 
 from commercetools.testing import BackendRepository
-from commercetools.testing.internal import InternalBackend
 
 logger = logging.getLogger(__name__)
 

src/commercetools/testing/shipping_methods.py

@@ -9,7 +9,10 @@
     ShippingMethodPagedQueryResponseSchema,
     ShippingMethodSchema,
     ShippingMethodUpdateSchema,
+    ShippingRateSchema,
+    ZoneRateSchema,
 )
+from commercetools._schemas._tax_category import TaxCategoryResourceIdentifierSchema
 from commercetools.testing import utils
 from commercetools.testing.abstract import BaseModel, ServiceBackend
 from commercetools.testing.utils import InternalUpdateError, update_attribute
@@ -65,14 +68,12 @@ def create_shipping_rate_from_draft(
         free_above = types.CentPrecisionMoney(
             cent_amount=draft.free_above.cent_amount,
             currency_code=draft.free_above.currency_code,
-            type=types.MoneyType.CENT_PRECISION,
             fraction_digits=2,
         )
     shipping_rate = types.ShippingRate(
         price=types.CentPrecisionMoney(
             cent_amount=draft.price.cent_amount,
             currency_code=draft.price.currency_code,
-            type=types.MoneyType.CENT_PRECISION,
             fraction_digits=2,
         ),
         free_above=free_above,

src/commercetools/testing/utils.py

@@ -54,8 +54,7 @@ def create_from_draft(draft):
 
     if isinstance(draft, types.CustomFieldsDraft):
         return types.CustomFields(
-            type=types.TypeReference(type_id=draft.type.type_id, id=draft.type.id),
-            fields=draft.fields,
+            type=types.TypeReference(id=draft.type.id), fields=draft.fields
         )
     if isinstance(draft, types.PriceTierDraft):
         return types.PriceTier(
@@ -164,7 +163,6 @@ def updater(self, obj, action):
             return new
         else:
             raise InternalUpdateError("No item found with id %r" % (obj["id"]))
-        return obj
 
     return updater