137 lines
4.2 KiB
TypeScript
137 lines
4.2 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import type { RawMessage } from '@/stores/chat';
|
|
import {
|
|
IMAGE_GENERATION_TIMEOUT_MS,
|
|
isImageGenerationPending,
|
|
} from '@/pages/Chat/image-generation-status';
|
|
|
|
const TASK_ID = '27443fdb-6cca-48e6-a3a7-ee34b0491aee';
|
|
const NOW = 1_700_000_000_000;
|
|
|
|
describe('isImageGenerationPending', () => {
|
|
it('returns true while waiting after image_generate without a final reply', () => {
|
|
const segmentMessages: RawMessage[] = [
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000,
|
|
content: [{ type: 'toolCall', id: 'call_1', name: 'image_generate', arguments: { prompt: 'astronaut' } }],
|
|
},
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000 + 5,
|
|
content: [{ type: 'text', text: '图片生成中，稍等片刻 🧑🚀✨' }],
|
|
},
|
|
];
|
|
|
|
expect(isImageGenerationPending(segmentMessages, [], NOW)).toBe(true);
|
|
});
|
|
|
|
it('returns false after the inter-session completion event arrives', () => {
|
|
const segmentMessages: RawMessage[] = [
|
|
{
|
|
role: 'toolresult',
|
|
toolName: 'image_generate',
|
|
timestamp: NOW / 1000,
|
|
content: [{
|
|
type: 'text',
|
|
text: `Background task started for image generation (${TASK_ID}).`,
|
|
}],
|
|
},
|
|
{
|
|
role: 'user',
|
|
content: [{
|
|
type: 'text',
|
|
text: `[Inter-session message] sourceSession=image_generate:${TASK_ID} sourceTool=image_generate`,
|
|
}],
|
|
},
|
|
];
|
|
|
|
expect(isImageGenerationPending(segmentMessages, [], NOW)).toBe(false);
|
|
});
|
|
|
|
it('returns false once a non-generating assistant reply arrives', () => {
|
|
const segmentMessages: RawMessage[] = [
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000,
|
|
content: [{ type: 'toolCall', id: 'call_1', name: 'image_generate', arguments: { prompt: 'astronaut' } }],
|
|
},
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000 + 5,
|
|
content: [{ type: 'text', text: '图片生成中，稍等片刻 🧑🚀✨' }],
|
|
},
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000 + 120,
|
|
content: [{ type: 'text', text: '抱歉，图片生成超时了，稍后重试一下吧 🧑🚀' }],
|
|
},
|
|
];
|
|
|
|
expect(isImageGenerationPending(segmentMessages, [], NOW + 120_000)).toBe(false);
|
|
});
|
|
|
|
it('returns false after the configured timeout even if only generating narration exists', () => {
|
|
const segmentMessages: RawMessage[] = [
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000,
|
|
content: [{ type: 'toolCall', id: 'call_1', name: 'image_generate', arguments: { prompt: 'astronaut' } }],
|
|
},
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000 + 5,
|
|
content: [{ type: 'text', text: '图片生成中，稍等片刻 🧑🚀✨' }],
|
|
},
|
|
];
|
|
|
|
expect(
|
|
isImageGenerationPending(
|
|
segmentMessages,
|
|
[],
|
|
NOW + IMAGE_GENERATION_TIMEOUT_MS + 20_000,
|
|
),
|
|
).toBe(false);
|
|
});
|
|
|
|
it('returns false when a stale streaming tool status is still marked running', () => {
|
|
expect(isImageGenerationPending([], [{
|
|
name: 'image_generate',
|
|
status: 'running',
|
|
updatedAt: NOW - IMAGE_GENERATION_TIMEOUT_MS - 60_000,
|
|
}], NOW)).toBe(false);
|
|
});
|
|
|
|
it('returns true while image_generate is actively running in streaming tools', () => {
|
|
expect(isImageGenerationPending([], [{
|
|
name: 'image_generate',
|
|
status: 'running',
|
|
updatedAt: NOW,
|
|
}], NOW)).toBe(true);
|
|
});
|
|
|
|
it('returns false once an image attachment is delivered', () => {
|
|
const segmentMessages: RawMessage[] = [
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000,
|
|
content: [{ type: 'toolCall', id: 'call_1', name: 'image_generate', arguments: { prompt: 'astronaut' } }],
|
|
},
|
|
{
|
|
role: 'assistant',
|
|
timestamp: NOW / 1000 + 90,
|
|
content: [{ type: 'text', text: '宇航员图片完成啦 🧑🚀✨' }],
|
|
_attachedFiles: [{
|
|
fileName: 'astronaut.png',
|
|
mimeType: 'image/png',
|
|
fileSize: 123,
|
|
preview: null,
|
|
filePath: '/tmp/astronaut.png',
|
|
}],
|
|
},
|
|
];
|
|
|
|
expect(isImageGenerationPending(segmentMessages, [], NOW + 90_000)).toBe(false);
|
|
});
|
|
});
|