feat: finally a good working state

tests/renderer/a2ui/inlineSurfaceAssociation.test.ts

@@ -0,0 +1,45 @@
+import { describe, expect, it } from 'vitest';
+import { computeTurnIndex } from '../../../src/renderer/a2ui/surfaceAssociation';
+import type { ChatMessage } from '../../../src/main/shared/electronApi';
+
+function msg(role: ChatMessage['role'], id = `msg-${Date.now()}-${Math.random()}`): ChatMessage {
+  return { id, conversationId: 'conv-1', role, content: '', createdAt: new Date().toISOString() };
+}
+
+describe('computeTurnIndex', () => {
+  it('returns 0 for the first user message', () => {
+    const messages = [msg('user')];
+    expect(computeTurnIndex(messages, 0)).toBe(0);
+  });
+
+  it('returns 0 for an assistant message following the first user message', () => {
+    const messages = [msg('user'), msg('assistant')];
+    expect(computeTurnIndex(messages, 1)).toBe(0);
+  });
+
+  it('increments turn index for each user message', () => {
+    const messages = [msg('user'), msg('assistant'), msg('user'), msg('assistant')];
+    expect(computeTurnIndex(messages, 0)).toBe(0);
+    expect(computeTurnIndex(messages, 1)).toBe(0);
+    expect(computeTurnIndex(messages, 2)).toBe(1);
+    expect(computeTurnIndex(messages, 3)).toBe(1);
+  });
+
+  it('skips system and tool messages in turn counting', () => {
+    const messages = [msg('system'), msg('user'), msg('tool'), msg('assistant')];
+    expect(computeTurnIndex(messages, 1)).toBe(0); // user
+    expect(computeTurnIndex(messages, 3)).toBe(0); // assistant after first user
+  });
+
+  it('returns -1 when no user messages precede the index', () => {
+    const messages = [msg('system'), msg('assistant')];
+    expect(computeTurnIndex(messages, 0)).toBe(-1);
+    expect(computeTurnIndex(messages, 1)).toBe(-1);
+  });
+
+  it('handles multiple assistant messages in the same turn', () => {
+    const messages = [msg('user'), msg('assistant'), msg('assistant')];
+    expect(computeTurnIndex(messages, 1)).toBe(0);
+    expect(computeTurnIndex(messages, 2)).toBe(0);
+  });
+});