@labdigital/commercetools-mock 2.38.0 → 2.40.0

package/dist/index.js

@@ -3171,6 +3171,11 @@ var CartDiscountUpdateHandler = class extends AbstractUpdateHandler {
   setKey(context, resource, { key }) {
     resource.key = key;
   }
+  setStores(context, resource, { stores }) {
+    resource.stores = stores?.map(
+      (s) => getStoreKeyReference(s, context.projectKey, this._storage)
+    );
+  }
   setValidFrom(context, resource, { validFrom }) {
     resource.validFrom = validFrom;
   }
@@ -3905,6 +3910,23 @@ var CustomerRepository = class extends AbstractResourceRepository {
     const billingAddressIds = draft.billingAddresses?.map(
       (addressId) => lookupAdressId(addresses, addressId)
     ) ?? [];
+    let storesForCustomer = [];
+    if (draft.stores) {
+      const storeIds = draft.stores.map((storeReference) => storeReference.id).filter(Boolean);
+      const stores = this._storage.query(context.projectKey, "store", {
+        where: storeIds.map((id) => `id="${id}"`)
+      }).results;
+      if (storeIds.length !== stores.length) {
+        throw new CommercetoolsError({
+          code: "ResourceNotFound",
+          message: `Store with ID '${storeIds.find((id) => !stores.some((store) => store.id === id))}' was not found.`
+        });
+      }
+      storesForCustomer = draft.stores.map((storeReference) => ({
+        typeId: "store",
+        key: storeReference.key ?? stores.find((store) => store.id === storeReference.id)?.key
+      }));
+    }
     const resource = {
       ...getBaseResourceProperties(),
       key: draft.key,
@@ -3930,7 +3952,7 @@ var CustomerRepository = class extends AbstractResourceRepository {
         context.projectKey,
         this._storage
       ),
-      stores: []
+      stores: storesForCustomer
     };
     return this.saveNew(context, resource);
   }