Files
oh-my-pi/packages/coding-agent/test/prompt-action-autocomplete.test.ts
T
2026-03-09 15:38:39 +01:00

97 lines
3.3 KiB
TypeScript

import { afterEach, beforeEach, describe, expect, it } from "bun:test";
import { EditorKeybindingsManager, setEditorKeybindings } from "../../tui/src/keybindings";
import { KeybindingsManager } from "../src/config/keybindings";
import { createPromptActionAutocompleteProvider } from "../src/modes/prompt-action-autocomplete";
describe("prompt action autocomplete", () => {
beforeEach(() => {
setEditorKeybindings(
new EditorKeybindingsManager({
cursorLineStart: ["home", "f6"],
cursorLineEnd: "f7",
}),
);
});
afterEach(() => {
setEditorKeybindings(new EditorKeybindingsManager());
});
it("shows prompt actions with configured shortcut hints", async () => {
const provider = createPromptActionAutocompleteProvider({
commands: [],
basePath: "/tmp",
keybindings: KeybindingsManager.inMemory({
copyLine: "ctrl+shift+l",
copyPrompt: ["alt+shift+c", "ctrl+shift+c"],
}),
copyCurrentLine: () => {},
copyPrompt: () => {},
moveCursorToLineStart: () => {},
moveCursorToLineEnd: () => {},
});
const suggestions = await provider.getSuggestions(["#"], 0, 1);
expect(suggestions).not.toBeNull();
expect(suggestions?.prefix).toBe("#");
expect(suggestions?.items.map(item => item.label)).toEqual([
"Copy current line",
"Copy whole prompt",
"Move cursor to beginning of line",
"Move cursor to end of line",
]);
expect(suggestions?.items.find(item => item.label === "Copy current line")?.description).toBe("Ctrl+Shift+L");
expect(suggestions?.items.find(item => item.label === "Copy whole prompt")?.description).toBe(
"Alt+Shift+C/Ctrl+Shift+C",
);
expect(suggestions?.items.find(item => item.label === "Move cursor to beginning of line")?.description).toBe(
"Home/F6",
);
expect(suggestions?.items.find(item => item.label === "Move cursor to end of line")?.description).toBe("F7");
});
it("executes selected prompt actions and removes the trigger text", async () => {
let promptCopies = 0;
const provider = createPromptActionAutocompleteProvider({
commands: [],
basePath: "/tmp",
keybindings: KeybindingsManager.inMemory(),
copyCurrentLine: () => {},
copyPrompt: () => {
promptCopies += 1;
},
moveCursorToLineStart: () => {},
moveCursorToLineEnd: () => {},
});
const suggestions = await provider.getSuggestions(["hello #cop"], 0, 10);
const item = suggestions?.items.find(entry => entry.label === "Copy whole prompt");
expect(item).toBeDefined();
if (!item || !suggestions) {
throw new Error("expected copy whole prompt suggestion");
}
const result = provider.applyCompletion(["hello #cop"], 0, 10, item, suggestions.prefix);
expect(result.lines).toEqual(["hello "]);
expect(result.cursorLine).toBe(0);
expect(result.cursorCol).toBe(6);
result.onApplied?.();
expect(promptCopies).toBe(1);
});
it("falls back to normal typing for literal hashtags with no matching action", async () => {
const provider = createPromptActionAutocompleteProvider({
commands: [],
basePath: "/tmp",
keybindings: KeybindingsManager.inMemory(),
copyCurrentLine: () => {},
copyPrompt: () => {},
moveCursorToLineStart: () => {},
moveCursorToLineEnd: () => {},
});
const suggestions = await provider.getSuggestions(["release #v1"], 0, 11);
expect(suggestions).toBeNull();
});
});