109 lines
3.2 KiB
TypeScript
109 lines
3.2 KiB
TypeScript
import { beforeAll, describe, expect, test } from "bun:test";
|
|
import { ApiVersion, CusProductStatus } from "@autumn/shared";
|
|
import chalk from "chalk";
|
|
import type Stripe from "stripe";
|
|
import { AutumnCli } from "tests/cli/AutumnCli.js";
|
|
import { expectCustomerV0Correct } from "tests/utils/expectUtils/expectCustomerV0Correct.js";
|
|
import ctx from "tests/utils/testInitUtils/createTestContext.js";
|
|
import { AutumnInt } from "@/external/autumn/autumnCli.js";
|
|
import { createStripeCli } from "@/external/connect/createStripeCli.js";
|
|
import { timeout } from "@/utils/genUtils.js";
|
|
import { initCustomerV3 } from "@/utils/scriptUtils/testUtils/initCustomerV3.js";
|
|
import { sharedDefaultFree, sharedProProduct, initBasicSharedProducts } from "./sharedProducts.js";
|
|
|
|
const testCase = "basic3";
|
|
const customerId = testCase;
|
|
|
|
describe(`${chalk.yellowBright("basic3: Testing cancel through Stripe at period end and now")}`, () => {
|
|
const autumnV1 = new AutumnInt({
|
|
secretKey: ctx.orgSecretKey,
|
|
version: ApiVersion.V1_2,
|
|
});
|
|
|
|
let stripeCli: Stripe;
|
|
|
|
beforeAll(async () => {
|
|
stripeCli = createStripeCli({ org: ctx.org, env: ctx.env });
|
|
|
|
// Explicitly ensure shared products exist
|
|
await initBasicSharedProducts();
|
|
|
|
// Then create customer with payment method
|
|
await initCustomerV3({
|
|
ctx,
|
|
customerId,
|
|
attachPm: "success",
|
|
withTestClock: true,
|
|
withDefault: true,
|
|
});
|
|
});
|
|
|
|
test("should attach pro product", async () => {
|
|
await autumnV1.attach({
|
|
customer_id: customerId,
|
|
product_id: sharedProProduct.id,
|
|
});
|
|
|
|
const res = await AutumnCli.getCustomer(customerId);
|
|
await expectCustomerV0Correct({
|
|
sent: sharedProProduct,
|
|
cusRes: res,
|
|
});
|
|
});
|
|
|
|
test("should cancel pro product (at period end)", async () => {
|
|
const cusRes: any = await AutumnCli.getCustomer(customerId);
|
|
|
|
const proProduct = cusRes.products.find(
|
|
(p: any) => p.id === sharedProProduct.id,
|
|
);
|
|
|
|
for (const subId of proProduct.subscription_ids) {
|
|
await stripeCli.subscriptions.update(subId, {
|
|
cancel_at_period_end: true,
|
|
});
|
|
}
|
|
await timeout(5000);
|
|
});
|
|
|
|
test("should have pro product active, and canceled_at != null, and free scheduled", async () => {
|
|
const cusRes: any = await AutumnCli.getCustomer(customerId);
|
|
await expectCustomerV0Correct({
|
|
sent: sharedProProduct,
|
|
cusRes: cusRes,
|
|
});
|
|
|
|
const proProduct = cusRes.products.find(
|
|
(p: any) => p.id === sharedProProduct.id,
|
|
);
|
|
expect(proProduct.canceled_at).not.toBe(null);
|
|
expect(proProduct.status).toBe(CusProductStatus.Active);
|
|
|
|
const freeProduct = cusRes.products.find(
|
|
(p: any) => p.id === sharedDefaultFree.id,
|
|
);
|
|
expect(freeProduct).toBeDefined();
|
|
expect(freeProduct.status).toBe(CusProductStatus.Scheduled);
|
|
});
|
|
|
|
test("should cancel pro product (now)", async () => {
|
|
const cusRes: any = await AutumnCli.getCustomer(customerId);
|
|
const proProduct = cusRes.products.find(
|
|
(p: any) => p.id === sharedProProduct.id,
|
|
);
|
|
|
|
for (const subId of proProduct.subscription_ids) {
|
|
await stripeCli.subscriptions.cancel(subId);
|
|
}
|
|
await timeout(5000);
|
|
});
|
|
|
|
test("should have free product active, and no pro product", async () => {
|
|
const cusRes: any = await AutumnCli.getCustomer(customerId);
|
|
await expectCustomerV0Correct({
|
|
sent: sharedDefaultFree,
|
|
cusRes: cusRes,
|
|
});
|
|
});
|
|
});
|