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

package/src/repositories/business-unit.ts

@@ -16,6 +16,7 @@ import type {
 	BusinessUnitChangeParentUnitAction,
 	BusinessUnitChangeStatusAction,
 	BusinessUnitDraft,
+	BusinessUnitKeyReference,
 	BusinessUnitRemoveAddressAction,
 	BusinessUnitRemoveAssociateAction,
 	BusinessUnitRemoveBillingAddressIdAction,
@@ -34,10 +35,12 @@ import type {
 	CompanyDraft,
 	Division,
 	DivisionDraft,
+	InvalidJsonInputError,
 	InvalidOperationError,
 } from "@commercetools/platform-sdk";
 import type { Config } from "#src/config.ts";
 import { CommercetoolsError } from "#src/exceptions.ts";
+import { BusinessUnitDraftSchema } from "#src/schemas/generated/business-unit.ts";
 import { generateRandomString, getBaseResourceProperties } from "../helpers.ts";
 import type { Writable } from "../types.ts";
 import type { UpdateHandlerInterface } from "./abstract.ts";
@@ -58,9 +61,13 @@ export class BusinessUnitRepository extends AbstractResourceRepository<"business
 	constructor(config: Config) {
 		super("business-unit", config);
 		this.actions = new BusinessUnitUpdateHandler(this._storage);
+		this.draftSchema = BusinessUnitDraftSchema;
 	}
 
-	create(context: RepositoryContext, draft: BusinessUnitDraft): BusinessUnit {
+	async create(
+		context: RepositoryContext,
+		draft: BusinessUnitDraft,
+	): Promise<BusinessUnit> {
 		const addresses =
 			draft.addresses?.map((address) => ({
 				...address,
@@ -83,18 +90,32 @@ export class BusinessUnitRepository extends AbstractResourceRepository<"business
 			(i) => addresses[i].id,
 		);
 
+		const stores = draft.stores
+			? await Promise.all(
+					draft.stores.map((s) =>
+						getStoreKeyReference(s, context.projectKey, this._storage),
+					),
+				)
+			: undefined;
+
+		const associates = draft.associates
+			? await Promise.all(
+					draft.associates.map((a) =>
+						createAssociate(a, context.projectKey, this._storage),
+					),
+				)
+			: [];
+
 		const resource = {
-			...getBaseResourceProperties(),
+			...getBaseResourceProperties(context.clientId),
 			key: draft.key,
 			status: draft.status,
-			stores: draft.stores?.map((s) =>
-				getStoreKeyReference(s, context.projectKey, this._storage),
-			),
+			stores,
 			storeMode: draft.storeMode,
 			name: draft.name,
 			contactEmail: draft.contactEmail,
 			addresses: addresses,
-			custom: createCustomFields(
+			custom: await createCustomFields(
 				draft.custom,
 				context.projectKey,
 				this._storage,
@@ -105,38 +126,56 @@ export class BusinessUnitRepository extends AbstractResourceRepository<"business
 			defaultBillingAddressId: defaultBillingAddressId,
 			associateMode: draft.associateMode,
 			approvalRuleMode: draft.approvalRuleMode,
-
-			associates:
-				draft.associates?.map((a) =>
-					createAssociate(a, context.projectKey, this._storage),
-				) ?? [],
+			associates,
 		};
 
 		if (this._isDivisionDraft(draft)) {
+			const parentUnit = await getBusinessUnitKeyReference(
+				draft.parentUnit,
+				context.projectKey,
+				this._storage,
+			);
+			// Look up the parent to determine the topLevelUnit
+			const parent = (await this._storage.getByResourceIdentifier(
+				context.projectKey,
+				parentUnit,
+			)) as BusinessUnit | undefined;
+			const topLevelUnit: BusinessUnitKeyReference = parent?.topLevelUnit ?? {
+				typeId: "business-unit",
+				key: parentUnit.key!,
+			};
 			const division = {
 				...resource,
 				unitType: "Division" as const,
-				parentUnit: getBusinessUnitKeyReference(
-					draft.parentUnit,
-					context.projectKey,
-					this._storage,
-				),
+				parentUnit,
+				topLevelUnit,
 			} as Division;
 
-			this.saveNew(context, division);
+			await this.saveNew(context, division);
 			return division;
 		}
 		if (this._isCompanyDraft(draft)) {
 			const company = {
 				...resource,
 				unitType: "Company" as const,
+				topLevelUnit: {
+					typeId: "business-unit" as const,
+					key: draft.key,
+				},
 			} as Company;
 
-			this.saveNew(context, company);
+			await this.saveNew(context, company);
 			return company;
 		}
 
-		throw new Error("Invalid business unit type");
+		throw new CommercetoolsError<InvalidJsonInputError>(
+			{
+				code: "InvalidJsonInput",
+				message: "Invalid business unit type",
+				detailedErrorMessage: "Invalid business unit type",
+			},
+			400,
+		);
 	}
 
 	private _isCompanyDraft(draft: BusinessUnitDraft): draft is CompanyDraft {
@@ -153,7 +192,7 @@ class BusinessUnitUpdateHandler
 	implements
 		Partial<UpdateHandlerInterface<BusinessUnit, BusinessUnitUpdateAction>>
 {
-	addAddress(
+	async addAddress(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ address }: BusinessUnitAddAddressAction,
@@ -168,12 +207,12 @@ class BusinessUnitUpdateHandler
 		}
 	}
 
-	addAssociate(
+	async addAssociate(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ associate }: BusinessUnitAddAssociateAction,
 	) {
-		const newAssociate = createAssociate(
+		const newAssociate = await createAssociate(
 			associate,
 			context.projectKey,
 			this._storage,
@@ -183,12 +222,12 @@ class BusinessUnitUpdateHandler
 		}
 	}
 
-	addStore(
+	async addStore(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ store }: BusinessUnitAddStoreAction,
 	) {
-		const newStore = getStoreKeyReference(
+		const newStore = await getStoreKeyReference(
 			store,
 			context.projectKey,
 			this._storage,
@@ -249,12 +288,12 @@ class BusinessUnitUpdateHandler
 		resource.name = name;
 	}
 
-	changeParentUnit(
+	async changeParentUnit(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ parentUnit }: BusinessUnitChangeParentUnitAction,
 	) {
-		resource.parentUnit = getBusinessUnitKeyReference(
+		resource.parentUnit = await getBusinessUnitKeyReference(
 			parentUnit,
 			context.projectKey,
 			this._storage,
@@ -269,14 +308,18 @@ class BusinessUnitUpdateHandler
 		resource.status = status;
 	}
 
-	setAssociates(
+	async setAssociates(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ associates }: BusinessUnitSetAssociatesAction,
 	) {
-		const newAssociates = associates
-			.map((a) => createAssociate(a, context.projectKey, this._storage))
-			.filter((a): a is Writable<Associate> => a !== undefined);
+		const newAssociates = (
+			await Promise.all(
+				associates.map((a) =>
+					createAssociate(a, context.projectKey, this._storage),
+				),
+			)
+		).filter((a): a is Writable<Associate> => a !== undefined);
 		resource.associates = newAssociates || undefined;
 	}
 
@@ -290,7 +333,7 @@ class BusinessUnitUpdateHandler
 		);
 	}
 
-	changeAssociate(
+	async changeAssociate(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ associate }: BusinessUnitChangeAssociateAction,
@@ -299,12 +342,16 @@ class BusinessUnitUpdateHandler
 			(a) => a.customer.id === associate.customer.id,
 		);
 		if (existingAssociateIndex === -1) {
-			throw new Error(
-				`Associate with customer id ${associate.customer.id} not found`,
+			throw new CommercetoolsError<InvalidOperationError>(
+				{
+					code: "InvalidOperation",
+					message: `Associate with customer id ${associate.customer.id} not found`,
+				},
+				400,
 			);
 		}
 
-		const newAssociate = createAssociate(
+		const newAssociate = await createAssociate(
 			associate,
 			context.projectKey,
 			this._storage,
@@ -322,20 +369,12 @@ class BusinessUnitUpdateHandler
 		resource.contactEmail = contactEmail;
 	}
 
-	setCustomType(
+	async setCustomType(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ type, fields }: BusinessUnitSetCustomTypeAction,
 	) {
-		if (type) {
-			resource.custom = createCustomFields(
-				{ type, fields },
-				context.projectKey,
-				this._storage,
-			);
-		} else {
-			resource.custom = undefined;
-		}
+		await this._setCustomType(context, resource, { type, fields });
 	}
 
 	setStoreMode(
@@ -378,7 +417,6 @@ class BusinessUnitUpdateHandler
 		if (!resource.shippingAddressIds.includes(address.id)) {
 			resource.shippingAddressIds.push(address.id);
 		}
-		return resource;
 	}
 
 	removeShippingAddressId(
@@ -450,10 +488,7 @@ class BusinessUnitUpdateHandler
 		resource: Writable<BusinessUnit>,
 		{ name, value }: BusinessUnitSetCustomFieldAction,
 	) {
-		if (!resource.custom) {
-			throw new Error("Resource has no custom type");
-		}
-		resource.custom.fields[name] = value;
+		this._setCustomFieldValues(resource, { name, value });
 	}
 
 	setAddressCustomField(
@@ -463,32 +498,49 @@ class BusinessUnitUpdateHandler
 	) {
 		const address = resource.addresses.find((addr) => addr.id === addressId);
 		if (!address) {
-			throw new Error(`Address with id ${addressId} not found`);
+			throw new CommercetoolsError<InvalidOperationError>(
+				{
+					code: "InvalidOperation",
+					message: `Address with id ${addressId} not found`,
+				},
+				400,
+			);
 		}
 		if (!address.custom) {
 			// If the address doesn't have custom fields, we need to initialize them
 			// This might require a type to be set first, but we'll just create minimal structure
-			throw new Error(
-				"Address has no custom type set. Use setAddressCustomType first.",
+			throw new CommercetoolsError<InvalidOperationError>(
+				{
+					code: "InvalidOperation",
+					message:
+						"Address has no custom type set. Use setAddressCustomType first.",
+				},
+				400,
 			);
 		}
 		address.custom.fields[name] = value;
 	}
 
-	setAddressCustomType(
+	async setAddressCustomType(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,
 		{ addressId, type, fields }: BusinessUnitSetAddressCustomTypeAction,
 	) {
 		const address = resource.addresses.find((addr) => addr.id === addressId);
 		if (!address) {
-			throw new Error(`Address with id ${addressId} not found`);
+			throw new CommercetoolsError<InvalidOperationError>(
+				{
+					code: "InvalidOperation",
+					message: `Address with id ${addressId} not found`,
+				},
+				400,
+			);
 		}
 
 		if (!type) {
 			address.custom = undefined;
 		} else {
-			address.custom = createCustomFields(
+			address.custom = await createCustomFields(
 				{ type, fields },
 				context.projectKey,
 				this._storage,