chore: Move factories and helpers to a better place (#4551)

* chore: Move factories and helpers to a better place * align factory product variant * fix factory cart * add simple store fac * fix tests * fix tests * fix * fix cart seeder
2023-07-20 13:16:04 +02:00
parent 26e2f81c24
commit 4d326fbbdf
197 changed files with 1073 additions and 4332 deletions
@@ -0,0 +1,81 @@
+import {
+  CustomerGroup,
+  MoneyAmount,
+  PriceList,
+  PriceListStatus,
+  PriceListType,
+} from "@medusajs/medusa"
+import faker from "faker"
+import { DataSource } from "typeorm"
+import { simpleCustomerGroupFactory } from "./simple-customer-group-factory"
+
+type ProductListPrice = {
+  variant_id: string
+  currency_code: string
+  region_id: string
+  amount: number
+}
+
+export type PriceListFactoryData = {
+  id?: string
+  name?: string
+  description?: string
+  type?: PriceListType
+  status?: PriceListStatus
+  starts_at?: Date
+  ends_at?: Date
+  customer_groups?: string[]
+  prices?: ProductListPrice[]
+  includes_tax?: boolean
+}
+
+export const simplePriceListFactory = async (
+  dataSource: DataSource,
+  data: PriceListFactoryData = {},
+  seed?: number
+): Promise<PriceList> => {
+  if (typeof seed !== "undefined") {
+    faker.seed(seed)
+  }
+
+  const manager = dataSource.manager
+
+  const listId = data.id || `simple-price-list-${Math.random() * 1000}`
+
+  let customerGroups: CustomerGroup[] = []
+  if (typeof data.customer_groups !== "undefined") {
+    customerGroups = await Promise.all(
+      data.customer_groups.map((group) =>
+        simpleCustomerGroupFactory(dataSource, { id: group })
+      )
+    )
+  }
+
+  const toCreate = {
+    id: listId,
+    name: data.name || faker.commerce.productName(),
+    description: data.description || "Some text",
+    status: data.status || PriceListStatus.ACTIVE,
+    type: data.type || PriceListType.OVERRIDE,
+    starts_at: data.starts_at || null,
+    ends_at: data.ends_at || null,
+    customer_groups: customerGroups,
+    includes_tax: data.includes_tax,
+  }
+
+  const toSave = manager.create(PriceList, toCreate)
+  const toReturn = await manager.save(toSave)
+
+  if (typeof data.prices !== "undefined") {
+    for (const ma of data.prices) {
+      const factoryData = {
+        ...ma,
+        price_list_id: listId,
+      }
+      const toSave = manager.create(MoneyAmount, factoryData)
+      await manager.save(toSave)
+    }
+  }
+
+  return toReturn
+}