feat(cart-discount): added setStores support (#242)

labd · Oct 23, 2024 · c970b6e · c970b6e
1 parent bd17654
commit c970b6e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/.changeset/shaggy-candles-nail.md b/.changeset/shaggy-candles-nail.md
@@ -0,0 +1,5 @@
+---
+"@labdigital/commercetools-mock": minor
+---
+
+Added cart discount set stores support
diff --git a/src/repositories/cart-discount/actions.ts b/src/repositories/cart-discount/actions.ts
@@ -7,6 +7,7 @@ import type {
 	CartDiscountSetCustomTypeAction,
 	CartDiscountSetDescriptionAction,
 	CartDiscountSetKeyAction,
+	CartDiscountSetStoresAction,
 	CartDiscountSetValidFromAction,
 	CartDiscountSetValidFromAndUntilAction,
 	CartDiscountSetValidUntilAction,
@@ -18,6 +19,7 @@ import type { UpdateHandlerInterface } from "../abstract";
 import { AbstractUpdateHandler, type RepositoryContext } from "../abstract";
 
 import { CommercetoolsError } from "~src/exceptions";
+import { getStoreKeyReference } from "~src/repositories/helpers";
 
 export class CartDiscountUpdateHandler
 	extends AbstractUpdateHandler
@@ -118,6 +120,16 @@ export class CartDiscountUpdateHandler
 		resource.key = key;
 	}
 
+	setStores(
+		context: RepositoryContext,
+		resource: Writable<CartDiscount>,
+		{ stores }: CartDiscountSetStoresAction,
+	) {
+		resource.stores = stores?.map((s) =>
+			getStoreKeyReference(s, context.projectKey, this._storage),
+		);
+	}
+
 	setValidFrom(
 		context: RepositoryContext,
 		resource: Writable<CartDiscount>,