chatContextInjection.test.js
2.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/**
* 回归测试：graph-console 底部聊天条的上下文注入不变式——
* context 变了才拼上下文行，没变就原样发，不能每条消息都刷一遍。
*/
import { describe, it, expect } from 'vitest';
import { contextKey, buildOutgoingMessage } from './chatContextInjection.js';
const ctxA = { taskId: 'T-1', taskTitle: '任务A', taskStatus: 'running', goalId: 'G-1', goalTitle: '目标A' };
const ctxB = { taskId: 'T-2', taskTitle: '任务B', taskStatus: 'pending', goalId: 'G-2', goalTitle: '目标B' };
describe('chatContextInjection · buildOutgoingMessage', () => {
it('首次发送（lastSentKey 为空）带上下文时要拼上下文行', () => {
const { payload, nextKey } = buildOutgoingMessage('你好', ctxA, '');
expect(payload).toContain('T-1');
expect(payload).toContain('任务A');
expect(payload.endsWith('你好')).toBe(true);
expect(nextKey).toBe(contextKey(ctxA));
});
it('context 跟上次相同时不重复拼上下文行，原样发送用户文本', () => {
const first = buildOutgoingMessage('第一条', ctxA, '');
const second = buildOutgoingMessage('第二条', ctxA, first.nextKey);
expect(second.payload).toBe('第二条');
expect(second.nextKey).toBe(first.nextKey);
});
it('context 换了（task 或 goal 变化）要重新拼一次', () => {
const first = buildOutgoingMessage('第一条', ctxA, '');
const second = buildOutgoingMessage('切到另一个任务了', ctxB, first.nextKey);
expect(second.payload).toContain('T-2');
expect(second.payload).toContain('任务B');
expect(second.nextKey).toBe(contextKey(ctxB));
expect(second.nextKey).not.toBe(first.nextKey);
});
it('没有 context（没选中任务）时原样发送，不拼任何说明', () => {
const { payload, nextKey } = buildOutgoingMessage('随便聊聊', null, 'stale-key');
expect(payload).toBe('随便聊聊');
expect(nextKey).toBe('stale-key'); // 透传，不产生新 key
});
});
describe('chatContextInjection · contextKey', () => {
it('null context 的 key 是空字符串', () => {
expect(contextKey(null)).toBe('');
});
it('taskId/goalId 相同则 key 相同，跟 title/status 无关', () => {
const same = { ...ctxA, taskTitle: '改了标题', taskStatus: 'completed' };
expect(contextKey(same)).toBe(contextKey(ctxA));
});
});