test: isolate pdf media state env

This commit is contained in:
Shakker
2026-06-05 16:56:05 +01:00
parent 002aa1061b
commit 415272d17e

View File

@@ -7,6 +7,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import type { OpenClawConfig } from "../../config/config.js"; import type { OpenClawConfig } from "../../config/config.js";
import * as pdfExtractModule from "../../media/pdf-extract.js"; import * as pdfExtractModule from "../../media/pdf-extract.js";
import * as webMedia from "../../media/web-media.js"; import * as webMedia from "../../media/web-media.js";
import { withEnvAsync } from "../../test-utils/env.js";
import * as modelDiscovery from "../agent-model-discovery.js"; import * as modelDiscovery from "../agent-model-discovery.js";
import type { AuthProfileStore } from "../auth-profiles/types.js"; import type { AuthProfileStore } from "../auth-profiles/types.js";
import * as modelAuth from "../model-auth.js"; import * as modelAuth from "../model-auth.js";
@@ -168,9 +169,10 @@ async function withManagedInboundPdf(
const mediaPath = path.join(inboundDir, mediaId); const mediaPath = path.join(inboundDir, mediaId);
await fs.mkdir(inboundDir, { recursive: true }); await fs.mkdir(inboundDir, { recursive: true });
await fs.writeFile(mediaPath, FAKE_PDF_MEDIA.buffer); await fs.writeFile(mediaPath, FAKE_PDF_MEDIA.buffer);
vi.stubEnv("OPENCLAW_STATE_DIR", stateDir);
try { try {
await run({ stateDir, mediaId, mediaPath }); await withEnvAsync({ OPENCLAW_STATE_DIR: stateDir }, async () => {
await run({ stateDir, mediaId, mediaPath });
});
} finally { } finally {
await fs.rm(stateDir, { recursive: true, force: true }); await fs.rm(stateDir, { recursive: true, force: true });
} }
@@ -186,7 +188,6 @@ describe("createPdfTool", () => {
afterEach(() => { afterEach(() => {
vi.restoreAllMocks(); vi.restoreAllMocks();
vi.unstubAllEnvs();
global.fetch = priorFetch; global.fetch = priorFetch;
}); });