test: 💍 bun

server/tests/advanced/customInterval/customInterval3.backup.ts

@@ -0,0 +1,160 @@
+import { type AppEnv, LegacyVersion, type Organization } from "@autumn/shared";
+import { expect } from "chai";
+import chalk from "chalk";
+import { addDays, addMonths } from "date-fns";
+import type Stripe from "stripe";
+import { addPrefixToProducts } from "tests/attach/utils.js";
+import { setupBefore } from "tests/before.js";
+import { TestFeature } from "tests/setup/v2Features.js";
+import { attachAndExpectCorrect } from "tests/utils/expectUtils/expectAttach.js";
+import { expectProductAttached } from "tests/utils/expectUtils/expectProductAttached.js";
+import { createProducts } from "tests/utils/productUtils.js";
+import { advanceTestClock } from "tests/utils/stripeUtils.js";
+import type { DrizzleCli } from "@/db/initDrizzle.js";
+import { AutumnInt } from "@/external/autumn/autumnCli.js";
+import { calculateProrationAmount } from "@/internal/invoices/prorationUtils.js";
+import {
+	constructFeatureItem,
+	constructPrepaidItem,
+} from "@/utils/scriptUtils/constructItem.js";
+import {
+	constructProduct,
+	constructRawProduct,
+} from "@/utils/scriptUtils/createTestProducts.js";
+import { initCustomer } from "@/utils/scriptUtils/initCustomer.js";
+
+const testCase = "customInterval3";
+
+export const pro = constructProduct({
+	type: "pro",
+	items: [
+		constructFeatureItem({
+			featureId: TestFeature.Words,
+			intervalCount: 2,
+		}),
+	],
+	intervalCount: 2,
+});
+
+const prepaidWordsItem = constructPrepaidItem({
+	featureId: TestFeature.Words,
+	price: 10,
+	billingUnits: 1,
+	includedUsage: 0,
+	intervalCount: 2,
+});
+
+export const addOn = constructRawProduct({
+	id: "addOn",
+	items: [prepaidWordsItem],
+	isAddOn: true,
+});
+
+describe(`${chalk.yellowBright(`${testCase}: Testing custom interval on add on merged product`)}`, () => {
+	const customerId = testCase;
+	const autumn: AutumnInt = new AutumnInt({ version: LegacyVersion.v1_4 });
+	let testClockId: string;
+	let db: DrizzleCli, org: Organization, env: AppEnv;
+	let stripeCli: Stripe;
+
+	before(async function () {
+		await setupBefore(this);
+		const { autumnJs } = this;
+		db = this.db;
+		org = this.org;
+		env = this.env;
+
+		stripeCli = this.stripeCli;
+
+		const { testClockId: testClockId1 } = await initCustomer({
+			autumn: autumnJs,
+			customerId,
+			db,
+			org,
+			env,
+			attachPm: "success",
+		});
+
+		addPrefixToProducts({
+			products: [pro, addOn],
+			prefix: testCase,
+		});
+
+		await createProducts({
+			autumn,
+			products: [pro, addOn],
+			db,
+			orgId: org.id,
+			env,
+		});
+
+		testClockId = testClockId1!;
+	});
+
+	it("should attach pro product", async () => {
+		await attachAndExpectCorrect({
+			autumn,
+			customerId,
+			product: pro,
+			stripeCli,
+			db,
+			org,
+			env,
+		});
+	});
+
+	it("should upgrade to attached add on and have correct invoice next cycle", async () => {
+		const curUnix = await advanceTestClock({
+			stripeCli,
+			testClockId,
+			advanceTo: addDays(new Date(), 20).getTime(),
+			waitForSeconds: 15,
+		});
+
+		const wordBillingSets = 2;
+		const wordsBillingUnits = prepaidWordsItem.billing_units! * wordBillingSets;
+		await autumn.attach({
+			customer_id: customerId,
+			product_id: addOn.id,
+			options: [
+				{
+					feature_id: TestFeature.Words,
+					quantity: wordsBillingUnits,
+				},
+			],
+		});
+
+		const customer = await autumn.customers.get(customerId);
+		const proProduct = customer.products.find((p) => p.id === pro.id);
+		const invoices = customer.invoices;
+		expectProductAttached({
+			customer,
+			product: pro,
+		});
+
+		expectProductAttached({
+			customer,
+			product: addOn,
+		});
+
+		const expectedPrice = wordsBillingUnits * prepaidWordsItem.price!;
+		const proratedPrice = calculateProrationAmount({
+			amount: expectedPrice,
+			periodStart: new Date().getTime(),
+			periodEnd: addMonths(new Date(), 2).getTime(),
+			now: curUnix!,
+		});
+
+		expect(invoices[0].product_ids).to.include(addOn.id);
+		expect(invoices[0].total).to.approximately(proratedPrice, 0.1);
+
+		const expectedAddonEnd = addMonths(new Date(), 2);
+		const approximate = 1000 * 60 * 60 * 24; // +- 1 day
+		const addOnProduct = customer.products.find((p) => p.id === addOn.id);
+
+		expect(addOnProduct?.current_period_end).to.be.approximately(
+			expectedAddonEnd.getTime(),
+			approximate,
+		);
+	});
+});