Complete _unique_values for all testing models

tleguijt · tleguijt · commit 989a60a6cffa · 2020-04-07T21:52:54.000+02:00
diff --git a/src/commercetools/testing/cart_discounts.py b/src/commercetools/testing/cart_discounts.py
@@ -11,6 +11,7 @@
 class CartDiscountsModel(BaseModel):
     _resource_schema = schemas.CartDiscountSchema
     _primary_type_name = "cart-discounts"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.CartDiscountDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/categories.py b/src/commercetools/testing/categories.py
@@ -10,6 +10,7 @@
 class CategoriesModel(BaseModel):
     _primary_type_name = "category"
     _resource_schema = schemas.CategorySchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.CategoryDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/channels.py b/src/commercetools/testing/channels.py
@@ -10,6 +10,7 @@
 class ChannelsModel(BaseModel):
     _resource_schema = schemas.ChannelSchema
     _primary_type_name = "channel"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.ChannelDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/customer_groups.py b/src/commercetools/testing/customer_groups.py
@@ -9,6 +9,7 @@
 class CustomerGroupModel(abstract.BaseModel):
     _resource_schema = schemas.CustomerGroupSchema
     _primary_type_name = "customer-group"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.CustomerGroupDraft, id: Optional[str] = None
diff --git a/src/commercetools/testing/customers.py b/src/commercetools/testing/customers.py
@@ -9,6 +9,7 @@
 class CustomerModel(abstract.BaseModel):
     _resource_schema = schemas.CustomerSchema
     _primary_type_name = "customer"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.CustomerDraft, id: Optional[str] = None
diff --git a/src/commercetools/testing/extensions.py b/src/commercetools/testing/extensions.py
@@ -9,6 +9,7 @@
 class ExtensionsModel(BaseModel):
     _primary_type_name = "extension"
     _resource_schema = schemas.ExtensionSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.ExtensionDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/payments.py b/src/commercetools/testing/payments.py
@@ -12,6 +12,7 @@
 class PaymentsModel(BaseModel):
     _primary_type_name = "payment"
     _resource_schema = schemas.PaymentSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.PaymentDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/reviews.py b/src/commercetools/testing/reviews.py
@@ -9,6 +9,7 @@
 class ReviewModel(abstract.BaseModel):
     _resource_schema = schemas.ReviewSchema
     _primary_type_name = "review"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.ReviewDraft, id: Optional[str] = None
diff --git a/src/commercetools/testing/shipping_methods.py b/src/commercetools/testing/shipping_methods.py
@@ -13,6 +13,7 @@ class ShippingMethodsModel(BaseModel):
     _primary_type_name = "shipping-method"
     _resource_schema = schemas.ShippingMethodSchema
     _schema_update = schemas.ShippingMethodUpdateSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.ShippingMethodDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/shopping_lists.py b/src/commercetools/testing/shopping_lists.py
@@ -10,6 +10,7 @@
 class ShoppingListModel(BaseModel):
     _resource_schema = schemas.ShoppingListSchema
     _primary_type_name = "shopping-list"
+    _unique_values = ["key"]
 
     def _create_line_item_from_draft(
         self, line_item_draft: types.ShoppingListLineItemDraft
diff --git a/src/commercetools/testing/states.py b/src/commercetools/testing/states.py
@@ -11,6 +11,7 @@
 class StateModel(BaseModel):
     _resource_schema = schemas.StateSchema
     _primary_type_name = "state"
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.StateDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/stores.py b/src/commercetools/testing/stores.py
@@ -10,6 +10,7 @@
 class StoresModel(BaseModel):
     _primary_type_name = "store"
     _resource_schema = schemas.StoreSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.StoreDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/subscriptions.py b/src/commercetools/testing/subscriptions.py
@@ -10,6 +10,7 @@
 class SubscriptionsModel(BaseModel):
     _primary_type_name = "subscription"
     _resource_schema = schemas.SubscriptionSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.SubscriptionDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/tax_categories.py b/src/commercetools/testing/tax_categories.py
@@ -20,6 +20,7 @@ def generate_tax_rate_id():
 class TaxCategoryModel(BaseModel):
     _primary_type_name = "tax-category"
     _resource_schema = schemas.TaxCategorySchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.TaxCategoryDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/types.py b/src/commercetools/testing/types.py
@@ -10,6 +10,7 @@
 class TypesModel(BaseModel):
     _primary_type_name = "type"
     _resource_schema = schemas.TypeSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.TypeDraft, id: typing.Optional[str] = None
diff --git a/src/commercetools/testing/zones.py b/src/commercetools/testing/zones.py
@@ -14,6 +14,7 @@
 class ZonesModel(BaseModel):
     _primary_type_name = "zones"
     _resource_schema = schemas.ZoneSchema
+    _unique_values = ["key"]
 
     def _create_from_draft(
         self, draft: types.ZoneDraft, id: typing.Optional[str] = None
