46 lines
1.9 KiB
TypeScript
46 lines
1.9 KiB
TypeScript
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);
|
|
});
|
|
});
|