Files
cfw-autumn/server/tests/check/basic/check4.test.ts
2025-10-22 22:03:52 +01:00

97 lines
2.4 KiB
TypeScript

import { beforeAll, describe, expect, test } from "bun:test";
import {
ApiVersion,
type CheckResponse,
type CheckResponseV0,
SuccessCode,
} from "@autumn/shared";
import chalk from "chalk";
import { TestFeature } from "tests/setup/v2Features.js";
import ctx from "tests/utils/testInitUtils/createTestContext.js";
import { AutumnInt } from "@/external/autumn/autumnCli.js";
import { constructFeatureItem } from "@/utils/scriptUtils/constructItem.js";
import { constructProduct } from "@/utils/scriptUtils/createTestProducts.js";
import { initCustomerV3 } from "@/utils/scriptUtils/testUtils/initCustomerV3.js";
import { initProductsV0 } from "@/utils/scriptUtils/testUtils/initProductsV0.js";
const messagesFeature = constructFeatureItem({
featureId: TestFeature.Messages,
unlimited: true,
});
const freeProd = constructProduct({
type: "free",
isDefault: false,
items: [messagesFeature],
});
const testCase = "check4";
describe(`${chalk.yellowBright("check4: test /check on unlimited feature")}`, () => {
const customerId = "check4";
const autumnV0: AutumnInt = new AutumnInt({ version: ApiVersion.V0_2 });
const autumnV1: AutumnInt = new AutumnInt({ version: ApiVersion.V1_2 });
beforeAll(async () => {
await initCustomerV3({
ctx,
customerId,
withTestClock: false,
});
await initProductsV0({
ctx,
products: [freeProd],
prefix: testCase,
});
await autumnV1.attach({
customer_id: customerId,
product_id: freeProd.id,
});
});
test("v0 response", async () => {
const res = (await autumnV0.check({
customer_id: customerId,
feature_id: TestFeature.Messages,
})) as unknown as CheckResponseV0;
expect(res.allowed).toBe(true);
expect(res.balances).toBeDefined();
expect(res.balances).toHaveLength(1);
expect(res.balances[0]).toStrictEqual({
balance: null,
feature_id: TestFeature.Messages,
unlimited: true,
usage_allowed: false,
required: null,
});
});
test("v1 response", async () => {
const res = (await autumnV1.check({
customer_id: customerId,
feature_id: TestFeature.Messages,
})) as unknown as CheckResponse;
console.log(res);
const expectedRes = {
allowed: true,
customer_id: customerId,
feature_id: TestFeature.Messages as string,
required_balance: 1,
code: SuccessCode.FeatureFound,
unlimited: true,
balance: null,
usage: 0,
included_usage: 0,
next_reset_at: null,
overage_allowed: false,
};
expect(expectedRes).toMatchObject(res);
});
});