test: reuse image generate tool imports

This commit is contained in:
Peter Steinberger
2026-04-08 19:48:44 +01:00
parent 4a60087cd0
commit 218078ffd4

View File

@@ -1,4 +1,4 @@
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
let imageGenerationRuntime: typeof import("../../image-generation/runtime.js");
let imageOps: typeof import("../../media/image-ops.js");
@@ -171,8 +171,7 @@ function createFalEditProvider(params?: {
}
describe("createImageGenerateTool", () => {
beforeEach(async () => {
vi.resetModules();
beforeAll(async () => {
vi.doUnmock("../../secrets/provider-env-vars.js");
imageGenerationRuntime = await import("../../image-generation/runtime.js");
imageOps = await import("../../media/image-ops.js");
@@ -180,7 +179,9 @@ describe("createImageGenerateTool", () => {
webMedia = await import("../../media/web-media.js");
({ createImageGenerateTool, resolveImageGenerationModelConfigForTool } =
await import("./image-generate-tool.js"));
});
beforeEach(() => {
vi.stubEnv("OPENAI_API_KEY", "");
vi.stubEnv("OPENAI_API_KEYS", "");
vi.stubEnv("GEMINI_API_KEY", "");