@labdigital/commercetools-mock 2.38.0 → 2.40.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@labdigital/commercetools-mock",
-  "version": "2.38.0",
+  "version": "2.40.0",
   "license": "MIT",
   "author": "Michael van Tellingen",
   "type": "module",

package/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>,

package/src/repositories/customer/index.test.ts

@@ -0,0 +1,75 @@
+import type { Store } from "@commercetools/platform-sdk";
+import { describe, expect, test } from "vitest";
+import { InMemoryStorage } from "~src/storage";
+import { CustomerRepository } from "./index";
+
+describe("Order repository", () => {
+	const storage = new InMemoryStorage();
+	const repository = new CustomerRepository(storage);
+
+	test("adding stores to customer", async () => {
+		const store1: Store = {
+			id: "d0016081-e9af-48a7-8133-1f04f340a335",
+			key: "store-1",
+			name: {
+				en: "Store 1",
+			},
+			version: 1,
+			createdAt: "2021-09-02T12:23:30.036Z",
+			lastModifiedAt: "2021-09-02T12:23:30.546Z",
+			languages: [],
+			distributionChannels: [],
+			countries: [],
+			supplyChannels: [],
+			productSelections: [],
+		};
+
+		const store2: Store = {
+			id: "6dac7d6d-2a48-4705-aa8b-17b0124a499a",
+			key: "store-2",
+			name: {
+				en: "Store 2",
+			},
+			version: 1,
+			createdAt: "2021-09-02T12:23:30.036Z",
+			lastModifiedAt: "2021-09-02T12:23:30.546Z",
+			languages: [],
+			distributionChannels: [],
+			countries: [],
+			supplyChannels: [],
+			productSelections: [],
+		};
+
+		storage.add("dummy", "store", store1);
+		storage.add("dummy", "store", store2);
+
+		const result = repository.create(
+			{ projectKey: "dummy" },
+			{
+				email: "my-customer@email.com",
+				stores: [
+					{
+						typeId: "store",
+						id: store1.id,
+					},
+					{
+						typeId: "store",
+						key: store2.key,
+					},
+				],
+			},
+		);
+
+		expect(result?.stores).toHaveLength(2);
+		expect(result?.stores).toEqual([
+			{
+				typeId: "store",
+				key: store1.key,
+			},
+			{
+				typeId: "store",
+				key: store2.key,
+			},
+		]);
+	});
+});

package/src/repositories/customer/index.ts

@@ -9,6 +9,7 @@ import type {
 	InvalidInputError,
 	MyCustomerResetPassword,
 	ResourceNotFoundError,
+	StoreKeyReference,
 } from "@commercetools/platform-sdk";
 import { CommercetoolsError } from "~src/exceptions";
 import { generateRandomString, getBaseResourceProperties } from "~src/helpers";
@@ -102,6 +103,33 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 				lookupAdressId(addresses, addressId),
 			) ?? [];
 
+		let storesForCustomer: StoreKeyReference[] = [];
+
+		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<ResourceNotFoundError>({
+					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 as string),
+			}));
+		}
+
 		const resource: Customer = {
 			...getBaseResourceProperties(),
 			key: draft.key,
@@ -127,7 +155,7 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 				context.projectKey,
 				this._storage,
 			),
-			stores: [],
+			stores: storesForCustomer,
 		};
 		return this.saveNew(context, resource);
 	}