@labdigital/commercetools-mock 1.8.1 → 1.10.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@labdigital/commercetools-mock",
   "author": "Michael van Tellingen",
-  "version": "1.8.1",
+  "version": "1.10.0",
   "license": "MIT",
   "main": "dist/index.cjs",
   "module": "dist/index.js",

package/src/repositories/business-unit.ts

@@ -1,14 +1,215 @@
-import type { BusinessUnit } from '@commercetools/platform-sdk'
+import {
+	type Associate,
+	type BusinessUnit,
+	type BusinessUnitAddAddressAction,
+	type BusinessUnitAddAssociateAction,
+	type BusinessUnitAddStoreAction,
+	type BusinessUnitChangeAddressAction,
+	type BusinessUnitChangeNameAction,
+	type BusinessUnitChangeParentUnitAction,
+	type BusinessUnitDraft,
+	type BusinessUnitSetAssociatesAction,
+	type BusinessUnitSetContactEmailAction,
+	type BusinessUnitSetStoreModeAction,
+	type Company,
+	type Division,
+	BusinessUnitChangeStatusAction,
+	CompanyDraft,
+	DivisionDraft,
+} from '@commercetools/platform-sdk'
 import {
 	AbstractResourceRepository,
 	type RepositoryContext,
 } from './abstract.js'
+import { getBaseResourceProperties } from '../helpers.js'
+import {
+	createAddress,
+	createAssociate,
+	createCustomFields,
+	getBusinessUnitKeyReference,
+	getStoreKeyReference,
+} from './helpers.js'
+import { Writable } from '../types.js'
 
 export class BusinessUnitRepository extends AbstractResourceRepository<'business-unit'> {
 	getTypeId() {
 		return 'business-unit' as const
 	}
-	create(context: RepositoryContext, draft: any): BusinessUnit {
-		throw new Error('Method not implemented.')
+
+	private _isCompanyDraft(draft: BusinessUnitDraft): draft is CompanyDraft {
+		return draft.unitType === 'Company'
+	}
+
+	private _isDivisionDraft(draft: BusinessUnitDraft): draft is DivisionDraft {
+		return draft.unitType === 'Division'
+	}
+
+	create(context: RepositoryContext, draft: BusinessUnitDraft): BusinessUnit {
+		const resource = {
+			...getBaseResourceProperties(),
+			key: draft.key,
+			status: draft.status,
+			stores: draft.stores?.map((s) =>
+				getStoreKeyReference(s, context.projectKey, this._storage)
+			),
+			storeMode: draft.storeMode,
+			name: draft.name,
+			contactEmail: draft.contactEmail,
+			addresses: draft.addresses?.map((a) =>
+				createAddress(a, context.projectKey, this._storage)
+			),
+			custom: createCustomFields(
+				draft.custom,
+				context.projectKey,
+				this._storage
+			),
+			shippingAddressIds: draft.shippingAddresses,
+			defaultShippingAddressId: draft.defaultShippingAddress,
+			billingAddressIds: draft.billingAddresses,
+			associateMode: draft.associateMode,
+			associates: draft.associates?.map((a) =>
+				createAssociate(a, context.projectKey, this._storage)
+			),
+		}
+
+		if (this._isDivisionDraft(draft)) {
+			const division = {
+				...resource,
+				parentUnit: getBusinessUnitKeyReference(
+					draft.parentUnit,
+					context.projectKey,
+					this._storage
+				),
+			} as Division
+
+			this.saveNew(context, division)
+			return division
+		} else if (this._isCompanyDraft(draft)) {
+			const company = resource as Company
+
+			this.saveNew(context, company)
+			return company
+		}
+
+		throw new Error('Invalid business unit type')
+	}
+
+	actions = {
+		addAddress: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ address }: BusinessUnitAddAddressAction
+		) => {
+			const newAddress = createAddress(
+				address,
+				context.projectKey,
+				this._storage
+			)
+			if (newAddress) {
+				resource.addresses.push(newAddress)
+			}
+		},
+		addAssociate: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ associate }: BusinessUnitAddAssociateAction
+		) => {
+			const newAssociate = createAssociate(
+				associate,
+				context.projectKey,
+				this._storage
+			)
+			if (newAssociate) {
+				resource.associates.push(newAssociate)
+			}
+		},
+		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)
+			resource.associates = newAssociates || undefined
+		},
+		setContactEmail: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ contactEmail }: BusinessUnitSetContactEmailAction
+		) => {
+			resource.contactEmail = contactEmail
+		},
+		setStoreMode: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ storeMode }: BusinessUnitSetStoreModeAction
+		) => {
+			resource.storeMode = storeMode
+		},
+		changeAssociateMode: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ storeMode }: BusinessUnitSetStoreModeAction
+		) => {
+			resource.associateMode = storeMode
+		},
+		changeName: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ name }: BusinessUnitChangeNameAction
+		) => {
+			resource.name = name
+		},
+		changeAddress: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ address }: BusinessUnitChangeAddressAction
+		) => {
+			const newAddress = createAddress(
+				address,
+				context.projectKey,
+				this._storage
+			)
+			if (newAddress) {
+				resource.addresses.push(newAddress)
+			}
+		},
+		addStore: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ store }: BusinessUnitAddStoreAction
+		) => {
+			const newStore = getStoreKeyReference(
+				store,
+				context.projectKey,
+				this._storage
+			)
+			if (newStore) {
+				if (!resource.stores) {
+					resource.stores = []
+				}
+
+				resource.stores.push(newStore)
+			}
+		},
+		changeParentUnit: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ parentUnit }: BusinessUnitChangeParentUnitAction
+		) => {
+			resource.parentUnit = getBusinessUnitKeyReference(
+				parentUnit,
+				context.projectKey,
+				this._storage
+			)
+		},
+		changeStatus: (
+			context: RepositoryContext,
+			resource: Writable<BusinessUnit>,
+			{ status }: BusinessUnitChangeStatusAction
+		) => {
+			resource.status = status
+		},
 	}
 }

package/src/repositories/cart-discount.ts

@@ -5,6 +5,7 @@ import type {
 	CartDiscountChangeTargetAction,
 	CartDiscountDraft,
 	CartDiscountSetCustomFieldAction,
+	CartDiscountSetCustomTypeAction,
 	CartDiscountSetDescriptionAction,
 	CartDiscountSetKeyAction,
 	CartDiscountSetValidFromAction,
@@ -162,7 +163,6 @@ export class CartDiscountRepository extends AbstractResourceRepository<'cart-dis
 		) => {
 			resource.target = target
 		},
-
 		setCustomField: (
 			context: RepositoryContext,
 			resource: Writable<CartDiscount>,
@@ -190,5 +190,30 @@ export class CartDiscountRepository extends AbstractResourceRepository<'cart-dis
 				resource.custom.fields[name] = value
 			}
 		},
+		setCustomType: (
+			context: RepositoryContext,
+			resource: Writable<CartDiscount>,
+			{ type, fields }: CartDiscountSetCustomTypeAction
+		) => {
+			if (!type) {
+				resource.custom = undefined
+			} else {
+				const resolvedType = this._storage.getByResourceIdentifier(
+					context.projectKey,
+					type
+				)
+				if (!resolvedType) {
+					throw new Error(`Type ${type} not found`)
+				}
+
+				resource.custom = {
+					type: {
+						typeId: 'type',
+						id: resolvedType.id,
+					},
+					fields: fields || {},
+				}
+			}
+		},
 	}
 }

package/src/repositories/cart.ts

@@ -21,6 +21,7 @@ import type {
 	Price,
 	Product,
 	ProductPagedQueryResponse,
+	CartRemoveDiscountCodeAction,
 	ProductVariant,
 } from '@commercetools/platform-sdk'
 import { v4 as uuidv4 } from 'uuid'
@@ -415,6 +416,15 @@ export class CartRepository extends AbstractResourceRepository<'cart'> {
 				custom: custom,
 			}
 		},
+		removeDiscountCode: (
+			context: RepositoryContext,
+			resource: Writable<Cart>,
+			{ discountCode }: CartRemoveDiscountCodeAction
+		) => {
+			resource.discountCodes = resource.discountCodes.filter(
+				(code) => code.discountCode.id !== discountCode.id
+			)
+		},
 	}
 	draftLineItemtoLineItem = (
 		projectKey: string,

package/src/repositories/helpers.ts

@@ -1,24 +1,34 @@
-import type {
-	Address,
-	BaseAddress,
-	CentPrecisionMoney,
-	CustomFields,
-	CustomFieldsDraft,
-	HighPrecisionMoney,
-	HighPrecisionMoneyDraft,
-	InvalidJsonInputError,
-	Price,
-	PriceDraft,
-	Reference,
-	ReferencedResourceNotFoundError,
-	ResourceIdentifier,
-	Store,
-	StoreKeyReference,
-	StoreReference,
-	StoreResourceIdentifier,
-	Type,
-	TypedMoney,
-	_Money,
+import {
+	AssociateRoleReference,
+	type Address,
+	type Associate,
+	type AssociateDraft,
+	type AssociateRoleAssignment,
+	type AssociateRoleAssignmentDraft,
+	type AssociateRoleKeyReference,
+	type AssociateRoleResourceIdentifier,
+	type BaseAddress,
+	type CentPrecisionMoney,
+	type CustomFields,
+	type CustomFieldsDraft,
+	type HighPrecisionMoney,
+	type HighPrecisionMoneyDraft,
+	type InvalidJsonInputError,
+	type Price,
+	type PriceDraft,
+	type Reference,
+	type ReferencedResourceNotFoundError,
+	type ResourceIdentifier,
+	type Store,
+	type StoreKeyReference,
+	type StoreReference,
+	type StoreResourceIdentifier,
+	type Type,
+	type TypedMoney,
+	type _Money,
+	BusinessUnitResourceIdentifier,
+	BusinessUnitKeyReference,
+	BusinessUnitReference,
 } from '@commercetools/platform-sdk'
 import type { Request } from 'express'
 import { v4 as uuidv4 } from 'uuid'
@@ -219,3 +229,90 @@ export const getRepositoryContext = (request: Request): RepositoryContext => ({
 	projectKey: request.params.projectKey,
 	storeKey: request.params.storeKey,
 })
+
+export const createAssociate = (
+	a: AssociateDraft,
+	projectKey: string,
+	storage: AbstractStorage
+): Associate | undefined => {
+	if (!a) return undefined
+
+	if (!a.associateRoleAssignments) {
+		throw new Error('AssociateRoleAssignments is required')
+	}
+
+	return {
+		customer: getReferenceFromResourceIdentifier(
+			a.customer,
+			projectKey,
+			storage
+		),
+		associateRoleAssignments: a.associateRoleAssignments?.map(
+			(a: AssociateRoleAssignmentDraft): AssociateRoleAssignment => ({
+				associateRole: getAssociateRoleKeyReference(
+					a.associateRole,
+					projectKey,
+					storage
+				),
+				inheritance: a.inheritance as string,
+			})
+		),
+		roles: a.roles as string[],
+	}
+}
+
+export const getAssociateRoleKeyReference = (
+	id: AssociateRoleResourceIdentifier,
+	projectKey: string,
+	storage: AbstractStorage
+): AssociateRoleKeyReference => {
+	if (id.key) {
+		return {
+			typeId: 'associate-role',
+			key: id.key,
+		}
+	}
+
+	const value = getReferenceFromResourceIdentifier<AssociateRoleReference>(
+		id,
+		projectKey,
+		storage
+	)
+
+	if (!value.obj?.key) {
+		throw new Error('No associate-role found for reference')
+	}
+
+	return {
+		typeId: 'associate-role',
+		key: value.obj?.key,
+	}
+}
+
+export const getBusinessUnitKeyReference = (
+	id: BusinessUnitResourceIdentifier,
+	projectKey: string,
+	storage: AbstractStorage
+): BusinessUnitKeyReference => {
+	if (id.key) {
+		return {
+			typeId: 'business-unit',
+			key: id.key,
+		}
+	}
+
+	const value = getReferenceFromResourceIdentifier<BusinessUnitReference>(
+		id,
+		projectKey,
+		storage
+	)
+
+	if (!value.obj?.key) {
+		throw new Error('No business-unit found for reference')
+	}
+
+	return {
+		typeId: 'business-unit',
+		key: value.obj?.key,
+	}
+}

package/src/services/associate-roles.test.ts

@@ -35,7 +35,7 @@ describe('Associate roles query', () => {
 		expect(response.status).toBe(200)
 		expect(response.body.count).toBe(1)
 
-		const associateRole = response.body.results[0] as AssociateRole
+		associateRole = response.body.results[0] as AssociateRole
 
 		expect(associateRole.key).toBe('example-role-associate-role')
 	})

package/src/services/business-units.test.ts

@@ -0,0 +1,42 @@
+import { afterEach, beforeEach, describe, expect, test } from 'vitest'
+import { CommercetoolsMock } from '../ctMock.js'
+import { BusinessUnit } from '@commercetools/platform-sdk'
+import supertest from 'supertest'
+
+describe('Business units query', () => {
+	const ctMock = new CommercetoolsMock()
+	let businessUnit: BusinessUnit | undefined
+
+	beforeEach(async () => {
+		const response = await supertest(ctMock.app)
+			.post('/dummy/business-units')
+			.send({
+				key: 'example-business-unit',
+				status: 'Active',
+				name: 'Example Business Unit',
+				unitType: 'Company',
+			})
+
+		expect(response.status).toBe(201)
+
+		businessUnit = response.body as BusinessUnit
+	})
+
+	afterEach(() => {
+		ctMock.clear()
+	})
+
+	test('no filter', async () => {
+		const response = await supertest(ctMock.app)
+			.get('/dummy/business-units')
+			.query('{}')
+			.send()
+
+		expect(response.status).toBe(200)
+		expect(response.body.count).toBe(1)
+
+		businessUnit = response.body.results[0] as BusinessUnit
+
+		expect(businessUnit.key).toBe('example-business-unit')
+	})
+})

package/src/services/business-units.ts

@@ -0,0 +1,17 @@
+import { Router } from 'express'
+import { BusinessUnitRepository } from '../repositories/business-unit.js'
+import AbstractService from './abstract.js'
+
+export class BusinessUnitServices extends AbstractService {
+	public repository: BusinessUnitRepository
+
+	constructor(parent: Router, repository: BusinessUnitRepository) {
+		super(parent)
+
+		this.repository = repository
+	}
+
+	protected getBasePath(): string {
+		return 'business-units'
+	}
+}

package/src/services/cart-discount.test.ts

@@ -359,4 +359,22 @@ describe('Cart Discounts Update Actions', () => {
 			})
 		expect(response.status).toBe(400)
 	})
+
+	test('remove all custom fields', async () => {
+		assert(cartDiscount, 'cart discount not created')
+
+		const response = await supertest(ctMock.app)
+			.post(`/dummy/cart-discounts/${cartDiscount.id}`)
+			.send({
+				version: 1,
+				actions: [
+					{
+						action: 'setCustomType',
+					},
+				],
+			})
+		expect(response.status).toBe(200)
+		expect(response.body.version).toBe(2)
+		expect(response.body.custom).toBeUndefined()
+	})
 })

package/src/services/category.test.ts

@@ -42,7 +42,7 @@ describe('Categories Query', () => {
 		expect(response.status).toBe(200)
 		expect(response.body.count).toBe(1)
 
-		const category = response.body.results[0] as Category
+		category = response.body.results[0] as Category
 
 		expect(category.name.en).toBe('Top hat')
 	})

package/src/services/customer.ts

@@ -27,6 +27,7 @@ export class CustomerService extends AbstractService {
 				? // @ts-ignore
 				  +request.params.ttlMinutes
 				: 34560
+			// eslint-disable-next-line @typescript-eslint/no-unused-vars
 			const { version, ...rest } = getBaseResourceProperties()
 
 			return response.status(200).send({

package/src/services/index.ts

@@ -1,4 +1,5 @@
-import { AssociateRoleServices } from './associate-roles'
+import { AssociateRoleServices } from './associate-roles.js'
+import { BusinessUnitServices } from './business-units.js'
 import { CartService } from './cart.js'
 import { CartDiscountService } from './cart-discount.js'
 import { CategoryServices } from './category.js'
@@ -32,6 +33,7 @@ import { AttributeGroupService } from './attribute-group.js'
 
 export const createServices = (router: any, repos: any) => ({
 	'associate-role': new AssociateRoleServices(router, repos['associate-role']),
+	'business-unit': new BusinessUnitServices(router, repos['business-unit']),
 	category: new CategoryServices(router, repos['category']),
 	cart: new CartService(router, repos['cart'], repos['order']),
 	'cart-discount': new CartDiscountService(router, repos['cart-discount']),

package/src/services/product.test.ts

@@ -163,7 +163,6 @@ describe('Product', () => {
 describe('Product update actions', () => {
 	const ctMock = new CommercetoolsMock()
 	let productPublished: Product | undefined
-	let productUnpublished: Product | undefined
 
 	beforeEach(async () => {
 		let response
@@ -179,7 +178,6 @@ describe('Product update actions', () => {
 			.send(unpublishedProductDraft)
 
 		expect(response.status).toBe(201)
-		productUnpublished = response.body
 	})
 
 	test('setAttribute masterVariant (staged)', async () => {

package/src/services/project.test.ts

@@ -0,0 +1,52 @@
+import type { Project } from '@commercetools/platform-sdk'
+import supertest from 'supertest'
+import { afterAll, afterEach, beforeAll, describe, expect, test } from 'vitest'
+import { CommercetoolsMock } from '../index.js'
+
+const ctMock = new CommercetoolsMock()
+
+describe('Project', () => {
+	beforeAll(() => {
+		ctMock.start()
+	})
+
+	afterEach(() => {
+		ctMock.clear()
+	})
+
+	afterAll(() => {
+		ctMock.stop()
+	})
+
+	test('Get project by key', async () => {
+		const response = await supertest(ctMock.app).get('/dummy/')
+
+		expect(response.status).toBe(200)
+		expect(response.body).toEqual({
+			version: 1,
+			carts: {
+				countryTaxRateFallbackEnabled: false,
+				deleteDaysAfterLastModification: 90,
+			},
+			countries: [],
+			createdAt: '2018-10-04T11:32:12.603Z',
+			currencies: [],
+			key: 'dummy',
+			languages: [],
+			messages: {
+				deleteDaysAfterCreation: 15,
+				enabled: false,
+			},
+			name: '',
+			searchIndexing: {
+				orders: {
+					status: 'Deactivated',
+				},
+				products: {
+					status: 'Deactivated',
+				},
+			},
+			trialUntil: '2018-12',
+		} as Project)
+	})
+})