mirror of
https://github.com/openclaw/openclaw.git
synced 2026-06-06 05:51:15 +08:00
test: isolate pdf media state env
This commit is contained in:
@@ -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;
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user