Spaces:
Paused
Paused
| import { describe, expect, it } from 'vitest'; | |
| import { prepareArtifactDeployment } from './artifact-runtime'; | |
| import { | |
| buildStreamingArtifactDraft, | |
| extractStreamingJsonStrings, | |
| } from './streaming-artifact'; | |
| describe('extractStreamingJsonStrings', () => { | |
| it('decodes complete properties and the active unfinished string', () => { | |
| expect(extractStreamingJsonStrings( | |
| '{"title":"Demo","file_1_path":"index.html","file_1_content":"<h1>Hello\\nwor', | |
| )).toEqual({ | |
| values: { | |
| title: 'Demo', | |
| file_1_path: 'index.html', | |
| }, | |
| partial: { | |
| key: 'file_1_content', | |
| value: '<h1>Hello\nwor', | |
| }, | |
| }); | |
| }); | |
| it('holds an unfinished JSON escape until it becomes decodable', () => { | |
| expect(extractStreamingJsonStrings('{"html":"hello\\u26')).toEqual({ | |
| values: {}, | |
| partial: { key: 'html', value: 'hello' }, | |
| }); | |
| }); | |
| }); | |
| describe('buildStreamingArtifactDraft', () => { | |
| it('renders partial HTML before the content property closes', () => { | |
| const draft = buildStreamingArtifactDraft( | |
| 'artifact_deploy', | |
| '{"title":"Live","file_1_path":"index.html","file_1_content":"<!doctype html><h1>Hel', | |
| 'draft-live', | |
| ); | |
| expect(draft?.files).toMatchObject([{ | |
| path: 'index.html', | |
| content: '<!doctype html><h1>Hel', | |
| partial: true, | |
| }]); | |
| expect(draft?.preview?.source).toContain('<h1>Hel'); | |
| }); | |
| it('shows partial JavaScript in the tree without executing it', () => { | |
| const draft = buildStreamingArtifactDraft( | |
| 'artifact_deploy', | |
| '{"file_1_path":"index.html","file_1_content":"<main>Ready</main>","file_2_path":"app.js","file_2_content":"document.querySele', | |
| 'draft-script', | |
| ); | |
| expect(draft?.files.find((file) => file.path === 'app.js')).toMatchObject({ | |
| content: 'document.querySele', | |
| partial: true, | |
| }); | |
| expect(draft?.preview?.files?.some((file) => file.path === 'app.js')).toBe(false); | |
| }); | |
| it('layers a partial write over the existing artifact without mutating it', () => { | |
| const original = prepareArtifactDeployment([ | |
| { path: 'index.html', content: '<link rel="stylesheet" href="./style.css"><main>Old</main>' }, | |
| { path: 'style.css', content: 'main { color: red; }' }, | |
| ], 'Original', 'index.html', 'artifact-1'); | |
| const draft = buildStreamingArtifactDraft( | |
| 'artifact_write', | |
| '{"artifact_id":"artifact-1","file_1_path":"style.css","file_1_content":"main { color: blue', | |
| 'draft-write', | |
| new Map([[original.id, original]]), | |
| ); | |
| expect(draft?.files.find((file) => file.path === 'style.css')).toMatchObject({ | |
| content: 'main { color: blue', | |
| partial: true, | |
| }); | |
| expect(original.files?.find((file) => file.path === 'style.css')?.content).toBe('main { color: red; }'); | |
| expect(draft?.preview?.files?.find((file) => file.path === 'style.css')?.content).toBe('main { color: blue'); | |
| }); | |
| }); | |