Files
cfw-autumn/server/tests/attach/basic/basic4.test.ts

114 lines
3.1 KiB
TypeScript

import { beforeAll, describe, expect, test } from "bun:test";
import chalk from "chalk";
import { AutumnCli } from "tests/cli/AutumnCli.js";
import { features, products } from "tests/global.js";
import { compareMainProduct } from "tests/utils/compare.js";
import { createProducts } from "tests/utils/productUtils.js";
import ctx from "tests/utils/testInitUtils/createTestContext.js";
import { AutumnInt } from "@/external/autumn/autumnCli.js";
import { constructPrepaidItem } from "@/utils/scriptUtils/constructItem.js";
import { constructRawProduct } from "@/utils/scriptUtils/createTestProducts.js";
import { initCustomerV3 } from "@/utils/scriptUtils/testUtils/initCustomerV3.js";
const monthlyItem = constructPrepaidItem({
featureId: features.metered1.id,
price: 9,
billingUnits: 250,
});
const monthly = constructRawProduct({
id: "basic4_monthly",
items: [monthlyItem],
});
const testCase = "basic4";
describe(`${chalk.yellowBright("basic4: Testing attach monthly add on")}`, () => {
const customerId = testCase;
const autumn: AutumnInt = new AutumnInt();
beforeAll(async () => {
await initCustomerV3({
ctx,
customerId,
attachPm: "success",
withTestClock: true,
});
await createProducts({
autumn: autumn,
db: ctx.db,
orgId: ctx.org.id,
env: ctx.env,
products: [monthly],
});
});
test("should attach pro", async () => {
await autumn.attach({
customer_id: customerId,
product_id: products.pro.id,
});
const res = await AutumnCli.getCustomer(customerId);
compareMainProduct({
sent: products.pro,
cusRes: res,
});
});
const monthlyQuantity = 500;
test("should attach monthly add on", async () => {
await AutumnCli.attach({
customerId: customerId,
productId: products.monthlyAddOnMetered1.id,
forceCheckout: false,
options: [
{
feature_id: features.metered1.id,
quantity: monthlyQuantity,
},
],
});
});
test("should have correct product & entitlements", async () => {
const cusRes = await AutumnCli.getCustomer(customerId);
const proMetered1 = products.pro.entitlements.metered1.allowance;
const monthlyMetered1Balance = cusRes.entitlements.find(
(e: any) =>
e.feature_id === features.metered1.id &&
e.interval ===
products.monthlyAddOnMetered1.entitlements.metered1.interval,
);
expect(monthlyMetered1Balance!.balance).toBe(proMetered1! + monthlyQuantity);
expect(cusRes.add_ons).toHaveLength(1);
const monthlyAddOnId = cusRes.add_ons.find(
(a: any) => a.id === products.monthlyAddOnMetered1.id,
);
expect(monthlyAddOnId).toBeDefined();
expect(cusRes.invoices.length).toBe(2);
});
test("should have correct /check result for metered1", async () => {
const res: any = await AutumnCli.entitled(customerId, features.metered1.id);
const metered1Balance = res!.balances.find(
(b: any) => b.feature_id === features.metered1.id,
);
const proMetered1Amt = products.pro.entitlements.metered1.allowance;
const monthlyAddOnMetered1Amt = monthlyQuantity;
expect(metered1Balance!.balance).toBe(
proMetered1Amt! + monthlyAddOnMetered1Amt,
);
});
});