writing tests for migrations

server/tests/attach/migrations/runMigrationTest.ts

@@ -0,0 +1,115 @@
+import { DrizzleCli } from "@/db/initDrizzle.js";
+import { AutumnInt } from "@/external/autumn/autumnCli.js";
+import { ProductV2, Organization } from "@autumn/shared";
+import { AppEnv } from "autumn-js";
+import {
+  expectSubItemsCorrect,
+  getSubsFromCusId,
+} from "tests/utils/expectUtils/expectSubUtils.js";
+import Stripe from "stripe";
+import { expect } from "chai";
+import { expectFeaturesCorrect } from "tests/utils/expectUtils/expectFeaturesCorrect.js";
+import { expectResetAtCorrect } from "tests/utils/expectUtils/expectAttach/expectResetAtCorrect.js";
+import { isFreeProductV2 } from "@/internal/products/productUtils/classifyProduct.js";
+import { expectTrialEndsAtCorrect } from "tests/utils/expectUtils/expectAttach/expectTrialEndsAt.js";
+import { timeout } from "@/utils/genUtils.js";
+
+export const expectSubsSame = ({
+  subsBefore,
+  subsAfter,
+}: {
+  subsBefore: Stripe.Subscription[];
+  subsAfter: Stripe.Subscription[];
+}) => {
+  let invoicesBefore = subsBefore.map((sub) => sub.latest_invoice);
+  let invoicesAfter = subsAfter.map((sub) => sub.latest_invoice);
+  let subIdsBefore = subsBefore.map((sub) => sub.id);
+  let subIdsAfter = subsAfter.map((sub) => sub.id);
+  let periodEndsBefore = subsBefore.map((sub) => sub.current_period_end);
+  let periodEndsAfter = subsAfter.map((sub) => sub.current_period_end);
+
+  expect(invoicesAfter).to.deep.equal(invoicesBefore);
+  expect(subIdsAfter).to.deep.equal(subIdsBefore);
+  expect(periodEndsAfter).to.deep.equal(periodEndsBefore);
+};
+
+export const runMigrationTest = async ({
+  autumn,
+  stripeCli,
+  customerId,
+  fromProduct,
+  toProduct,
+  db,
+  org,
+  env,
+  usage,
+  numInvoices = 1,
+}: {
+  autumn: AutumnInt;
+  stripeCli: Stripe;
+  customerId: string;
+  fromProduct: ProductV2;
+  toProduct: ProductV2;
+  db: DrizzleCli;
+  org: Organization;
+  env: AppEnv;
+  usage?: {
+    featureId: string;
+    value: number;
+  }[];
+  numInvoices?: number;
+}) => {
+  const { subs: subsBefore } = await getSubsFromCusId({
+    stripeCli,
+    customerId,
+    productId: fromProduct.id,
+    db,
+    org,
+    env,
+  });
+
+  const cusBefore = await autumn.customers.get(customerId);
+
+  await autumn.migrate({
+    from_product_id: fromProduct.id,
+    to_product_id: toProduct.id,
+    from_version: fromProduct.version,
+    to_version: toProduct.version,
+  });
+
+  await timeout(5000);
+
+  const { subs: subsAfter } = await getSubsFromCusId({
+    stripeCli,
+    customerId,
+    productId: toProduct.id,
+    db,
+    org,
+    env,
+  });
+
+  expectSubsSame({ subsBefore, subsAfter });
+
+  const cusAfter = await autumn.customers.get(customerId);
+  expectFeaturesCorrect({
+    customer: cusAfter,
+    product: toProduct,
+    usage,
+  });
+
+  expectResetAtCorrect({ cusBefore, cusAfter });
+  expectTrialEndsAtCorrect({ cusBefore, cusAfter });
+
+  await expectSubItemsCorrect({
+    stripeCli,
+    customerId,
+    product: toProduct,
+    db,
+    org,
+    env,
+  });
+
+  if (!isFreeProductV2({ product: toProduct })) {
+    expect(cusAfter.invoices.length).to.equal(numInvoices);
+  }
+};