enieuwy
54 lines
2.1 KiB
TypeScript
54 lines
2.1 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "bun:test";
|
|
import { CommandController } from "@oh-my-pi/pi-coding-agent/modes/controllers/command-controller";
|
|
import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
|
|
import * as native from "@oh-my-pi/pi-natives";
|
|
|
|
function createController(options: { assistantText?: string; hasAssistantMessage?: boolean; handoffText?: string }) {
|
|
const showStatus = vi.fn();
|
|
const showError = vi.fn();
|
|
const ctx = {
|
|
session: {
|
|
getLastAssistantText: () => options.assistantText,
|
|
hasCopyCandidateAssistantMessage: () => options.hasAssistantMessage ?? options.assistantText !== undefined,
|
|
getLastVisibleHandoffText: () => options.handoffText,
|
|
},
|
|
showStatus,
|
|
showError,
|
|
} as unknown as InteractiveModeContext;
|
|
|
|
return { controller: new CommandController(ctx), showStatus, showError };
|
|
}
|
|
|
|
describe("/copy command", () => {
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
it("falls back to the fresh handoff context when no assistant message exists", () => {
|
|
const copySpy = vi.spyOn(native, "copyToClipboard").mockImplementation(() => undefined);
|
|
const { controller, showStatus, showError } = createController({
|
|
handoffText: "<handoff-context>\n## Goal\nContinue\n</handoff-context>",
|
|
});
|
|
|
|
controller.handleCopyCommand();
|
|
|
|
expect(copySpy).toHaveBeenCalledWith("<handoff-context>\n## Goal\nContinue\n</handoff-context>");
|
|
expect(showStatus).toHaveBeenCalledWith("Copied handoff context to clipboard");
|
|
expect(showError).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("does not fall back to stale handoff context after a textless assistant response", () => {
|
|
const copySpy = vi.spyOn(native, "copyToClipboard").mockImplementation(() => undefined);
|
|
const { controller, showStatus, showError } = createController({
|
|
hasAssistantMessage: true,
|
|
handoffText: "<handoff-context>\n## Goal\nContinue\n</handoff-context>",
|
|
});
|
|
|
|
controller.handleCopyCommand();
|
|
|
|
expect(copySpy).not.toHaveBeenCalled();
|
|
expect(showStatus).not.toHaveBeenCalled();
|
|
expect(showError).toHaveBeenCalledWith("No agent messages to copy yet.");
|
|
});
|
|
});
|