feat(custom-fields): Added custom field mutations

Author: demeyerthom

.changeset/pink-hornets-compete.md

@@ -0,0 +1,5 @@
+---
+"@labdigital/commercetools-mock": patch
+---
+
+Added missing mutations for custom fields on associate roles, product selections and business units

src/repositories/associate-role.ts

@@ -5,6 +5,7 @@ import type {
 	AssociateRoleDraft,
 	AssociateRoleRemovePermissionAction,
 	AssociateRoleSetCustomFieldAction,
+	AssociateRoleSetCustomTypeAction,
 	AssociateRoleSetNameAction,
 	AssociateRoleSetPermissionsAction,
 	AssociateRoleUpdateAction,
@@ -107,6 +108,22 @@ class AssociateRoleUpdateHandler
 		}
 	}
 
+	setCustomType(
+		context: RepositoryContext,
+		resource: Writable<AssociateRole>,
+		{ type, fields }: AssociateRoleSetCustomTypeAction,
+	) {
+		if (type) {
+			resource.custom = createCustomFields(
+				{ type, fields },
+				context.projectKey,
+				this._storage,
+			);
+		} else {
+			resource.custom = undefined;
+		}
+	}
+
 	setName(
 		context: RepositoryContext,
 		resource: Writable<AssociateRole>,

src/repositories/business-unit.ts

@@ -2,6 +2,7 @@ import type {
 	BusinessUnitChangeApprovalRuleModeAction,
 	BusinessUnitChangeAssociateModeAction,
 	BusinessUnitChangeStatusAction,
+	BusinessUnitSetCustomTypeAction,
 	BusinessUnitUpdateAction,
 	CompanyDraft,
 	DivisionDraft,
@@ -260,6 +261,22 @@ class BusinessUnitUpdateHandler
 		resource.contactEmail = contactEmail;
 	}
 
+	setCustomType(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ type, fields }: BusinessUnitSetCustomTypeAction,
+	) {
+		if (type) {
+			resource.custom = createCustomFields(
+				{ type, fields },
+				context.projectKey,
+				this._storage,
+			);
+		} else {
+			resource.custom = undefined;
+		}
+	}
+
 	setStoreMode(
 		context: RepositoryContext,
 		resource: Writable<BusinessUnit>,

src/repositories/product-selection.ts

@@ -2,8 +2,10 @@ import type {
 	ProductSelection,
 	ProductSelectionChangeNameAction,
 	ProductSelectionDraft,
+	ProductSelectionSetCustomTypeAction,
 	ProductSelectionUpdateAction,
 } from "@commercetools/platform-sdk";
+import { createCustomFields } from "~src/repositories/helpers";
 import { getBaseResourceProperties } from "../helpers";
 import type { AbstractStorage } from "../storage/abstract";
 import type { Writable } from "../types";
@@ -45,4 +47,20 @@ class ProductSelectionUpdateHandler
 	) {
 		resource.name = name;
 	}
+
+	setCustomType(
+		context: RepositoryContext,
+		resource: Writable<ProductSelection>,
+		{ type, fields }: ProductSelectionSetCustomTypeAction,
+	) {
+		if (type) {
+			resource.custom = createCustomFields(
+				{ type, fields },
+				context.projectKey,
+				this._storage,
+			);
+		} else {
+			resource.custom = undefined;
+		}
+	}
 }