@labdigital/commercetools-mock 1.5.0 → 1.6.0

package/src/repositories/order.ts

@@ -1,434 +1,434 @@
 import type {
-  Cart,
-  CartReference,
-  CustomLineItem,
-  CustomLineItemDraft,
-  GeneralError,
-  LineItem,
-  LineItemImportDraft,
-  Order,
-  OrderAddPaymentAction,
-  OrderChangeOrderStateAction,
-  OrderChangePaymentStateAction,
-  OrderFromCartDraft,
-  OrderImportDraft,
-  OrderSetBillingAddressAction,
-  OrderSetCustomerEmailAction,
-  OrderSetCustomFieldAction,
-  OrderSetCustomTypeAction,
-  OrderSetLocaleAction,
-  OrderSetOrderNumberAction,
-  OrderSetShippingAddressAction,
-  OrderSetStoreAction,
-  OrderTransitionStateAction,
-  Product,
-  ProductPagedQueryResponse,
-  ProductVariant,
-  State,
-  Store,
+	Cart,
+	CartReference,
+	CustomLineItem,
+	CustomLineItemDraft,
+	GeneralError,
+	LineItem,
+	LineItemImportDraft,
+	Order,
+	OrderAddPaymentAction,
+	OrderChangeOrderStateAction,
+	OrderChangePaymentStateAction,
+	OrderFromCartDraft,
+	OrderImportDraft,
+	OrderSetBillingAddressAction,
+	OrderSetCustomerEmailAction,
+	OrderSetCustomFieldAction,
+	OrderSetCustomTypeAction,
+	OrderSetLocaleAction,
+	OrderSetOrderNumberAction,
+	OrderSetShippingAddressAction,
+	OrderSetStoreAction,
+	OrderTransitionStateAction,
+	Product,
+	ProductPagedQueryResponse,
+	ProductVariant,
+	State,
+	Store,
 } from '@commercetools/platform-sdk'
 import assert from 'assert'
 import { CommercetoolsError } from '../exceptions.js'
 import { getBaseResourceProperties } from '../helpers.js'
 import type { Writable } from '../types.js'
 import {
-  AbstractResourceRepository,
-  type QueryParams,
-  RepositoryContext,
+	AbstractResourceRepository,
+	type QueryParams,
+	RepositoryContext,
 } from './abstract.js'
 import {
-  createAddress,
-  createCentPrecisionMoney,
-  createCustomFields,
-  createPrice,
-  createTypedMoney,
-  resolveStoreReference,
+	createAddress,
+	createCentPrecisionMoney,
+	createCustomFields,
+	createPrice,
+	createTypedMoney,
+	resolveStoreReference,
 } from './helpers.js'
 
 export class OrderRepository extends AbstractResourceRepository<'order'> {
-  getTypeId() {
-    return 'order' as const
-  }
+	getTypeId() {
+		return 'order' as const
+	}
 
-  create(context: RepositoryContext, draft: OrderFromCartDraft): Order {
-    assert(draft.cart, 'draft.cart is missing')
-    return this.createFromCart(
-      context,
-      {
-        id: draft.cart.id!,
-        typeId: 'cart',
-      },
-      draft.orderNumber
-    )
-  }
+	create(context: RepositoryContext, draft: OrderFromCartDraft): Order {
+		assert(draft.cart, 'draft.cart is missing')
+		return this.createFromCart(
+			context,
+			{
+				id: draft.cart.id!,
+				typeId: 'cart',
+			},
+			draft.orderNumber
+		)
+	}
 
-  createFromCart(
-    context: RepositoryContext,
-    cartReference: CartReference,
-    orderNumber?: string
-  ) {
-    const cart = this._storage.getByResourceIdentifier(
-      context.projectKey,
-      cartReference
-    ) as Cart | null
-    if (!cart) {
-      throw new Error('Cannot find cart')
-    }
+	createFromCart(
+		context: RepositoryContext,
+		cartReference: CartReference,
+		orderNumber?: string
+	) {
+		const cart = this._storage.getByResourceIdentifier(
+			context.projectKey,
+			cartReference
+		) as Cart | null
+		if (!cart) {
+			throw new Error('Cannot find cart')
+		}
 
-    const resource: Order = {
-      ...getBaseResourceProperties(),
-      orderNumber,
-      cart: cartReference,
-      orderState: 'Open',
-      lineItems: [],
-      customLineItems: [],
-      totalPrice: cart.totalPrice,
-      refusedGifts: [],
-      origin: 'Customer',
-      syncInfo: [],
-      shippingMode: cart.shippingMode,
-      shipping: cart.shipping,
-      store: context.storeKey
-        ? {
-            key: context.storeKey,
-            typeId: 'store',
-          }
-        : undefined,
-      lastMessageSequenceNumber: 0,
-    }
-    this.saveNew(context, resource)
-    return resource
-  }
+		const resource: Order = {
+			...getBaseResourceProperties(),
+			orderNumber,
+			cart: cartReference,
+			orderState: 'Open',
+			lineItems: [],
+			customLineItems: [],
+			totalPrice: cart.totalPrice,
+			refusedGifts: [],
+			origin: 'Customer',
+			syncInfo: [],
+			shippingMode: cart.shippingMode,
+			shipping: cart.shipping,
+			store: context.storeKey
+				? {
+						key: context.storeKey,
+						typeId: 'store',
+				  }
+				: undefined,
+			lastMessageSequenceNumber: 0,
+		}
+		this.saveNew(context, resource)
+		return resource
+	}
 
-  import(context: RepositoryContext, draft: OrderImportDraft): Order {
-    // TODO: Check if order with given orderNumber already exists
-    assert(this, 'OrderRepository not valid')
-    const resource: Order = {
-      ...getBaseResourceProperties(),
+	import(context: RepositoryContext, draft: OrderImportDraft): Order {
+		// TODO: Check if order with given orderNumber already exists
+		assert(this, 'OrderRepository not valid')
+		const resource: Order = {
+			...getBaseResourceProperties(),
 
-      billingAddress: createAddress(
-        draft.billingAddress,
-        context.projectKey,
-        this._storage
-      ),
-      shippingAddress: createAddress(
-        draft.shippingAddress,
-        context.projectKey,
-        this._storage
-      ),
+			billingAddress: createAddress(
+				draft.billingAddress,
+				context.projectKey,
+				this._storage
+			),
+			shippingAddress: createAddress(
+				draft.shippingAddress,
+				context.projectKey,
+				this._storage
+			),
 
-      custom: createCustomFields(
-        draft.custom,
-        context.projectKey,
-        this._storage
-      ),
-      customerEmail: draft.customerEmail,
-      lastMessageSequenceNumber: 0,
-      orderNumber: draft.orderNumber,
-      orderState: draft.orderState || 'Open',
-      origin: draft.origin || 'Customer',
-      paymentState: draft.paymentState,
-      refusedGifts: [],
-      shippingMode: 'Single',
-      shipping: [],
+			custom: createCustomFields(
+				draft.custom,
+				context.projectKey,
+				this._storage
+			),
+			customerEmail: draft.customerEmail,
+			lastMessageSequenceNumber: 0,
+			orderNumber: draft.orderNumber,
+			orderState: draft.orderState || 'Open',
+			origin: draft.origin || 'Customer',
+			paymentState: draft.paymentState,
+			refusedGifts: [],
+			shippingMode: 'Single',
+			shipping: [],
 
-      store: resolveStoreReference(
-        draft.store,
-        context.projectKey,
-        this._storage
-      ),
-      syncInfo: [],
+			store: resolveStoreReference(
+				draft.store,
+				context.projectKey,
+				this._storage
+			),
+			syncInfo: [],
 
-      lineItems:
-        draft.lineItems?.map((item) =>
-          this.lineItemFromImportDraft.bind(this)(context, item)
-        ) || [],
-      customLineItems:
-        draft.customLineItems?.map((item) =>
-          this.customLineItemFromImportDraft.bind(this)(context, item)
-        ) || [],
+			lineItems:
+				draft.lineItems?.map((item) =>
+					this.lineItemFromImportDraft.bind(this)(context, item)
+				) || [],
+			customLineItems:
+				draft.customLineItems?.map((item) =>
+					this.customLineItemFromImportDraft.bind(this)(context, item)
+				) || [],
 
-      totalPrice: createCentPrecisionMoney(draft.totalPrice),
-    }
-    this.saveNew(context, resource)
-    return resource
-  }
+			totalPrice: createCentPrecisionMoney(draft.totalPrice),
+		}
+		this.saveNew(context, resource)
+		return resource
+	}
 
-  private lineItemFromImportDraft(
-    context: RepositoryContext,
-    draft: LineItemImportDraft
-  ): LineItem {
-    let product: Product
-    let variant: ProductVariant | undefined
+	private lineItemFromImportDraft(
+		context: RepositoryContext,
+		draft: LineItemImportDraft
+	): LineItem {
+		let product: Product
+		let variant: ProductVariant | undefined
 
-    if (draft.variant.sku) {
-      variant = {
-        id: 0,
-        sku: draft.variant.sku,
-      }
+		if (draft.variant.sku) {
+			variant = {
+				id: 0,
+				sku: draft.variant.sku,
+			}
 
-      const items = this._storage.query(context.projectKey, 'product', {
-        where: [
-          `masterData(current(masterVariant(sku="${draft.variant.sku}"))) or masterData(current(variants(sku="${draft.variant.sku}")))`,
-        ],
-      }) as ProductPagedQueryResponse
+			const items = this._storage.query(context.projectKey, 'product', {
+				where: [
+					`masterData(current(masterVariant(sku="${draft.variant.sku}"))) or masterData(current(variants(sku="${draft.variant.sku}")))`,
+				],
+			}) as ProductPagedQueryResponse
 
-      if (items.count !== 1) {
-        throw new CommercetoolsError<GeneralError>({
-          code: 'General',
-          message: `A product containing a variant with SKU '${draft.variant.sku}' not found.`,
-        })
-      }
+			if (items.count !== 1) {
+				throw new CommercetoolsError<GeneralError>({
+					code: 'General',
+					message: `A product containing a variant with SKU '${draft.variant.sku}' not found.`,
+				})
+			}
 
-      product = items.results[0]
-      if (product.masterData.current.masterVariant.sku === draft.variant.sku) {
-        variant = product.masterData.current.masterVariant
-      } else {
-        variant = product.masterData.current.variants.find(
-          (v) => v.sku === draft.variant.sku
-        )
-      }
-      if (!variant) {
-        throw new Error('Internal state error')
-      }
-    } else {
-      throw new Error('No product found')
-    }
+			product = items.results[0]
+			if (product.masterData.current.masterVariant.sku === draft.variant.sku) {
+				variant = product.masterData.current.masterVariant
+			} else {
+				variant = product.masterData.current.variants.find(
+					(v) => v.sku === draft.variant.sku
+				)
+			}
+			if (!variant) {
+				throw new Error('Internal state error')
+			}
+		} else {
+			throw new Error('No product found')
+		}
 
-    const lineItem: LineItem = {
-      ...getBaseResourceProperties(),
-      custom: createCustomFields(
-        draft.custom,
-        context.projectKey,
-        this._storage
-      ),
-      discountedPricePerQuantity: [],
-      lineItemMode: 'Standard',
-      name: draft.name,
-      price: createPrice(draft.price),
-      priceMode: 'Platform',
-      productId: product.id,
-      productType: product.productType,
-      quantity: draft.quantity,
-      state: draft.state || [],
-      taxRate: draft.taxRate,
-      taxedPricePortions: [],
-      perMethodTaxRate: [],
-      totalPrice: createCentPrecisionMoney(draft.price.value),
-      variant: {
-        id: variant.id,
-        sku: variant.sku,
-        price: createPrice(draft.price),
-      },
-    }
+		const lineItem: LineItem = {
+			...getBaseResourceProperties(),
+			custom: createCustomFields(
+				draft.custom,
+				context.projectKey,
+				this._storage
+			),
+			discountedPricePerQuantity: [],
+			lineItemMode: 'Standard',
+			name: draft.name,
+			price: createPrice(draft.price),
+			priceMode: 'Platform',
+			productId: product.id,
+			productType: product.productType,
+			quantity: draft.quantity,
+			state: draft.state || [],
+			taxRate: draft.taxRate,
+			taxedPricePortions: [],
+			perMethodTaxRate: [],
+			totalPrice: createCentPrecisionMoney(draft.price.value),
+			variant: {
+				id: variant.id,
+				sku: variant.sku,
+				price: createPrice(draft.price),
+			},
+		}
 
-    return lineItem
-  }
+		return lineItem
+	}
 
-  private customLineItemFromImportDraft(
-    context: RepositoryContext,
-    draft: CustomLineItemDraft
-  ): CustomLineItem {
-    const lineItem: CustomLineItem = {
-      ...getBaseResourceProperties(),
-      custom: createCustomFields(
-        draft.custom,
-        context.projectKey,
-        this._storage
-      ),
-      discountedPricePerQuantity: [],
-      money: createTypedMoney(draft.money),
-      name: draft.name,
-      quantity: draft.quantity ?? 0,
-      perMethodTaxRate: [],
-      priceMode: draft.priceMode,
-      slug: draft.slug,
-      state: [],
-      totalPrice: createCentPrecisionMoney(draft.money),
-    }
+	private customLineItemFromImportDraft(
+		context: RepositoryContext,
+		draft: CustomLineItemDraft
+	): CustomLineItem {
+		const lineItem: CustomLineItem = {
+			...getBaseResourceProperties(),
+			custom: createCustomFields(
+				draft.custom,
+				context.projectKey,
+				this._storage
+			),
+			discountedPricePerQuantity: [],
+			money: createTypedMoney(draft.money),
+			name: draft.name,
+			quantity: draft.quantity ?? 0,
+			perMethodTaxRate: [],
+			priceMode: draft.priceMode,
+			slug: draft.slug,
+			state: [],
+			totalPrice: createCentPrecisionMoney(draft.money),
+		}
 
-    return lineItem
-  }
+		return lineItem
+	}
 
-  getWithOrderNumber(
-    context: RepositoryContext,
-    orderNumber: string,
-    params: QueryParams = {}
-  ): Order | undefined {
-    const result = this._storage.query(context.projectKey, this.getTypeId(), {
-      ...params,
-      where: [`orderNumber="${orderNumber}"`],
-    })
-    if (result.count === 1) {
-      return result.results[0] as Order
-    }
+	getWithOrderNumber(
+		context: RepositoryContext,
+		orderNumber: string,
+		params: QueryParams = {}
+	): Order | undefined {
+		const result = this._storage.query(context.projectKey, this.getTypeId(), {
+			...params,
+			where: [`orderNumber="${orderNumber}"`],
+		})
+		if (result.count === 1) {
+			return result.results[0] as Order
+		}
 
-    // Catch this for now, should be checked when creating/updating
-    if (result.count > 1) {
-      throw new Error('Duplicate order numbers')
-    }
+		// Catch this for now, should be checked when creating/updating
+		if (result.count > 1) {
+			throw new Error('Duplicate order numbers')
+		}
 
-    return
-  }
+		return
+	}
 
-  actions = {
-    addPayment: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { payment }: OrderAddPaymentAction
-    ) => {
-      const resolvedPayment = this._storage.getByResourceIdentifier(
-        context.projectKey,
-        payment
-      )
-      if (!resolvedPayment) {
-        throw new Error(`Payment ${payment.id} not found`)
-      }
+	actions = {
+		addPayment: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ payment }: OrderAddPaymentAction
+		) => {
+			const resolvedPayment = this._storage.getByResourceIdentifier(
+				context.projectKey,
+				payment
+			)
+			if (!resolvedPayment) {
+				throw new Error(`Payment ${payment.id} not found`)
+			}
 
-      if (!resource.paymentInfo) {
-        resource.paymentInfo = {
-          payments: [],
-        }
-      }
+			if (!resource.paymentInfo) {
+				resource.paymentInfo = {
+					payments: [],
+				}
+			}
 
-      resource.paymentInfo.payments.push({
-        typeId: 'payment',
-        id: payment.id!,
-      })
-    },
-    changeOrderState: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { orderState }: OrderChangeOrderStateAction
-    ) => {
-      resource.orderState = orderState
-    },
-    changePaymentState: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { paymentState }: OrderChangePaymentStateAction
-    ) => {
-      resource.paymentState = paymentState
-    },
-    transitionState: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { state }: OrderTransitionStateAction
-    ) => {
-      const resolvedType = this._storage.getByResourceIdentifier(
-        context.projectKey,
-        state
-      ) as State | null
+			resource.paymentInfo.payments.push({
+				typeId: 'payment',
+				id: payment.id!,
+			})
+		},
+		changeOrderState: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ orderState }: OrderChangeOrderStateAction
+		) => {
+			resource.orderState = orderState
+		},
+		changePaymentState: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ paymentState }: OrderChangePaymentStateAction
+		) => {
+			resource.paymentState = paymentState
+		},
+		transitionState: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ state }: OrderTransitionStateAction
+		) => {
+			const resolvedType = this._storage.getByResourceIdentifier(
+				context.projectKey,
+				state
+			) as State | null
 
-      if (!resolvedType) {
-        throw new Error(
-          `No state found with key=${state.key} or id=${state.key}`
-        )
-      }
+			if (!resolvedType) {
+				throw new Error(
+					`No state found with key=${state.key} or id=${state.key}`
+				)
+			}
 
-      resource.state = {
-        typeId: 'state',
-        id: resolvedType.id,
-        obj: { ...resolvedType, key: state.key ?? '' },
-      }
-    },
-    setBillingAddress: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { address }: OrderSetBillingAddressAction
-    ) => {
-      resource.billingAddress = createAddress(
-        address,
-        context.projectKey,
-        this._storage
-      )
-    },
-    setCustomerEmail: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { email }: OrderSetCustomerEmailAction
-    ) => {
-      resource.customerEmail = email
-    },
-    setCustomField: (
-      context: RepositoryContext,
-      resource: Order,
-      { name, value }: OrderSetCustomFieldAction
-    ) => {
-      if (!resource.custom) {
-        throw new Error('Resource has no custom field')
-      }
-      resource.custom.fields[name] = value
-    },
-    setCustomType: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { type, fields }: OrderSetCustomTypeAction
-    ) => {
-      if (!type) {
-        resource.custom = undefined
-      } else {
-        const resolvedType = this._storage.getByResourceIdentifier(
-          context.projectKey,
-          type
-        )
-        if (!resolvedType) {
-          throw new Error(`Type ${type} not found`)
-        }
+			resource.state = {
+				typeId: 'state',
+				id: resolvedType.id,
+				obj: { ...resolvedType, key: state.key ?? '' },
+			}
+		},
+		setBillingAddress: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ address }: OrderSetBillingAddressAction
+		) => {
+			resource.billingAddress = createAddress(
+				address,
+				context.projectKey,
+				this._storage
+			)
+		},
+		setCustomerEmail: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ email }: OrderSetCustomerEmailAction
+		) => {
+			resource.customerEmail = email
+		},
+		setCustomField: (
+			context: RepositoryContext,
+			resource: Order,
+			{ name, value }: OrderSetCustomFieldAction
+		) => {
+			if (!resource.custom) {
+				throw new Error('Resource has no custom field')
+			}
+			resource.custom.fields[name] = value
+		},
+		setCustomType: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ type, fields }: OrderSetCustomTypeAction
+		) => {
+			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 || {},
-        }
-      }
-    },
-    setLocale: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { locale }: OrderSetLocaleAction
-    ) => {
-      resource.locale = locale
-    },
-    setOrderNumber: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { orderNumber }: OrderSetOrderNumberAction
-    ) => {
-      resource.orderNumber = orderNumber
-    },
-    setShippingAddress: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { address }: OrderSetShippingAddressAction
-    ) => {
-      resource.shippingAddress = createAddress(
-        address,
-        context.projectKey,
-        this._storage
-      )
-    },
-    setStore: (
-      context: RepositoryContext,
-      resource: Writable<Order>,
-      { store }: OrderSetStoreAction
-    ) => {
-      if (!store) return
-      const resolvedType = this._storage.getByResourceIdentifier(
-        context.projectKey,
-        store
-      )
-      if (!resolvedType) {
-        throw new Error(`No store found with key=${store.key}`)
-      }
+				resource.custom = {
+					type: {
+						typeId: 'type',
+						id: resolvedType.id,
+					},
+					fields: fields || {},
+				}
+			}
+		},
+		setLocale: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ locale }: OrderSetLocaleAction
+		) => {
+			resource.locale = locale
+		},
+		setOrderNumber: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ orderNumber }: OrderSetOrderNumberAction
+		) => {
+			resource.orderNumber = orderNumber
+		},
+		setShippingAddress: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ address }: OrderSetShippingAddressAction
+		) => {
+			resource.shippingAddress = createAddress(
+				address,
+				context.projectKey,
+				this._storage
+			)
+		},
+		setStore: (
+			context: RepositoryContext,
+			resource: Writable<Order>,
+			{ store }: OrderSetStoreAction
+		) => {
+			if (!store) return
+			const resolvedType = this._storage.getByResourceIdentifier(
+				context.projectKey,
+				store
+			)
+			if (!resolvedType) {
+				throw new Error(`No store found with key=${store.key}`)
+			}
 
-      const storeReference = resolvedType as Store
-      resource.store = {
-        typeId: 'store',
-        key: storeReference.key,
-      }
-    },
-  }
+			const storeReference = resolvedType as Store
+			resource.store = {
+				typeId: 'store',
+				key: storeReference.key,
+			}
+		},
+	}
 }