61 lines
2.0 KiB
TypeScript
61 lines
2.0 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import { runUserPromptSubmit, userPromptSubmitHandler } from '../../src/hooks/user-prompt-submit.js';
|
|
import { makeHookCaptures, makeMockBridge } from './_test-helpers.js';
|
|
|
|
describe('user-prompt-submit handler', () => {
|
|
it('extracts prompt from payload.prompt or payload.user_message', () => {
|
|
expect(userPromptSubmitHandler.parse({ prompt: 'hi' }).prompt).toBe('hi');
|
|
expect(userPromptSubmitHandler.parse({ user_message: 'hello' }).prompt).toBe('hello');
|
|
expect(userPromptSubmitHandler.parse({}).prompt).toBe('');
|
|
});
|
|
|
|
it('saves a temporary frame containing the prompt', async () => {
|
|
const bridge = makeMockBridge();
|
|
const cap = makeHookCaptures();
|
|
await runUserPromptSubmit({
|
|
readStdin: async () => JSON.stringify({
|
|
prompt: 'How do I X?',
|
|
cwd: '/proj/foo',
|
|
session_id: 'sess-7',
|
|
}),
|
|
writeStdout: cap.writeStdout,
|
|
exit: cap.exit,
|
|
bridge,
|
|
});
|
|
expect(bridge.saveMemory).toHaveBeenCalledTimes(1);
|
|
const arg = bridge.saveMemory.mock.calls[0][0];
|
|
expect(arg).toMatchObject({
|
|
content: 'How do I X?',
|
|
importance: 'temporary',
|
|
scope: 'sess-7',
|
|
source: 'claude-code',
|
|
});
|
|
expect(cap.exits).toEqual([0]);
|
|
});
|
|
|
|
it('skips save when no prompt is present', async () => {
|
|
const bridge = makeMockBridge();
|
|
const cap = makeHookCaptures();
|
|
await runUserPromptSubmit({
|
|
readStdin: async () => '{}',
|
|
writeStdout: cap.writeStdout,
|
|
exit: cap.exit,
|
|
bridge,
|
|
});
|
|
expect(bridge.saveMemory).not.toHaveBeenCalled();
|
|
expect(cap.exits).toEqual([0]);
|
|
});
|
|
|
|
it('exits 0 even if saveMemory rejects', async () => {
|
|
const bridge = makeMockBridge({ saveMemoryThrows: new Error('cli down') });
|
|
const cap = makeHookCaptures();
|
|
await runUserPromptSubmit({
|
|
readStdin: async () => JSON.stringify({ prompt: 'x' }),
|
|
writeStdout: cap.writeStdout,
|
|
exit: cap.exit,
|
|
bridge,
|
|
});
|
|
expect(cap.exits).toEqual([0]);
|
|
});
|
|
});
|