@things-factory/ai-assistant 10.1.6

package/client/components/board-ai-chat.test.ts

@@ -0,0 +1,149 @@
+/**
+ * board-ai-chat.ts 의 pure helper 회귀 방지.
+ *
+ * 핵심 보호:
+ *  - GraphQL mutation input 에 selectedRefids 가 항상 포함되는지 — 누락이 사용자
+ *    보고된 "AI 가 선택된 컴포넌트를 모른다" 의 root cause 였다.
+ *  - 식별자 정책 — refid (universal numeric) 가 selection 채널의 표준. id 는 데이터
+ *    바인딩 이름이며 unique 가 아니라 selection 에 부적합.
+ */
+import { buildChatMutationInput } from './chat-input-builder'
+
+describe('buildChatMutationInput — mutation input 회귀 방지', () => {
+  test('selectedRefids 가 input 에 반드시 포함된다 (root cause regression)', () => {
+    const input = buildChatMutationInput({
+      sessionId: 'session-1',
+      history: [{ role: 'user', content: 'hi' }],
+      liveBoard: { width: 1000, height: 600, components: [] },
+      selectedRefids: [35]
+    })
+    expect(input).toHaveProperty('selectedRefids', [35])
+  })
+
+  test('selectedRefids 빈 배열도 누락 없이 포함', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: []
+    })
+    expect(input.selectedRefids).toEqual([])
+  })
+
+  test('모든 expected 필드 존재', () => {
+    const input = buildChatMutationInput({
+      sessionId: 's1',
+      history: [{ role: 'user', content: 'hi' }],
+      liveBoard: { components: [] },
+      scopes: ['fmsim'],
+      knownTypes: ['rect'],
+      categories: ['equipment'],
+      componentSchemas: [{ type: 'rect' }],
+      selectedRefids: [1, 2]
+    })
+    expect(Object.keys(input).sort()).toEqual([
+      'categories',
+      'componentSchemas',
+      'currentBoard',
+      'knownTypes',
+      'messages',
+      'scopes',
+      'selectedRefids',
+      'sessionId'
+    ])
+  })
+
+  test('sessionId 누락 → null 정규화', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: []
+    })
+    expect(input.sessionId).toBeNull()
+  })
+
+  test('liveBoard 누락 → null 정규화', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: undefined,
+      selectedRefids: []
+    })
+    expect(input.currentBoard).toBeNull()
+  })
+
+  test('messages 가 history 그대로', () => {
+    const history = [
+      { role: 'user' as const, content: 'msg1' },
+      { role: 'assistant' as const, content: 'reply1' }
+    ]
+    const input = buildChatMutationInput({
+      history,
+      liveBoard: null,
+      selectedRefids: []
+    })
+    expect(input.messages).toBe(history)
+  })
+
+  test('liveBoard 가 그대로 currentBoard 로 매핑', () => {
+    const board = {
+      width: 1920,
+      height: 1080,
+      components: [{ refid: 1, type: 'rect' }]
+    }
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: board,
+      selectedRefids: []
+    })
+    expect(input.currentBoard).toBe(board)
+  })
+
+  test('scopes / knownTypes / categories 누락 시 undefined', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: []
+    })
+    expect(input.scopes).toBeUndefined()
+    expect(input.knownTypes).toBeUndefined()
+    expect(input.categories).toBeUndefined()
+    expect(input.componentSchemas).toBeUndefined()
+  })
+
+  test('selectedRefids 다중 항목 (number 배열, stringify 없음)', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: [35, 127, 4096]
+    })
+    expect(input.selectedRefids).toEqual([35, 127, 4096])
+  })
+
+  test('mentions (popup picks) — 빈 배열은 input 에 안 들어감 (server 부담 X)', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: [],
+      mentions: []
+    })
+    expect(input.mentions).toBeUndefined()
+  })
+
+  test('mentions — 항목 있으면 input 에 그대로 통과', () => {
+    const input = buildChatMutationInput({
+      history: [{ role: 'user', content: '#경광등1 빨갛게' }],
+      liveBoard: null,
+      selectedRefids: [],
+      mentions: [{ token: '경광등1', refid: 35 }]
+    })
+    expect(input.mentions).toEqual([{ token: '경광등1', refid: 35 }])
+  })
+
+  test('mentions 미지정 시 input 에 키 자체가 없음 (backwards compat)', () => {
+    const input = buildChatMutationInput({
+      history: [],
+      liveBoard: null,
+      selectedRefids: []
+    })
+    expect('mentions' in input).toBe(false)
+  })
+})