Files
connectai/tests/devilAgent.test.ts
T
2026-05-18 08:15:01 +09:00

62 lines
3.1 KiB
TypeScript

import { buildDevilSystemPrompt, buildDevilUserPrompt } from '../src/features/devilAgent/devilPrompt';
describe('Devil Agent prompt builder', () => {
test('system prompt enforces the 6 core rules', () => {
const sys = buildDevilSystemPrompt();
// 핵심 규칙들이 모두 들어있는지 검증 — 정확한 문자열은 prompt 본문 변경에 깨지지만,
// 가치 보존되는 핵심 키워드 기준으로 체크.
expect(sys).toContain('도현');
expect(sys).toContain('한 가지 약점'); // rule 1
expect(sys).toContain('만약'); // rule 2 framing
expect(sys).toContain('칭찬'); // rule 3 — 금지
expect(sys).toContain('통계'); // rule 4 — 환각 가드
expect(sys).toContain('우려'); // rule 5 출력 형식
expect(sys).toContain('검증'); // rule 5 출력 형식
expect(sys).toContain('근거:'); // rule 6 출처 태그
});
test('user prompt includes both user prompt and assistant answer', () => {
const u = buildDevilUserPrompt({
userPrompt: '회원가입 흐름 설계 도와줘',
assistantAnswer: '이메일+비번 기본 flow 를 권장합니다.',
});
expect(u).toContain('회원가입 흐름 설계 도와줘');
expect(u).toContain('이메일+비번 기본 flow 를 권장합니다');
expect(u).toContain('Astra 의 직전 답변');
expect(u).toContain('가장 본질적인 약점 하나');
});
test('user prompt with brainContext adds reference section', () => {
const u = buildDevilUserPrompt({
userPrompt: 'Q',
assistantAnswer: 'A',
brainContext: '## brain/sample.md\n샘플 자료입니다.',
});
expect(u).toContain('참고 가능한 Second Brain 자료');
expect(u).toContain('샘플 자료입니다');
});
test('user prompt with userRebuttal switches mode to deeper round', () => {
const u = buildDevilUserPrompt({
userPrompt: 'Q',
assistantAnswer: 'A',
userRebuttal: '그건 사용자 입장에선 오히려 편리한데?',
});
expect(u).toContain('사용자의 재반박');
expect(u).toContain('그건 사용자 입장에선 오히려 편리한데');
expect(u).toContain('한 단계 더 깊은 약점');
// 첫 라운드 표시는 보이지 말아야 — userRebuttal 있으면 다른 분기.
expect(u).not.toContain('가장 본질적인 약점 하나를 골라');
});
test('user prompt handles empty fields gracefully', () => {
const u = buildDevilUserPrompt({ userPrompt: '', assistantAnswer: '' });
expect(u).toContain('(빈 질문)');
expect(u).toContain('(빈 답변)');
});
});
// generateDevilRebuttal 은 vscode 의존 — settings 토글 / module dynamic import 흐름 때문에
// 본 unit suite 에서는 직접 호출 X. 대신 prompt 빌더에 집중.
// settings 통합 흐름은 통합 테스트(향후)로 분리.