59 lines
2.4 KiB
TypeScript
59 lines
2.4 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
|
|
import { extractOutputText } from '../../../src/renderer/components/chat/items/linkedTool/renderHelpers';
|
|
|
|
describe('renderHelpers', () => {
|
|
describe('extractOutputText', () => {
|
|
it('should return plain string content as-is', () => {
|
|
expect(extractOutputText('hello world')).toBe('hello world');
|
|
});
|
|
|
|
it('should pretty-print string content that is valid JSON', () => {
|
|
const json = '{"name":"test","value":42}';
|
|
expect(extractOutputText(json)).toBe('{\n "name": "test",\n "value": 42\n}');
|
|
});
|
|
|
|
it('should extract text from content block arrays', () => {
|
|
const content = [{ type: 'text', text: 'hello world' }];
|
|
expect(extractOutputText(content)).toBe('hello world');
|
|
});
|
|
|
|
it('should extract and pretty-print JSON from content block arrays', () => {
|
|
const inner = { teams: [{ id: '1', name: 'Test' }] };
|
|
const content = [{ type: 'text', text: JSON.stringify(inner) }];
|
|
expect(extractOutputText(content)).toBe(JSON.stringify(inner, null, 2));
|
|
});
|
|
|
|
it('should handle serialized content block arrays (string wrapping content blocks)', () => {
|
|
// This is what SemanticStepExtractor produces when content is an array
|
|
const inner = { teams: [{ id: '1', name: 'Test' }] };
|
|
const contentBlocks = [{ type: 'text', text: JSON.stringify(inner) }];
|
|
const serialized = JSON.stringify(contentBlocks);
|
|
|
|
const result = extractOutputText(serialized);
|
|
expect(result).toBe(JSON.stringify(inner, null, 2));
|
|
});
|
|
|
|
it('should handle serialized content blocks with plain text', () => {
|
|
const contentBlocks = [{ type: 'text', text: 'Some plain text\nwith newlines' }];
|
|
const serialized = JSON.stringify(contentBlocks);
|
|
|
|
const result = extractOutputText(serialized);
|
|
expect(result).toBe('Some plain text\nwith newlines');
|
|
});
|
|
|
|
it('should join multiple content blocks with newlines', () => {
|
|
const content = [
|
|
{ type: 'text', text: 'first' },
|
|
{ type: 'text', text: 'second' },
|
|
];
|
|
expect(extractOutputText(content)).toBe('first\nsecond');
|
|
});
|
|
|
|
it('should stringify non-text content blocks', () => {
|
|
const content = [{ type: 'image', url: 'http://example.com/img.png' }];
|
|
const result = extractOutputText(content);
|
|
expect(result).toContain('"type": "image"');
|
|
});
|
|
});
|
|
});
|