@labdigital/commercetools-mock 2.66.0 → 3.0.0-beta.0

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

@@ -10,12 +10,12 @@ describe("Customer repository", () => {
 	const repository = new CustomerRepository(config);
 
 	test("query by lowercaseEmail", async () => {
-		const customer = repository.create(
+		const customer = await repository.create(
 			{ projectKey: "dummy" },
 			{ email: "my-customer-UPPERCASE@email.com" },
 		);
 
-		const result = repository.query(
+		const result = await repository.query(
 			{ projectKey: "dummy" },
 			{ where: [`lowercaseEmail = "my-customer-uppercase@email.com"`] },
 		);
@@ -25,18 +25,18 @@ describe("Customer repository", () => {
 	});
 
 	test("updating lowercaseEmail", async () => {
-		const customer = repository.create(
+		const customer = await repository.create(
 			{ projectKey: "dummy" },
 			{ email: "my-customer-UPPERCASE-v1@email.com" },
 		);
 
-		repository.saveUpdate({ projectKey: "dummy" }, customer.version, {
+		await repository.saveUpdate({ projectKey: "dummy" }, customer.version, {
 			...customer,
 			email: "my-customer-UPPERCASE-v2@email.com",
 			version: customer.version + 1,
 		});
 
-		const result = repository.query(
+		const result = await repository.query(
 			{ projectKey: "dummy" },
 			{ where: [`lowercaseEmail = "my-customer-uppercase-v2@email.com"`] },
 		);
@@ -81,10 +81,10 @@ describe("Customer repository", () => {
 			productSelections: [],
 		};
 
-		storage.add("dummy", "store", store1);
-		storage.add("dummy", "store", store2);
+		await storage.add("dummy", "store", store1);
+		await storage.add("dummy", "store", store2);
 
-		const result = repository.create(
+		const result = await repository.create(
 			{ projectKey: "dummy" },
 			{
 				email: "my-customer@email.com",
@@ -131,9 +131,9 @@ describe("Customer repository", () => {
 			productSelections: [],
 		};
 
-		storage.add("dummy", "store", store1);
+		await storage.add("dummy", "store", store1);
 
-		const result = repository.create(
+		const result = await repository.create(
 			{ projectKey: "dummy" },
 			{
 				email: "my-customer2@email.com",
@@ -155,7 +155,7 @@ describe("Customer repository", () => {
 	});
 
 	test("adding customer without linked stores", async () => {
-		const result = repository.create(
+		const result = await repository.create(
 			{ projectKey: "dummy" },
 			{
 				email: "my-customer-without-stores@email.com",

package/src/repositories/customer/index.ts

@@ -8,6 +8,7 @@ import type {
 	DuplicateFieldError,
 	InvalidInputError,
 	MyCustomerResetPassword,
+	RequiredFieldError,
 	ResourceNotFoundError,
 	Store,
 	StoreKeyReference,
@@ -26,6 +27,7 @@ import {
 	validateEmailVerifyToken,
 	validatePasswordResetToken,
 } from "#src/lib/password.ts";
+import { CustomerDraftSchema } from "#src/schemas/generated/customer.ts";
 import type { ResourceMap, ShallowWritable, Writable } from "#src/types.ts";
 import {
 	AbstractResourceRepository,
@@ -38,13 +40,21 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 	constructor(config: Config) {
 		super("customer", config);
 		this.actions = new CustomerUpdateHandler(config.storage);
+		this.draftSchema = CustomerDraftSchema;
 	}
 
-	create(context: RepositoryContext, draft: CustomerDraft): Customer {
+	async create(
+		context: RepositoryContext,
+		draft: CustomerDraft,
+	): Promise<Customer> {
 		// Check uniqueness
-		const results = this._storage.query(context.projectKey, this.getTypeId(), {
-			where: [`lowercaseEmail="${draft.email.toLowerCase()}"`],
-		});
+		const results = await this._storage.query(
+			context.projectKey,
+			this.getTypeId(),
+			{
+				where: [`lowercaseEmail="${draft.email.toLowerCase()}"`],
+			},
+		);
 		if (results.count > 0) {
 			throw new CommercetoolsError<any>({
 				code: "CustomerAlreadyExists",
@@ -75,7 +85,14 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 			if (addressId < addresses.length) {
 				const id = addresses[addressId].id;
 				if (!id) {
-					throw new Error("Address ID is missing");
+					throw new CommercetoolsError<RequiredFieldError>(
+						{
+							code: "RequiredField",
+							message: "Address ID is missing",
+							field: "addressId",
+						},
+						400,
+					);
 				}
 				return id;
 			}
@@ -112,14 +129,14 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		let storesForCustomer: StoreKeyReference[] = [];
 
 		if (draft.stores && draft.stores.length > 0) {
-			storesForCustomer = this.storeReferenceToStoreKeyReference(
+			storesForCustomer = await this.storeReferenceToStoreKeyReference(
 				draft.stores,
 				context.projectKey,
 			);
 		}
 
 		const resource: Customer = {
-			...getBaseResourceProperties(),
+			...getBaseResourceProperties(context.clientId),
 			key: draft.key,
 			authenticationMode: draft.authenticationMode || "Password",
 			firstName: draft.firstName,
@@ -139,38 +156,43 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 			defaultShippingAddressId: defaultShippingAddressId,
 			shippingAddressIds: shippingAddressIds,
 			billingAddressIds: billingAddressIds,
-			custom: createCustomFields(
+			custom: await createCustomFields(
 				draft.custom,
 				context.projectKey,
 				this._storage,
 			),
 			stores: storesForCustomer,
+			customerGroupAssignments: [],
 		} satisfies unknown as Customer;
 
-		return this.saveNew(context, resource);
+		return await this.saveNew(context, resource);
 	}
 
-	saveUpdate(
+	async saveUpdate(
 		context: RepositoryContext,
 		version: number,
 		resource: ShallowWritable<ResourceMap["customer"]>,
-	): ShallowWritable<ResourceMap["customer"]> {
+	): Promise<ShallowWritable<ResourceMap["customer"]>> {
 		// Also update lowercaseEmail attribute
 		const updatedResource: Customer = {
 			...resource,
 			lowercaseEmail: resource.email.toLowerCase(),
 		} satisfies unknown as Customer;
 
-		return super.saveUpdate(context, version, updatedResource);
+		return await super.saveUpdate(context, version, updatedResource);
 	}
 
-	passwordResetToken(
+	async passwordResetToken(
 		context: RepositoryContext,
 		request: CustomerCreatePasswordResetToken,
-	): CustomerToken {
-		const results = this._storage.query(context.projectKey, this.getTypeId(), {
-			where: [`email="${request.email.toLocaleLowerCase()}"`],
-		});
+	): Promise<CustomerToken> {
+		const results = await this._storage.query(
+			context.projectKey,
+			this.getTypeId(),
+			{
+				where: [`email="${request.email.toLocaleLowerCase()}"`],
+			},
+		);
 		if (results.count === 0) {
 			throw new CommercetoolsError<ResourceNotFoundError>({
 				code: "ResourceNotFound",
@@ -182,7 +204,7 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 
 		const expiresAt = new Date(Date.now() + ttlMinutes * 60 * 1000);
 		const customer = results.results[0] as Customer;
-		const rest = getBaseResourceProperties();
+		const rest = getBaseResourceProperties(context.clientId);
 
 		const token = createPasswordResetToken(customer, expiresAt);
 
@@ -197,7 +219,7 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		};
 	}
 
-	passwordReset(
+	async passwordReset(
 		context: RepositoryContext,
 		resetPassword: CustomerResetPassword | MyCustomerResetPassword,
 	) {
@@ -211,11 +233,11 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 			});
 		}
 
-		const customer = this._storage.get(
+		const customer = (await this._storage.get(
 			context.projectKey,
 			"customer",
 			customerId,
-		) as Writable<Customer> | undefined;
+		)) as Writable<Customer> | undefined;
 
 		if (!customer) {
 			throw new CommercetoolsError<ResourceNotFoundError>({
@@ -228,14 +250,21 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		customer.version += 1;
 
 		// Update storage
-		this._storage.add(context.projectKey, "customer", customer);
+		await this._storage.add(context.projectKey, "customer", customer);
 		return customer;
 	}
 
-	emailToken(context: RepositoryContext, id: string): CustomerToken {
-		const results = this._storage.query(context.projectKey, this.getTypeId(), {
-			where: [`id="${id.toLocaleLowerCase()}"`],
-		});
+	async emailToken(
+		context: RepositoryContext,
+		id: string,
+	): Promise<CustomerToken> {
+		const results = await this._storage.query(
+			context.projectKey,
+			this.getTypeId(),
+			{
+				where: [`id="${id.toLocaleLowerCase()}"`],
+			},
+		);
 		if (results.count === 0) {
 			throw new CommercetoolsError<ResourceNotFoundError>({
 				code: "ResourceNotFound",
@@ -244,7 +273,7 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		}
 		const expiresAt = new Date(Date.now() + 30 * 60);
 		const customer = results.results[0] as Customer;
-		const rest = getBaseResourceProperties();
+		const rest = getBaseResourceProperties(context.clientId);
 
 		const token = createEmailVerifyToken(customer);
 		return {
@@ -258,7 +287,7 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		};
 	}
 
-	emailTokenConfirm(
+	async emailTokenConfirm(
 		context: RepositoryContext,
 		request: { tokenValue: string },
 	) {
@@ -270,11 +299,11 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 			});
 		}
 
-		const customer = this._storage.get(
+		const customer = (await this._storage.get(
 			context.projectKey,
 			"customer",
 			customerId,
-		) as Writable<Customer> | undefined;
+		)) as Writable<Customer> | undefined;
 
 		if (!customer) {
 			throw new CommercetoolsError<ResourceNotFoundError>({
@@ -287,14 +316,14 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		customer.version += 1;
 
 		// Update storage
-		this._storage.add(context.projectKey, "customer", customer);
+		await this._storage.add(context.projectKey, "customer", customer);
 		return customer;
 	}
 
-	private storeReferenceToStoreKeyReference(
+	private async storeReferenceToStoreKeyReference(
 		draftStores: StoreResourceIdentifier[],
 		projectKey: string,
-	): StoreKeyReference[] {
+	): Promise<StoreKeyReference[]> {
 		const storeIds = draftStores
 			.map((storeReference) => storeReference.id)
 			.filter(Boolean);
@@ -302,9 +331,10 @@ export class CustomerRepository extends AbstractResourceRepository<"customer"> {
 		let stores: Store[] = [];
 
 		if (storeIds.length > 0) {
-			stores = this._storage.query(projectKey, "store", {
+			const storeResult = await this._storage.query(projectKey, "store", {
 				where: storeIds.map((id) => `id="${id}"`),
-			}).results;
+			});
+			stores = storeResult.results;
 
 			if (storeIds.length !== stores.length) {
 				throw new CommercetoolsError<ResourceNotFoundError>({

package/src/repositories/customer-group.test.ts

@@ -5,7 +5,7 @@ import type {
 	CustomerGroupSetCustomTypeAction,
 	CustomerGroupSetKeyAction,
 } from "@commercetools/platform-sdk";
-import { describe, expect, test } from "vitest";
+import { beforeAll, describe, expect, test } from "vitest";
 import type { Config } from "#src/config.ts";
 import { getBaseResourceProperties } from "#src/helpers.ts";
 import { InMemoryStorage } from "#src/storage/index.ts";
@@ -16,32 +16,34 @@ describe("CustomerGroup Repository", () => {
 	const config: Config = { storage, strict: false };
 	const repository = new CustomerGroupRepository(config);
 
-	// Add a custom type for testing
-	storage.add("dummy", "type", {
-		...getBaseResourceProperties(),
-		id: "custom-type-id",
-		key: "custom-type-key",
-		name: { "en-US": "Custom Type" },
-		resourceTypeIds: ["customer-group"],
-		fieldDefinitions: [
-			{
-				name: "description",
-				label: { "en-US": "Description" },
-				required: false,
-				type: { name: "String" },
-				inputHint: "SingleLine",
-			},
-		],
+	beforeAll(async () => {
+		// Add a custom type for testing
+		await storage.add("dummy", "type", {
+			...getBaseResourceProperties(),
+			id: "custom-type-id",
+			key: "custom-type-key",
+			name: { "en-US": "Custom Type" },
+			resourceTypeIds: ["customer-group"],
+			fieldDefinitions: [
+				{
+					name: "description",
+					label: { "en-US": "Description" },
+					required: false,
+					type: { name: "String" },
+					inputHint: "SingleLine",
+				},
+			],
+		});
 	});
 
-	test("create customer group", () => {
+	test("create customer group", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "premium-customers",
 			groupName: "Premium Customers",
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const result = repository.create(ctx, draft);
+		const result = await repository.create(ctx, draft);
 
 		expect(result.id).toBeDefined();
 		expect(result.version).toBe(1);
@@ -50,12 +52,12 @@ describe("CustomerGroup Repository", () => {
 		expect(result.custom).toBeUndefined();
 
 		// Test that the customer group is stored
-		const items = repository.query(ctx);
+		const items = await repository.query(ctx);
 		expect(items.count).toBe(1);
 		expect(items.results[0].id).toBe(result.id);
 	});
 
-	test("create customer group with custom fields", () => {
+	test("create customer group with custom fields", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "vip-customers",
 			groupName: "VIP Customers",
@@ -71,7 +73,7 @@ describe("CustomerGroup Repository", () => {
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const result = repository.create(ctx, draft);
+		const result = await repository.create(ctx, draft);
 
 		expect(result.id).toBeDefined();
 		expect(result.key).toBe(draft.key);
@@ -80,16 +82,16 @@ describe("CustomerGroup Repository", () => {
 		expect(result.custom?.fields.description).toBe("Very important customers");
 	});
 
-	test("update customer group - changeName", () => {
+	test("update customer group - changeName", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "test-customers",
 			groupName: "Test Customers",
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const customerGroup = repository.create(ctx, draft);
+		const customerGroup = await repository.create(ctx, draft);
 
-		const result = repository.processUpdateActions(
+		const result = await repository.processUpdateActions(
 			ctx,
 			customerGroup,
 			customerGroup.version,
@@ -105,16 +107,16 @@ describe("CustomerGroup Repository", () => {
 		expect(result.version).toBe(customerGroup.version + 1);
 	});
 
-	test("update customer group - setKey", () => {
+	test("update customer group - setKey", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "test-customers-2",
 			groupName: "Test Customers 2",
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const customerGroup = repository.create(ctx, draft);
+		const customerGroup = await repository.create(ctx, draft);
 
-		const result = repository.processUpdateActions(
+		const result = await repository.processUpdateActions(
 			ctx,
 			customerGroup,
 			customerGroup.version,
@@ -130,17 +132,17 @@ describe("CustomerGroup Repository", () => {
 		expect(result.version).toBe(customerGroup.version + 1);
 	});
 
-	test("update customer group - setCustomType", () => {
+	test("update customer group - setCustomType", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "test-customers-3",
 			groupName: "Test Customers 3",
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const customerGroup = repository.create(ctx, draft);
+		const customerGroup = await repository.create(ctx, draft);
 
 		// Set custom type
-		const result = repository.processUpdateActions(
+		const result = await repository.processUpdateActions(
 			ctx,
 			customerGroup,
 			customerGroup.version,
@@ -163,7 +165,7 @@ describe("CustomerGroup Repository", () => {
 		expect(result.version).toBe(customerGroup.version + 1);
 
 		// Remove custom type
-		const result2 = repository.processUpdateActions(
+		const result2 = await repository.processUpdateActions(
 			ctx,
 			result,
 			result.version,
@@ -178,7 +180,7 @@ describe("CustomerGroup Repository", () => {
 		expect(result2.version).toBe(result.version + 1);
 	});
 
-	test("update customer group - setCustomField", () => {
+	test("update customer group - setCustomField", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "test-customers-4",
 			groupName: "Test Customers 4",
@@ -194,10 +196,10 @@ describe("CustomerGroup Repository", () => {
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const customerGroup = repository.create(ctx, draft);
+		const customerGroup = await repository.create(ctx, draft);
 
 		// Update custom field
-		const result = repository.processUpdateActions(
+		const result = await repository.processUpdateActions(
 			ctx,
 			customerGroup,
 			customerGroup.version,
@@ -214,7 +216,7 @@ describe("CustomerGroup Repository", () => {
 		expect(result.version).toBe(customerGroup.version + 1);
 
 		// Remove custom field
-		const result2 = repository.processUpdateActions(
+		const result2 = await repository.processUpdateActions(
 			ctx,
 			result,
 			result.version,
@@ -231,32 +233,32 @@ describe("CustomerGroup Repository", () => {
 		expect(result2.version).toBe(result.version + 1);
 	});
 
-	test("get and delete customer group", () => {
+	test("get and delete customer group", async () => {
 		const draft: CustomerGroupDraft = {
 			key: "delete-test",
 			groupName: "Delete Test",
 		};
 
 		const ctx = { projectKey: "dummy" };
-		const customerGroup = repository.create(ctx, draft);
+		const customerGroup = await repository.create(ctx, draft);
 
 		// Test get
-		const retrieved = repository.get(ctx, customerGroup.id);
+		const retrieved = await repository.get(ctx, customerGroup.id);
 		expect(retrieved).toBeDefined();
 		expect(retrieved?.id).toBe(customerGroup.id);
 
 		// Test getByKey
-		const retrievedByKey = repository.getByKey(ctx, customerGroup.key!);
+		const retrievedByKey = await repository.getByKey(ctx, customerGroup.key!);
 		expect(retrievedByKey).toBeDefined();
 		expect(retrievedByKey?.key).toBe(customerGroup.key);
 
 		// Test delete
-		const deleted = repository.delete(ctx, customerGroup.id);
+		const deleted = await repository.delete(ctx, customerGroup.id);
 		expect(deleted).toBeDefined();
 		expect(deleted?.id).toBe(customerGroup.id);
 
 		// Verify it's deleted
-		const notFound = repository.get(ctx, customerGroup.id);
+		const notFound = await repository.get(ctx, customerGroup.id);
 		expect(notFound).toBeNull();
 	});
 });

package/src/repositories/customer-group.ts

@@ -8,6 +8,7 @@ import type {
 	CustomerGroupUpdateAction,
 } from "@commercetools/platform-sdk";
 import type { Config } from "#src/config.ts";
+import { CustomerGroupDraftSchema } from "#src/schemas/generated/customer-group.ts";
 import { getBaseResourceProperties } from "../helpers.ts";
 import type { Writable } from "../types.ts";
 import type { UpdateHandlerInterface } from "./abstract.ts";
@@ -22,20 +23,24 @@ export class CustomerGroupRepository extends AbstractResourceRepository<"custome
 	constructor(config: Config) {
 		super("customer-group", config);
 		this.actions = new CustomerGroupUpdateHandler(config.storage);
+		this.draftSchema = CustomerGroupDraftSchema;
 	}
 
-	create(context: RepositoryContext, draft: CustomerGroupDraft): CustomerGroup {
+	async create(
+		context: RepositoryContext,
+		draft: CustomerGroupDraft,
+	): Promise<CustomerGroup> {
 		const resource: CustomerGroup = {
-			...getBaseResourceProperties(),
+			...getBaseResourceProperties(context.clientId),
 			key: draft.key,
 			name: draft.groupName,
-			custom: createCustomFields(
+			custom: await createCustomFields(
 				draft.custom,
 				context.projectKey,
 				this._storage,
 			),
 		};
-		return this.saveNew(context, resource);
+		return await this.saveNew(context, resource);
 	}
 }
 
@@ -56,30 +61,15 @@ class CustomerGroupUpdateHandler
 		resource: Writable<CustomerGroup>,
 		{ name, value }: CustomerGroupSetCustomFieldAction,
 	) {
-		if (!resource.custom) {
-			return;
-		}
-		if (value === null) {
-			delete resource.custom.fields[name];
-		} else {
-			resource.custom.fields[name] = value;
-		}
+		this._setCustomFieldValues(resource, { name, value });
 	}
 
-	setCustomType(
+	async setCustomType(
 		context: RepositoryContext,
 		resource: Writable<CustomerGroup>,
 		{ type, fields }: CustomerGroupSetCustomTypeAction,
 	) {
-		if (type) {
-			resource.custom = createCustomFields(
-				{ type, fields },
-				context.projectKey,
-				this._storage,
-			);
-		} else {
-			resource.custom = undefined;
-		}
+		await this._setCustomType(context, resource, { type, fields });
 	}
 
 	setKey(

package/src/repositories/discount-code/actions.ts

@@ -18,7 +18,6 @@ import type {
 import type { Writable } from "#src/types.ts";
 import type { UpdateHandlerInterface } from "../abstract.ts";
 import { AbstractUpdateHandler, type RepositoryContext } from "../abstract.ts";
-import { createCustomFields } from "../helpers.ts";
 
 export class DiscountCodeUpdateHandler
 	extends AbstractUpdateHandler
@@ -59,30 +58,15 @@ export class DiscountCodeUpdateHandler
 		resource: Writable<DiscountCode>,
 		{ name, value }: DiscountCodeSetCustomFieldAction,
 	) {
-		if (!resource.custom) {
-			return;
-		}
-		if (value === null) {
-			delete resource.custom.fields[name];
-		} else {
-			resource.custom.fields[name] = value;
-		}
+		this._setCustomFieldValues(resource, { name, value });
 	}
 
-	setCustomType(
+	async setCustomType(
 		context: RepositoryContext,
 		resource: Writable<DiscountCode>,
 		{ type, fields }: DiscountCodeSetCustomTypeAction,
 	) {
-		if (type) {
-			resource.custom = createCustomFields(
-				{ type, fields },
-				context.projectKey,
-				this._storage,
-			);
-		} else {
-			resource.custom = undefined;
-		}
+		await this._setCustomType(context, resource, { type, fields });
 	}
 
 	setDescription(