79 lines
3.3 KiB
TypeScript
79 lines
3.3 KiB
TypeScript
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { DEFAULT_BILLING_PRICE_RULES, ensureDefaultBillingPriceRules } from "@/lib/server/billing-catalog";
|
|
import { quoteGenerationCharge } from "@/lib/server/billing-service";
|
|
import { createBillingPriceRule, getBillingPriceRule, updateBillingPriceTierMultiplier } from "@/lib/server/billing-store";
|
|
|
|
let runtimeDir = "";
|
|
|
|
describe("default billing catalog refresh", () => {
|
|
beforeEach(async () => {
|
|
runtimeDir = await mkdtemp(join(tmpdir(), "zhinian-billing-seed-"));
|
|
vi.stubEnv("ZHINIAN_RUNTIME_DIR", runtimeDir);
|
|
vi.stubEnv("ZHINIAN_DATA_BACKEND", "local");
|
|
});
|
|
|
|
afterEach(async () => {
|
|
vi.unstubAllEnvs();
|
|
await rm(runtimeDir, { recursive: true, force: true });
|
|
});
|
|
|
|
it("keeps a saved tier multiplier across catalog refresh and quotation", async () => {
|
|
await ensureDefaultBillingPriceRules();
|
|
await updateBillingPriceTierMultiplier({
|
|
ruleId: "base-seedream-5-0-pro",
|
|
dimensionKey: "size",
|
|
tierValue: "1.5K",
|
|
markupMultiplier: 1.3
|
|
});
|
|
|
|
const quote = await quoteGenerationCharge({
|
|
provider: "seedream",
|
|
capability: "image.generate",
|
|
reqKey: "doubao-seedream-5-0-pro-260628",
|
|
requestPayload: { settings: { size: "1.5K" } },
|
|
usageContext: { source: "platform", accountId: "user-1", displayName: "测试用户", organizationId: "org-1" }
|
|
});
|
|
expect(quote).toMatchObject({ standardUnitPriceFen: 30, markupMultiplier: 1.3, amountFen: 39 });
|
|
|
|
const saved = await getBillingPriceRule("base-seedream-5-0-pro");
|
|
const tiers = saved?.parameterDimensions?.find((dimension) => dimension.key === "size")?.tiers;
|
|
expect(tiers?.find((tier) => tier.value === "1.5K")?.markupMultiplier).toBe(1.3);
|
|
expect(tiers?.find((tier) => tier.value === "2K")?.markupMultiplier).toBe(1.2);
|
|
});
|
|
|
|
it("takes current default definitions while retaining valid matching tier markups", async () => {
|
|
const candidate = DEFAULT_BILLING_PRICE_RULES.find((rule) => rule.id === "base-seedream-5-0-pro")!;
|
|
const size = candidate.parameterDimensions!.find((dimension) => dimension.key === "size")!;
|
|
await createBillingPriceRule({
|
|
...candidate,
|
|
standardUnitPriceFen: 5,
|
|
markupMultiplier: 1.7,
|
|
parameterDimensions: [{
|
|
...size,
|
|
label: "旧分辨率",
|
|
tiers: [
|
|
{ ...size.tiers[0], standardFactor: 99, markupMultiplier: 0 },
|
|
{ ...size.tiers[1], standardFactor: 99, markupMultiplier: 1.4 },
|
|
{ value: "3K", label: "旧档位", standardFactor: 9, markupMultiplier: 3, enabled: true }
|
|
]
|
|
}]
|
|
});
|
|
|
|
await ensureDefaultBillingPriceRules();
|
|
const saved = await getBillingPriceRule(candidate.id!);
|
|
expect(saved?.standardUnitPriceFen).toBe(30);
|
|
expect(saved?.markupMultiplier).toBe(1.7);
|
|
const dimension = saved?.parameterDimensions?.find((item) => item.key === "size");
|
|
expect(dimension?.label).toBe("分辨率");
|
|
expect(dimension?.tiers.map((tier) => tier.value)).toEqual(["1K", "1.5K", "2K"]);
|
|
expect(dimension?.tiers.map((tier) => [tier.standardFactor, tier.markupMultiplier])).toEqual([
|
|
[1, 1.2],
|
|
[1, 1.4],
|
|
[2, 1.2]
|
|
]);
|
|
});
|
|
});
|