@labdigital/commercetools-mock 1.0.0 → 1.1.0

package/dist/index.global.js

@@ -48024,6 +48024,122 @@ ${this.pendingMocks().join("\n")}`
           }
           checkForStagedChanges(resource);
           return resource;
+        },
+        addPrice: (context, resource, { variantId, sku, price, staged }) => {
+          const addVariantPrice = (data) => {
+            const { variant, isMasterVariant, variantIndex } = getVariant(
+              data,
+              variantId,
+              sku
+            );
+            if (!variant) {
+              throw new Error(
+                `Variant with id ${variantId} or sku ${sku} not found on product ${resource.id}`
+              );
+            }
+            if (variant.prices === void 0) {
+              variant.prices = [priceFromDraft(price)];
+            } else {
+              variant.prices.push(priceFromDraft(price));
+            }
+            if (isMasterVariant) {
+              data.masterVariant = variant;
+            } else {
+              data.variants[variantIndex] = variant;
+            }
+          };
+          const onlyStaged = staged !== void 0 ? staged : true;
+          addVariantPrice(resource.masterData.staged);
+          if (!onlyStaged) {
+            addVariantPrice(resource.masterData.current);
+          }
+          checkForStagedChanges(resource);
+          return resource;
+        },
+        changePrice: (context, resource, { priceId, price, staged }) => {
+          const changeVariantPrice = (data) => {
+            var _a;
+            const allVariants = [data.masterVariant, ...data.variants ?? []];
+            const priceVariant = allVariants.find(
+              (variant2) => {
+                var _a2;
+                return (_a2 = variant2.prices) == null ? void 0 : _a2.some((x) => x.id === priceId);
+              }
+            );
+            if (!priceVariant) {
+              throw new Error(
+                `Price with id ${priceId} not found on product ${resource.id}`
+              );
+            }
+            const { variant, isMasterVariant, variantIndex } = getVariant(
+              data,
+              priceVariant.id,
+              priceVariant.sku
+            );
+            if (!variant) {
+              throw new Error(
+                `Variant with id ${priceVariant.id} or sku ${priceVariant.sku} not found on product ${resource.id}`
+              );
+            }
+            variant.prices = (_a = variant.prices) == null ? void 0 : _a.map((x) => {
+              if (x.id === priceId) {
+                return { ...x, ...price };
+              }
+              return x;
+            });
+            if (isMasterVariant) {
+              data.masterVariant = variant;
+            } else {
+              data.variants[variantIndex] = variant;
+            }
+          };
+          const onlyStaged = staged !== void 0 ? staged : true;
+          changeVariantPrice(resource.masterData.staged);
+          if (!onlyStaged) {
+            changeVariantPrice(resource.masterData.current);
+          }
+          checkForStagedChanges(resource);
+          return resource;
+        },
+        removePrice: (context, resource, { priceId, staged }) => {
+          const removeVariantPrice = (data) => {
+            var _a;
+            const allVariants = [data.masterVariant, ...data.variants ?? []];
+            const priceVariant = allVariants.find(
+              (variant2) => {
+                var _a2;
+                return (_a2 = variant2.prices) == null ? void 0 : _a2.some((x) => x.id === priceId);
+              }
+            );
+            if (!priceVariant) {
+              throw new Error(
+                `Price with id ${priceId} not found on product ${resource.id}`
+              );
+            }
+            const { variant, isMasterVariant, variantIndex } = getVariant(
+              data,
+              priceVariant.id,
+              priceVariant.sku
+            );
+            if (!variant) {
+              throw new Error(
+                `Variant with id ${priceVariant.id} or sku ${priceVariant.sku} not found on product ${resource.id}`
+              );
+            }
+            variant.prices = (_a = variant.prices) == null ? void 0 : _a.filter((x) => x.id !== priceId);
+            if (isMasterVariant) {
+              data.masterVariant = variant;
+            } else {
+              data.variants[variantIndex] = variant;
+            }
+          };
+          const onlyStaged = staged !== void 0 ? staged : true;
+          removeVariantPrice(resource.masterData.staged);
+          if (!onlyStaged) {
+            removeVariantPrice(resource.masterData.current);
+          }
+          checkForStagedChanges(resource);
+          return resource;
         }
       };
     }
@@ -48117,6 +48233,7 @@ ${this.pendingMocks().join("\n")}`
   };
   var priceFromDraft = (draft) => ({
     id: v4(),
+    country: draft.country,
     value: {
       currencyCode: draft.value.currencyCode,
       centAmount: draft.value.centAmount,