70 lines
1.6 KiB
TypeScript
70 lines
1.6 KiB
TypeScript
/**
|
|
* @jest-environment node
|
|
*/
|
|
|
|
import { createApp, h, defineAsyncComponent } from 'vue'
|
|
import { ReadableStream, TransformStream } from 'stream/web'
|
|
import { pipeToWebWritable, renderToWebStream } from '../src'
|
|
|
|
beforeEach(() => {
|
|
global.ReadableStream = ReadableStream
|
|
})
|
|
|
|
afterEach(() => {
|
|
// @ts-ignore
|
|
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><!--]-->`)
|
|
})
|