wehub-resource-sync
42 lines
1.3 KiB
TypeScript
42 lines
1.3 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import type { ImageContent } from "@oh-my-pi/pi-ai";
|
|
import type { Args } from "@oh-my-pi/pi-coding-agent/cli/args";
|
|
import { buildInitialMessage } from "@oh-my-pi/pi-coding-agent/cli/initial-message";
|
|
|
|
function createArgs(messages: string[]): Args {
|
|
return {
|
|
messages,
|
|
fileArgs: [],
|
|
unknownFlags: new Map(),
|
|
unrecognizedFlags: [],
|
|
};
|
|
}
|
|
|
|
describe("buildInitialMessage", () => {
|
|
it("combines stdin, file text, and the first CLI message", () => {
|
|
const parsed = createArgs(["first", "second"]);
|
|
const images: ImageContent[] = [{ type: "image", data: "abc123", mimeType: "image/png" }];
|
|
|
|
const result = buildInitialMessage({
|
|
parsed,
|
|
stdinContent: "stdin",
|
|
fileText: "file-",
|
|
fileImages: images,
|
|
});
|
|
|
|
expect(result.initialMessage).toBe("stdin\nfile-first");
|
|
expect(result.initialImages).toEqual(images);
|
|
expect(parsed.messages).toEqual(["second"]);
|
|
});
|
|
|
|
it("leaves plain CLI messages untouched when there is no initial file or stdin input", () => {
|
|
const parsed = createArgs(["first", "second"]);
|
|
|
|
const result = buildInitialMessage({ parsed });
|
|
|
|
expect(result.initialMessage).toBeUndefined();
|
|
expect(result.initialImages).toBeUndefined();
|
|
expect(parsed.messages).toEqual(["first", "second"]);
|
|
});
|
|
});
|