openclaw/src/media-understanding/runner.test-utils.ts

35 lines
1.1 KiB
TypeScript

import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import type { MsgContext } from "../auto-reply/templating.js";
import { withEnvAsync } from "../test-utils/env.js";
import { createMediaAttachmentCache, normalizeMediaAttachments } from "./runner.js";
type AudioFixtureParams = {
ctx: MsgContext;
media: ReturnType<typeof normalizeMediaAttachments>;
cache: ReturnType<typeof createMediaAttachmentCache>;
};
export async function withAudioFixture(
filePrefix: string,
run: (params: AudioFixtureParams) => Promise<void>,
) {
const tmpPath = path.join(os.tmpdir(), filePrefix + "-" + Date.now().toString() + ".wav");
await fs.writeFile(tmpPath, Buffer.from("RIFF"));
const ctx: MsgContext = { MediaPath: tmpPath, MediaType: "audio/wav" };
const media = normalizeMediaAttachments(ctx);
const cache = createMediaAttachmentCache(media, {
localPathRoots: [path.dirname(tmpPath)],
});
try {
await withEnvAsync({ PATH: "" }, async () => {
await run({ ctx, media, cache });
});
} finally {
await cache.cleanup();
await fs.unlink(tmpPath).catch(() => {});
}
}