From faea23cffab17611a13b9e81f379f97bddfbdfe8 Mon Sep 17 00:00:00 2001
From: Shakker <shakkerdroid@gmail.com>
Date: Sat, 9 May 2026 16:59:41 +0100
Subject: [PATCH] test: avoid provider policy module resets

---
 .../models-config.providers.policy.test.ts    | 22 ++++++-------------
 1 file changed, 7 insertions(+), 15 deletions(-)

diff --git a/src/agents/models-config.providers.policy.test.ts b/src/agents/models-config.providers.policy.test.ts
index 1d683f4dd7a..f0056432903 100644
--- a/src/agents/models-config.providers.policy.test.ts
+++ b/src/agents/models-config.providers.policy.test.ts
@@ -1,13 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
-
-type NormalizeProviderSpecificConfig =
-  typeof import("./models-config.providers.policy.js").normalizeProviderSpecificConfig;
-type ResolveProviderConfigApiKeyResolver =
-  typeof import("./models-config.providers.policy.js").resolveProviderConfigApiKeyResolver;
-
-const GOOGLE_BASE_URL = "https://generativelanguage.googleapis.com";
-let normalizeProviderSpecificConfig: NormalizeProviderSpecificConfig;
-let resolveProviderConfigApiKeyResolver: ResolveProviderConfigApiKeyResolver;
+import { describe, expect, it, vi } from "vitest";
 
 vi.mock("../plugins/provider-runtime.js", () => ({
   applyProviderNativeStreamingUsageCompatWithPlugin: () => undefined,
@@ -46,11 +37,12 @@ vi.mock("../plugins/provider-runtime.js", () => ({
   },
 }));
 
-beforeEach(async () => {
-  vi.resetModules();
-  ({ normalizeProviderSpecificConfig, resolveProviderConfigApiKeyResolver } =
-    await import("./models-config.providers.policy.js"));
-});
+import {
+  normalizeProviderSpecificConfig,
+  resolveProviderConfigApiKeyResolver,
+} from "./models-config.providers.policy.js";
+
+const GOOGLE_BASE_URL = "https://generativelanguage.googleapis.com";
 
 describe("models-config.providers.policy", () => {
   it("resolves config apiKey markers through provider plugin hooks", () => {