working on multi attach

server/tests/core/multiAttach/multiAttach1.test.ts

@@ -0,0 +1,155 @@
+import chalk from "chalk";
+import { setupBefore } from "tests/before.js";
+import { Stripe } from "stripe";
+import { createProducts } from "tests/utils/productUtils.js";
+import { constructProduct } from "@/utils/scriptUtils/createTestProducts.js";
+import { TestFeature } from "tests/setup/v2Features.js";
+import { AutumnInt } from "@/external/autumn/autumnCli.js";
+import { initCustomer } from "@/utils/scriptUtils/initCustomer.js";
+import {
+  APIVersion,
+  AppEnv,
+  CusProductStatus,
+  Organization,
+} from "@autumn/shared";
+import {
+  constructArrearItem,
+  constructFeatureItem,
+} from "@/utils/scriptUtils/constructItem.js";
+import { DrizzleCli } from "@/db/initDrizzle.js";
+import {
+  addPrefixToProducts,
+  getBasePrice,
+} from "tests/utils/testProductUtils/testProductUtils.js";
+import { expect } from "chai";
+import { advanceToNextInvoice } from "tests/utils/testAttachUtils/testAttachUtils.js";
+import { attachAndExpectCorrect } from "tests/utils/expectUtils/expectAttach.js";
+import {
+  advanceTestClock,
+  completeCheckoutForm,
+} from "tests/utils/stripeUtils.js";
+import { addWeeks } from "date-fns";
+import { getExpectedInvoiceTotal } from "tests/utils/expectUtils/expectInvoiceUtils.js";
+
+let growth = constructProduct({
+  id: "growth",
+  items: [
+    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 100 }),
+  ],
+  type: "growth",
+});
+
+let premium = constructProduct({
+  id: "premium",
+  items: [
+    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 200 }),
+  ],
+  type: "premium",
+});
+
+let pro = constructProduct({
+  id: "pro",
+  items: [
+    constructFeatureItem({
+      featureId: TestFeature.Words,
+      includedUsage: 300,
+    }),
+  ],
+  type: "pro",
+});
+
+const ops = [
+  {
+    entityId: "1",
+    product: pro,
+    results: [{ product: pro, status: CusProductStatus.Active }],
+  },
+  {
+    entityId: "2",
+    product: pro,
+    results: [{ product: pro, status: CusProductStatus.Active }],
+  },
+];
+
+const testCase = "multiAttach1";
+describe(`${chalk.yellowBright("multiAttach1: Testing attach multiple products with diff quantities")}`, () => {
+  let customerId = testCase;
+  let autumn: AutumnInt = new AutumnInt({ version: APIVersion.v1_4 });
+
+  let stripeCli: Stripe;
+  let testClockId: string;
+  let curUnix: number;
+  let db: DrizzleCli;
+  let org: Organization;
+  let env: AppEnv;
+
+  before(async function () {
+    await setupBefore(this);
+    const { autumnJs } = this;
+    db = this.db;
+    org = this.org;
+    env = this.env;
+
+    stripeCli = this.stripeCli;
+
+    addPrefixToProducts({
+      products: [pro, premium, growth],
+      prefix: testCase,
+    });
+
+    await createProducts({
+      autumn: autumnJs,
+      products: [pro, premium, growth],
+      db,
+      orgId: org.id,
+      env,
+      customerId,
+    });
+
+    const { testClockId: testClockId1 } = await initCustomer({
+      autumn: autumnJs,
+      customerId,
+      db,
+      org,
+      env,
+      attachPm: "success",
+    });
+
+    testClockId = testClockId1!;
+  });
+
+  const entities = [
+    {
+      id: "1",
+      name: "Entity 1",
+      feature_id: TestFeature.Users,
+    },
+    {
+      id: "2",
+      name: "Entity 2",
+      feature_id: TestFeature.Users,
+    },
+  ];
+
+  it("should attach multiple products with diff quantities", async function () {
+    const productsList = [
+      {
+        product_id: pro.id,
+        quantity: 5,
+      },
+      {
+        product_id: premium.id,
+        quantity: 3,
+      },
+    ];
+
+    const { checkout_url } = await autumn.attach({
+      customer_id: customerId,
+      // @ts-ignore
+      products: productsList,
+      force_checkout: true,
+    });
+
+    await completeCheckoutForm(checkout_url);
+  });
+});