From 325af4dda21c618855b6b43975c536d3912df346 Mon Sep 17 00:00:00 2001 From: Charlie Lamb Date: Wed, 10 Jun 2026 19:59:35 +0100 Subject: [PATCH] fix: mock models.dev pricing in credit cost unit test Co-Authored-By: Claude Fable 5 --- .../unit/features/get-credit-cost.test.ts | 20 ++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/server/tests/unit/features/get-credit-cost.test.ts b/server/tests/unit/features/get-credit-cost.test.ts index 5f64b31fa..5eb2d6f31 100644 --- a/server/tests/unit/features/get-credit-cost.test.ts +++ b/server/tests/unit/features/get-credit-cost.test.ts @@ -1,17 +1,23 @@ -import { describe, expect, test } from "bun:test"; +import { describe, expect, mock, test } from "bun:test"; import { ErrCode, type Feature, FeatureType, FeatureUsageType, } from "@autumn/shared"; -import { - getModelCreditCost, - getModelCreditCostBreakdown, -} from "@/internal/features/aiCreditSystemUtils.js"; -import { getCreditCost } from "@/internal/features/creditSystemUtils.js"; -// Uses custom/* models so pricing resolves offline (no models.dev fetch). +mock.module("@/internal/features/utils/getModelPricing.js", () => ({ + getModelsDevPricing: async () => ({}), +})); + +const { getModelCreditCost, getModelCreditCostBreakdown } = await import( + "@/internal/features/aiCreditSystemUtils.js" +); +const { getCreditCost } = await import( + "@/internal/features/creditSystemUtils.js" +); + +// custom/* models price from model_markups; pricing data is mocked empty. const CUSTOM_MODEL = "custom/foo"; const aiCreditFeature: Feature = {