@labdigital/commercetools-mock 0.5.13 → 0.5.16

package/src/repositories/cart.ts

@@ -1,9 +1,30 @@
-import { Cart, CartDraft, ReferenceTypeId } from '@commercetools/platform-sdk'
+import {
+  Cart,
+  CartAddLineItemAction,
+  CartDraft,
+  CartRemoveLineItemAction,
+  CartSetBillingAddressAction,
+  CartSetCountryAction,
+  CartSetCustomerEmailAction,
+  CartSetCustomFieldAction,
+  CartSetCustomTypeAction,
+  CartSetLocaleAction,
+  CartSetShippingAddressAction,
+  GeneralError,
+  LineItem,
+  Product,
+  ProductPagedQueryResponse,
+  ProductVariant,
+  ReferenceTypeId,
+} from '@commercetools/platform-sdk'
+import { v4 as uuidv4 } from 'uuid'
 import { getBaseResourceProperties } from '../helpers'
-import AbstractRepository from './abstract'
+import { AbstractResourceRepository } from './abstract'
 import { createCustomFields } from './helpers'
+import { Writable } from '../types'
+import { CommercetoolsError } from '../exceptions'
 
-export class CartRepository extends AbstractRepository {
+export class CartRepository extends AbstractResourceRepository {
   getTypeId(): ReferenceTypeId {
     return 'cart'
   }
@@ -30,4 +51,227 @@ export class CartRepository extends AbstractRepository {
     this.save(projectKey, resource)
     return resource
   }
+
+  getActiveCart(projectKey: string): Cart | undefined {
+    // Get first active cart
+    const results = this._storage.query(projectKey, this.getTypeId(), {
+      where: [`cartState="Active"`],
+    })
+    if (results.count > 0) {
+      return results.results[0] as Cart
+    }
+
+    return
+  }
+
+  actions = {
+    addLineItem: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { productId, variantId, sku, quantity = 1 }: CartAddLineItemAction
+    ) => {
+      let product: Product | null = null
+      let variant: ProductVariant | undefined
+
+      if (productId && variantId) {
+        // Fetch product and variant by ID
+        product = this._storage.get(projectKey, 'product', productId, {})
+      } else if (sku) {
+        // Fetch product and variant by SKU
+        const items = this._storage.query(projectKey, 'product', {
+          where: [
+            `masterData(current(masterVariant(sku="${sku}"))) or masterData(current(variants(sku="${sku}")))`,
+          ],
+        }) as ProductPagedQueryResponse
+
+        if (items.count === 1) {
+          product = items.results[0]
+        }
+      }
+
+      if (!product) {
+        // Check if product is found
+        throw new CommercetoolsError<GeneralError>({
+          code: 'General',
+          message: sku
+            ? `A product containing a variant with SKU '${sku}' not found.`
+            : `A product with ID '${productId}' not found.`,
+        })
+      }
+
+      // Find matching variant
+      variant = [
+        product.masterData.current.masterVariant,
+        ...product.masterData.current.variants,
+      ].find(x => {
+        if (sku) return x.sku === sku
+        if (variantId) return x.id === variantId
+        return false
+      })
+
+      if (!variant) {
+        // Check if variant is found
+        throw new CommercetoolsError<GeneralError>({
+          code: 'General',
+          message: sku
+            ? `A variant with SKU '${sku}' for product '${product.id}' not found.`
+            : `A variant with ID '${variantId}' for product '${product.id}' not found.`,
+        })
+      }
+
+      const alreadyAdded = resource.lineItems.some(
+        x => x.productId === product?.id && x.variant.id === variant?.id
+      )
+      if (alreadyAdded) {
+        // increase quantity and update total price
+        resource.lineItems.map(x => {
+          if (x.productId === product?.id && x.variant.id === variant?.id) {
+            x.quantity += quantity
+            x.totalPrice.centAmount = calculateLineItemTotalPrice(x)
+          }
+          return x
+        })
+      } else {
+        // add line item
+        if (!variant.prices?.length) {
+          throw new CommercetoolsError<GeneralError>({
+            code: 'General',
+            message: `A product with ID '${productId}' doesn't have any prices.`,
+          })
+        }
+
+        const price = variant.prices[0]
+        resource.lineItems.push({
+          id: uuidv4(),
+          productId: product.id,
+          productKey: product.key,
+          name: product.masterData.current.name,
+          productSlug: product.masterData.current.slug,
+          productType: product.productType,
+          variant,
+          price: price,
+          totalPrice: {
+            ...price.value,
+            centAmount: price.value.centAmount * quantity,
+          },
+          quantity,
+          discountedPricePerQuantity: [],
+          lineItemMode: 'Standard',
+          priceMode: 'Platform',
+          state: [],
+        })
+      }
+
+      // Update cart total price
+      resource.totalPrice.centAmount = calculateCartTotalPrice(resource)
+    },
+    removeLineItem: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { lineItemId, quantity }: CartRemoveLineItemAction
+    ) => {
+      const lineItem = resource.lineItems.find(x => x.id === lineItemId)
+      if (!lineItem) {
+        // Check if product is found
+        throw new CommercetoolsError<GeneralError>({
+          code: 'General',
+          message: `A line item with ID '${lineItemId}' not found.`,
+        })
+      }
+
+      const shouldDelete = !quantity || quantity >= lineItem.quantity
+      if (shouldDelete) {
+        // delete line item
+        resource.lineItems = resource.lineItems.filter(x => x.id !== lineItemId)
+      } else {
+        // decrease quantity and update total price
+        resource.lineItems.map(x => {
+          if (x.id === lineItemId && quantity) {
+            x.quantity -= quantity
+            x.totalPrice.centAmount = calculateLineItemTotalPrice(x)
+          }
+          return x
+        })
+      }
+
+      // Update cart total price
+      resource.totalPrice.centAmount = calculateCartTotalPrice(resource)
+    },
+    setBillingAddress: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { address }: CartSetBillingAddressAction
+    ) => {
+      resource.billingAddress = address
+    },
+    setCountry: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { country }: CartSetCountryAction
+    ) => {
+      resource.country = country
+    },
+    setCustomerEmail: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { email }: CartSetCustomerEmailAction
+    ) => {
+      resource.customerEmail = email
+    },
+    setCustomField: (
+      projectKey: string,
+      resource: Cart,
+      { name, value }: CartSetCustomFieldAction
+    ) => {
+      if (!resource.custom) {
+        throw new Error('Resource has no custom field')
+      }
+      resource.custom.fields[name] = value
+    },
+    setCustomType: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { type, fields }: CartSetCustomTypeAction
+    ) => {
+      if (!type) {
+        resource.custom = undefined
+      } else {
+        const resolvedType = this._storage.getByResourceIdentifier(
+          projectKey,
+          type
+        )
+        if (!resolvedType) {
+          throw new Error(`Type ${type} not found`)
+        }
+
+        resource.custom = {
+          type: {
+            typeId: 'type',
+            id: resolvedType.id,
+          },
+          fields: fields || [],
+        }
+      }
+    },
+    setLocale: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { locale }: CartSetLocaleAction
+    ) => {
+      resource.locale = locale
+    },
+    setShippingAddress: (
+      projectKey: string,
+      resource: Writable<Cart>,
+      { address }: CartSetShippingAddressAction
+    ) => {
+      resource.shippingAddress = address
+    },
+  }
 }
+
+const calculateLineItemTotalPrice = (lineItem: LineItem): number =>
+  lineItem.price!.value.centAmount * lineItem.quantity
+
+const calculateCartTotalPrice = (cart: Cart): number =>
+  cart.lineItems.reduce((cur, item) => cur + item.totalPrice.centAmount, 0)

package/src/repositories/category.ts

@@ -0,0 +1,140 @@
+import { v4 as uuidv4 } from 'uuid'
+import {
+  Category,
+  CategoryChangeAssetNameAction,
+  CategoryChangeSlugAction,
+  CategoryDraft,
+  CategorySetAssetDescriptionAction,
+  CategorySetAssetSourcesAction,
+  CategorySetDescriptionAction,
+  CategorySetKeyAction,
+  CategorySetMetaDescriptionAction,
+  CategorySetMetaKeywordsAction,
+  CategorySetMetaTitleAction,
+  ReferenceTypeId,
+} from '@commercetools/platform-sdk'
+import { Writable } from 'types'
+import { getBaseResourceProperties } from '../helpers'
+import { createCustomFields } from './helpers'
+import { AbstractResourceRepository } from './abstract'
+
+export class CategoryRepository extends AbstractResourceRepository {
+  getTypeId(): ReferenceTypeId {
+    return 'category'
+  }
+
+  create(projectKey: string, draft: CategoryDraft): Category {
+    const resource: Category = {
+      ...getBaseResourceProperties(),
+      key: draft.key,
+      name: draft.name,
+      slug: draft.slug,
+      orderHint: draft.orderHint || '',
+      parent: draft.parent
+        ? { typeId: 'category', id: draft.parent.id! }
+        : undefined,
+      ancestors: [], // TODO
+      assets:
+        draft.assets?.map(d => {
+          return {
+            id: uuidv4(),
+            name: d.name,
+            description: d.description,
+            sources: d.sources,
+            tags: d.tags,
+            key: d.key,
+            custom: createCustomFields(draft.custom, projectKey, this._storage),
+          }
+        }) || [],
+    }
+    this.save(projectKey, resource)
+    return resource
+  }
+
+  actions = {
+    changeAssetName: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { assetId, assetKey, name }: CategoryChangeAssetNameAction
+    ) => {
+      resource.assets?.forEach(asset => {
+        if (assetId && assetId == asset.id) {
+          asset.name = name
+        }
+        if (assetKey && assetKey == asset.key) {
+          asset.name = name
+        }
+      })
+    },
+    changeSlug: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { slug }: CategoryChangeSlugAction
+    ) => {
+      resource.slug = slug
+    },
+    setKey: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { key }: CategorySetKeyAction
+    ) => {
+      resource.key = key
+    },
+    setAssetDescription: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { assetId, assetKey, description }: CategorySetAssetDescriptionAction
+    ) => {
+      resource.assets?.forEach(asset => {
+        if (assetId && assetId == asset.id) {
+          asset.description = description
+        }
+        if (assetKey && assetKey == asset.key) {
+          asset.description = description
+        }
+      })
+    },
+    setAssetSources: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { assetId, assetKey, sources }: CategorySetAssetSourcesAction
+    ) => {
+      resource.assets?.forEach(asset => {
+        if (assetId && assetId == asset.id) {
+          asset.sources = sources
+        }
+        if (assetKey && assetKey == asset.key) {
+          asset.sources = sources
+        }
+      })
+    },
+    setDescription: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { description }: CategorySetDescriptionAction
+    ) => {
+      resource.description = description
+    },
+    setMetaDescription: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { metaDescription }: CategorySetMetaDescriptionAction
+    ) => {
+      resource.metaDescription = metaDescription
+    },
+    setMetaKeywords: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { metaKeywords }: CategorySetMetaKeywordsAction
+    ) => {
+      resource.metaKeywords = metaKeywords
+    },
+    setMetaTitle: (
+      projectKey: string,
+      resource: Writable<Category>,
+      { metaTitle }: CategorySetMetaTitleAction
+    ) => {
+      resource.metaTitle = metaTitle
+    },
+  }
+}

package/src/repositories/channel.ts

@@ -0,0 +1,23 @@
+import {
+  Channel,
+  ChannelDraft,
+  ReferenceTypeId,
+} from '@commercetools/platform-sdk'
+import { getBaseResourceProperties } from '../helpers'
+import { AbstractResourceRepository } from './abstract'
+
+export class ChannelRepository extends AbstractResourceRepository {
+  getTypeId(): ReferenceTypeId {
+    return 'channel'
+  }
+
+  create(projectKey: string, draft: ChannelDraft): Channel {
+    const resource: Channel = {
+      ...getBaseResourceProperties(),
+      key: draft.key,
+      roles: draft.roles || [],
+    }
+    this.save(projectKey, resource)
+    return resource
+  }
+}

package/src/repositories/custom-object.ts

@@ -4,11 +4,11 @@ import {
   ReferenceTypeId,
 } from '@commercetools/platform-sdk'
 import { checkConcurrentModification } from './errors'
-import AbstractRepository from './abstract'
+import { AbstractResourceRepository } from './abstract'
 import { Writable } from '../types'
 import { getBaseResourceProperties } from '../helpers'
 
-export class CustomObjectRepository extends AbstractRepository {
+export class CustomObjectRepository extends AbstractResourceRepository {
   getTypeId(): ReferenceTypeId {
     return 'key-value-document'
   }

package/src/repositories/customer-group.ts

@@ -0,0 +1,42 @@
+import {
+  CustomerGroup,
+  CustomerGroupChangeNameAction,
+  CustomerGroupDraft,
+  CustomerGroupSetKeyAction,
+  ReferenceTypeId,
+} from '@commercetools/platform-sdk'
+import { Writable } from 'types'
+import { getBaseResourceProperties } from '../helpers'
+import { AbstractResourceRepository } from './abstract'
+
+export class CustomerGroupRepository extends AbstractResourceRepository {
+  getTypeId(): ReferenceTypeId {
+    return 'customer'
+  }
+  create(projectKey: string, draft: CustomerGroupDraft): CustomerGroup {
+    const resource: CustomerGroup = {
+      ...getBaseResourceProperties(),
+      key: draft.key,
+      name: draft.groupName,
+    }
+    this.save(projectKey, resource)
+    return resource
+  }
+
+  actions = {
+    setKey: (
+      projectKey: string,
+      resource: Writable<CustomerGroup>,
+      { key }: CustomerGroupSetKeyAction
+    ) => {
+      resource.key = key
+    },
+    changeName: (
+      projectKey: string,
+      resource: Writable<CustomerGroup>,
+      { name }: CustomerGroupChangeNameAction
+    ) => {
+      resource.name = name
+    },
+  }
+}

package/src/repositories/customer.ts

@@ -4,9 +4,9 @@ import {
   ReferenceTypeId,
 } from '@commercetools/platform-sdk'
 import { getBaseResourceProperties } from '../helpers'
-import AbstractRepository from './abstract'
+import { AbstractResourceRepository } from './abstract'
 
-export class CustomerRepository extends AbstractRepository {
+export class CustomerRepository extends AbstractResourceRepository {
   getTypeId(): ReferenceTypeId {
     return 'customer'
   }

package/src/repositories/discount-code.ts

@@ -0,0 +1,143 @@
+import {
+  CartDiscountReference,
+  DiscountCode,
+  DiscountCodeChangeCartDiscountsAction,
+  DiscountCodeChangeIsActiveAction,
+  DiscountCodeDraft,
+  DiscountCodeSetCartPredicateAction,
+  DiscountCodeSetDescriptionAction,
+  DiscountCodeSetMaxApplicationsAction,
+  DiscountCodeSetMaxApplicationsPerCustomerAction,
+  DiscountCodeSetNameAction,
+  DiscountCodeSetValidFromAction,
+  DiscountCodeSetValidFromAndUntilAction,
+  DiscountCodeSetValidUntilAction,
+  DiscountCodeUpdateAction,
+  ReferenceTypeId,
+} from '@commercetools/platform-sdk'
+import { Writable } from 'types'
+import { getBaseResourceProperties } from '../helpers'
+import { AbstractResourceRepository } from './abstract'
+
+export class DiscountCodeRepository extends AbstractResourceRepository {
+  getTypeId(): ReferenceTypeId {
+    return 'cart-discount'
+  }
+
+  create(projectKey: string, draft: DiscountCodeDraft): DiscountCode {
+    console.log(draft)
+    const resource: DiscountCode = {
+      ...getBaseResourceProperties(),
+      applicationVersion: 1,
+      cartDiscounts: draft.cartDiscounts.map(
+        (obj): CartDiscountReference => ({
+          typeId: 'cart-discount',
+          id: obj.id!,
+        })
+      ),
+      cartPredicate: draft.cartPredicate,
+      code: draft.code,
+      description: draft.description,
+      groups: draft.groups || [],
+      isActive: draft.isActive || true,
+      name: draft.name,
+      references: [],
+      validFrom: draft.validFrom,
+      validUntil: draft.validUntil,
+      maxApplications: draft.maxApplications,
+      maxApplicationsPerCustomer: draft.maxApplicationsPerCustomer,
+    }
+    this.save(projectKey, resource)
+    return resource
+  }
+
+  actions: Partial<
+    Record<
+      DiscountCodeUpdateAction['action'],
+      (
+        projectKey: string,
+        resource: Writable<DiscountCode>,
+        action: any
+      ) => void
+    >
+  > = {
+    changeIsActive: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { isActive }: DiscountCodeChangeIsActiveAction
+    ) => {
+      resource.isActive = isActive
+    },
+    changeCartDiscounts: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { cartDiscounts }: DiscountCodeChangeCartDiscountsAction
+    ) => {
+      resource.cartDiscounts = cartDiscounts.map(
+        (obj): CartDiscountReference => ({
+          typeId: 'cart-discount',
+          id: obj.id!,
+        })
+      )
+    },
+    setDescription: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { description }: DiscountCodeSetDescriptionAction
+    ) => {
+      resource.description = description
+    },
+    setCartPredicate: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { cartPredicate }: DiscountCodeSetCartPredicateAction
+    ) => {
+      resource.cartPredicate = cartPredicate
+    },
+    setName: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { name }: DiscountCodeSetNameAction
+    ) => {
+      resource.name = name
+    },
+    setMaxApplications: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { maxApplications }: DiscountCodeSetMaxApplicationsAction
+    ) => {
+      resource.maxApplications = maxApplications
+    },
+    setMaxApplicationsPerCustomer: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      {
+        maxApplicationsPerCustomer,
+      }: DiscountCodeSetMaxApplicationsPerCustomerAction
+    ) => {
+      resource.maxApplicationsPerCustomer = maxApplicationsPerCustomer
+    },
+    setValidFrom: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { validFrom }: DiscountCodeSetValidFromAction
+    ) => {
+      resource.validFrom = validFrom
+    },
+    setValidUntil: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { validUntil }: DiscountCodeSetValidUntilAction
+    ) => {
+      resource.validUntil = validUntil
+    },
+    setValidFromAndUntil: (
+      projectKey: string,
+      resource: Writable<DiscountCode>,
+      { validFrom, validUntil }: DiscountCodeSetValidFromAndUntilAction
+    ) => {
+      resource.validFrom = validFrom
+      resource.validUntil = validUntil
+    },
+  }
+}

package/src/repositories/errors.ts

@@ -1,18 +1,23 @@
 import {
   BaseResource,
   ConcurrentModificationError,
+  Project,
 } from '@commercetools/platform-sdk'
 import { CommercetoolsError } from '../exceptions'
 
 export const checkConcurrentModification = (
-  resource: BaseResource,
+  resource: BaseResource | Project,
   expectedVersion: number
 ) => {
   if (resource.version === expectedVersion) return
 
+  const identifier = (<BaseResource>resource).id
+    ? (<BaseResource>resource).id
+    : (<Project>resource).key
+
   throw new CommercetoolsError<ConcurrentModificationError>(
     {
-      message: `Object ${resource.id} has a different version than expected. Expected: ${expectedVersion} - Actual: ${resource.version}.`,
+      message: `Object ${identifier} has a different version than expected. Expected: ${expectedVersion} - Actual: ${resource.version}.`,
       currentVersion: resource.version,
       code: 'ConcurrentModification',
     },