vue3-core/packages/server-renderer/__tests__/webStream.spec.ts

67 lines
1.6 KiB
TypeScript

import { createApp, defineAsyncComponent, h } from 'vue'
import { ReadableStream, TransformStream } from 'node:stream/web'
import { pipeToWebWritable, renderToWebStream } from '../src'
beforeEach(() => {
// @ts-expect-error
global.ReadableStream = ReadableStream
})
afterEach(() => {
// @ts-expect-error
delete global.ReadableStream
})
test('renderToWebStream', async () => {
const Async = defineAsyncComponent(() =>
Promise.resolve({
render: () => h('div', 'async'),
}),
)
const App = {
render: () => [h('div', 'parent'), h(Async)],
}
const stream = renderToWebStream(createApp(App))
const reader = stream.getReader()
const decoder = new TextDecoder()
let res = ''
await reader.read().then(function read({ done, value }): any {
if (!done) {
res += decoder.decode(value)
return reader.read().then(read)
}
})
expect(res).toBe(`<!--[--><div>parent</div><div>async</div><!--]-->`)
})
test('pipeToWebWritable', async () => {
const Async = defineAsyncComponent(() =>
Promise.resolve({
render: () => h('div', 'async'),
}),
)
const App = {
render: () => [h('div', 'parent'), h(Async)],
}
const { readable, writable } = new TransformStream()
pipeToWebWritable(createApp(App), {}, writable)
const reader = readable.getReader()
const decoder = new TextDecoder()
let res = ''
await reader.read().then(function read({ done, value }): any {
if (!done) {
res += decoder.decode(value)
return reader.read().then(read)
}
})
expect(res).toBe(`<!--[--><div>parent</div><div>async</div><!--]-->`)
})