diff --git a/tests/unit/native-codex-turn-pin-10379.test.ts b/tests/unit/native-codex-turn-pin-10379.test.ts index df469648b7c..a9c0cef4468 100644 --- a/tests/unit/native-codex-turn-pin-10379.test.ts +++ b/tests/unit/native-codex-turn-pin-10379.test.ts @@ -20,6 +20,7 @@ const { getCircuitBreaker, resetAllCircuitBreakers } = await import("../../src/shared/utils/circuitBreaker.ts"); const { recordProviderCooldown, clearCooldownState } = await import("../../open-sse/services/providerCooldownTracker.ts"); +const { PROVIDER_PROFILES } = await import("../../open-sse/config/constants.ts"); const { resolveResilienceSettings } = await import("../../src/lib/resilience/settings.ts"); const BODY = { @@ -224,7 +225,9 @@ test("isPinnedTargetModelScopedUnusable distinguishes model lockout from provide // Reset breaker, test provider cooldown cb.reset(); - recordProviderCooldown("antigravity", undefined, resilienceSettings); + for (let i = 0; i < PROVIDER_PROFILES.oauth.providerFailureThreshold; i += 1) { + recordProviderCooldown("antigravity", undefined, resilienceSettings); + } assert.equal( await isPinnedTargetModelScopedUnusable({ target, diff --git a/tests/unit/native-codex-turn-pin-model-scoped-fallback.test.ts b/tests/unit/native-codex-turn-pin-model-scoped-fallback.test.ts index 314dd3efdba..6ca38922dc5 100644 --- a/tests/unit/native-codex-turn-pin-model-scoped-fallback.test.ts +++ b/tests/unit/native-codex-turn-pin-model-scoped-fallback.test.ts @@ -19,6 +19,7 @@ const { } = await import("../../open-sse/services/combo/nativeCodexTurnPin.ts"); const { recordProviderCooldown, isProviderInCooldown, clearCooldownState } = await import("../../open-sse/services/providerCooldownTracker.ts"); +const { PROVIDER_PROFILES } = await import("../../open-sse/config/constants.ts"); const { getCircuitBreaker, resetAllCircuitBreakers } = await import("../../src/shared/utils/circuitBreaker.ts"); const { resolveResilienceSettings } = await import("../../src/lib/resilience/settings.ts"); @@ -403,7 +404,9 @@ describe("Native Codex Turn Pin model-scoped fallback", () => { allCombos: null, }); - recordProviderCooldown("antigravity", undefined, settings); + for (let i = 0; i < PROVIDER_PROFILES.oauth.providerFailureThreshold; i += 1) { + recordProviderCooldown("antigravity", undefined, settings); + } assert.equal(isProviderInCooldown("antigravity", undefined, settings), true); const attempted: string[] = [];