cfw-autumn/server/tests/advanced/coupons/coupon3.test.ts

import {
	type AppEnv,
	CouponDurationType,
	type CreateReward,
	LegacyVersion,
	type Organization,
	RewardType,
} from "@autumn/shared";
import { beforeAll, describe, expect, test } from "bun:test";
import chalk from "chalk";
import type Stripe from "stripe";
import { TestFeature } from "tests/setup/v2Features.js";
import { expectAttachCorrect } from "tests/utils/expectUtils/expectAttach.js";
import { createProducts, createReward } from "tests/utils/productUtils.js";
import {
	addPrefixToProducts,
	getBasePrice,
} from "tests/utils/testProductUtils/testProductUtils.js";
import ctx from "tests/utils/testInitUtils/createTestContext.js";
import type { DrizzleCli } from "@/db/initDrizzle.js";
import { AutumnInt } from "@/external/autumn/autumnCli.js";
import {
	constructArrearItem,
	constructFeatureItem,
} from "@/utils/scriptUtils/constructItem.js";
import { constructProduct } from "@/utils/scriptUtils/createTestProducts.js";
import { initCustomerV3 } from "@/utils/scriptUtils/testUtils/initCustomerV3.js";

const pro = constructProduct({
	type: "pro",
	items: [constructArrearItem({ featureId: TestFeature.Words })],
});

const oneOff = constructProduct({
	type: "one_off",
	items: [
		constructFeatureItem({
			featureId: TestFeature.Words,
			includedUsage: 500,
		}),
	],
});

// Create reward input
const rewardId = "attach_coupon";
const promoCode = "attach_coupon_code";
const reward: CreateReward = {
	id: rewardId,
	name: "attach_coupon",
	promo_codes: [{ code: promoCode }],
	type: RewardType.FixedDiscount,
	discount_config: {
		discount_value: 5,
		duration_type: CouponDurationType.OneOff,
		duration_value: 1,
		should_rollover: true,
		apply_to_all: true,
		price_ids: [],
	},
};

const testCase = "coupon3";
describe(chalk.yellow(`${testCase} - Testing attach coupon`), () => {
	const customerId = testCase;
	let stripeCli: Stripe;
	let testClockId: string;

	const autumn: AutumnInt = new AutumnInt({ version: LegacyVersion.v1_4 });
	let org: Organization;
	let env: AppEnv;
	let db: DrizzleCli;

	const couponAmount = reward.discount_config!.discount_value;

	beforeAll(async () => {
		org = ctx.org;
		env = ctx.env;
		db = ctx.db;
		stripeCli = ctx.stripeCli;

		const { testClockId: testClockId1 } = await initCustomerV3({
			ctx,
			customerId,
			attachPm: "success",
		});

		testClockId = testClockId1;

		addPrefixToProducts({
			products: [pro, oneOff],
			prefix: testCase,
		});

		await createProducts({
			orgId: ctx.org.id,
			env: ctx.env,
			db: ctx.db,
			autumn,
			products: [pro, oneOff],
		});

		await createReward({
			orgId: org.id,
			env,
			db,
			autumn,
			reward,
			productId: pro.id,
		});
	});

	// CYCLE 0
	test("should attach pro with reward ID", async () => {
		const res = await autumn.attach({
			customer_id: customerId,
			product_id: pro.id,
			reward: rewardId,
		});

		const customer = await autumn.customers.get(customerId);
		expectAttachCorrect({
			customer,
			product: pro,
		});

		const invoice = customer.invoices![0];
		const basePrice = getBasePrice({ product: pro });
		expect(invoice.total).toBe(basePrice - couponAmount);
	});

	test("should attach one off with reward ID", async () => {
		await autumn.attach({
			customer_id: customerId,
			product_id: oneOff.id,
			reward: rewardId,
		});

		const customer = await autumn.customers.get(customerId);
		expectAttachCorrect({
			customer,
			product: oneOff,
		});

		const invoice = customer.invoices![0];
		const basePrice = getBasePrice({ product: oneOff });
		expect(invoice.total).toBe(basePrice - couponAmount);
		expect(invoice.product_ids).toContain(oneOff.id);
	});

	test("should attach one off with promo code", async () => {
		await autumn.attach({
			customer_id: customerId,
			product_id: oneOff.id,
			reward: promoCode,
		});

		const customer = await autumn.customers.get(customerId);
		expectAttachCorrect({
			customer,
			product: oneOff,
		});

		expect(customer.invoices!.length).toBe(3);
		const basePrice = getBasePrice({ product: oneOff });
		for (let i = 0; i < 2; i++) {
			const invoice = customer.invoices![i];
			expect(invoice.total).toBe(basePrice - couponAmount);
			expect(invoice.product_ids).toContain(oneOff.id);
		}
	});
});