60 lines
2.3 KiB
JavaScript
60 lines
2.3 KiB
JavaScript
const assert = require('node:assert/strict');
|
|
const test = require('node:test');
|
|
|
|
test('model response progress forwards each downloaded chunk and preserves the response', async () => {
|
|
const {trackDownloadProgress} = await import('../piper-download-progress.mjs');
|
|
const source = new ReadableStream({
|
|
start(controller) {
|
|
controller.enqueue(new Uint8Array([1, 2]));
|
|
controller.enqueue(new Uint8Array([3, 4, 5]));
|
|
controller.close();
|
|
},
|
|
});
|
|
const response = new Response(source, {headers: {'content-length': '5'}});
|
|
const messages = [];
|
|
const tracked = trackDownloadProgress(response, {id: 7, postMessage: message => messages.push(message)});
|
|
|
|
assert.deepEqual([...new Uint8Array(await tracked.arrayBuffer())], [1, 2, 3, 4, 5]);
|
|
assert.deepEqual(messages.map(message => message.progress), [
|
|
{asset: 'model', loaded: 2, total: 5},
|
|
{asset: 'model', loaded: 5, total: 5},
|
|
]);
|
|
assert.ok(messages.every(message => message.id === 7 && message.type === 'progress'));
|
|
});
|
|
|
|
test('model response progress works when content length is absent', async () => {
|
|
const {trackDownloadProgress} = await import('../piper-download-progress.mjs');
|
|
const source = new Response(new Uint8Array([1, 2, 3]));
|
|
const messages = [];
|
|
const tracked = trackDownloadProgress(source, {
|
|
id: 9,
|
|
asset: 'runtime',
|
|
file: 'runtime.wasm',
|
|
postMessage: message => messages.push(message),
|
|
});
|
|
|
|
assert.equal((await tracked.arrayBuffer()).byteLength, 3);
|
|
assert.deepEqual(messages
|
|
.map(message => message.progress)
|
|
.filter(progress => Number.isFinite(progress.loaded)),
|
|
[{asset: 'runtime', file: 'runtime.wasm', loaded: 3, total: 0}]);
|
|
});
|
|
|
|
test('runtime asset completion reports that initialization follows the download', async () => {
|
|
const {trackDownloadProgress} = await import('../piper-download-progress.mjs');
|
|
const response = new Response(new Uint8Array([1, 2]), {headers: {'content-length': '2'}});
|
|
const messages = [];
|
|
const tracked = trackDownloadProgress(response, {
|
|
id: 10,
|
|
asset: 'runtime',
|
|
file: 'ort-wasm-simd-threaded.wasm',
|
|
postMessage: message => messages.push(message),
|
|
});
|
|
|
|
await tracked.arrayBuffer();
|
|
assert.deepEqual(messages.at(-1).progress, {
|
|
stage: 'initializing-runtime',
|
|
file: 'ort-wasm-simd-threaded.wasm',
|
|
});
|
|
});
|