fix(billing): apply margin multiplier to subscription quota consumption

Subscription plans now deduct period_tokens_used as upstream tokens × marginMultiplier, aligning package quota burn with wallet cost-mode pricing. Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-28 20:31:38 +08:00
parent e85b9b986a
commit 6f48cad4c2
6 changed files with 146 additions and 9 deletions
@@ -1,6 +1,10 @@
 import test from 'node:test';
 import assert from 'node:assert/strict';
-import { computeDeltaCostCents, normalizeTokenState } from './billing.mjs';
+import {
+  computeDeltaCostCents,
+  computeSubscriptionQuotaTokens,
+  normalizeTokenState,
+} from './billing.mjs';

 const config = {
  useBackendCost: false,
@@ -76,3 +80,14 @@ test('computeDeltaCostCents falls back to token path when accumulatedCost missin
  const costConfig = { ...config, useBackendCost: true, marginMultiplier: 3 };
  assert.equal(computeDeltaCostCents(previous, current, costConfig), 5);
 });
+
+test('computeSubscriptionQuotaTokens scales subscription quota by marginMultiplier', () => {
+  const billing = { ...config, marginMultiplier: 6 };
+  assert.equal(computeSubscriptionQuotaTokens(8000, 2000, billing), 60_000);
+  assert.equal(computeSubscriptionQuotaTokens(100, 0, billing), 600);
+});
+
+test('computeSubscriptionQuotaTokens leaves raw tokens when margin is 1', () => {
+  assert.equal(computeSubscriptionQuotaTokens(8000, 2000, config), 10_000);
+  assert.equal(computeSubscriptionQuotaTokens(0, 0, config), 0);
+});