mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
28907d8c59
### Description Sometimes the `npm test` failed with an error of "TypeError: Failed to fetch". I checked the callback entry of the localhost server started by karma. When the "Failed to fetch" happens, no request is reflected on the server side. The root cause is still not identified. However, as this issue only happens sometimes when the browser is just launched by karma runner, doing retry can workaround this issue for most of the time.
66 lines
1.8 KiB
TypeScript
66 lines
1.8 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import * as base64 from 'base64-js';
|
|
import * as fs from 'node:fs/promises';
|
|
|
|
import {Attribute} from '../lib/onnxjs/attribute';
|
|
import {Graph} from '../lib/onnxjs/graph';
|
|
|
|
export function base64toBuffer(data: string): Uint8Array {
|
|
return base64.toByteArray(data);
|
|
}
|
|
|
|
export function bufferToBase64(buffer: Uint8Array): string {
|
|
return base64.fromByteArray(buffer);
|
|
}
|
|
|
|
async function retry<T>(fn: () => Promise<T>, maxRetries = 3, delay = 100): Promise<T> {
|
|
let retries = maxRetries;
|
|
do {
|
|
try {
|
|
return await fn();
|
|
} catch (err) {
|
|
if (retries-- === 0) {
|
|
throw err;
|
|
}
|
|
await new Promise(resolve => setTimeout(resolve, delay));
|
|
}
|
|
// eslint-disable-next-line no-constant-condition
|
|
} while (true);
|
|
}
|
|
|
|
export async function readFile(file: string) {
|
|
if (typeof process !== 'undefined' && process.versions && process.versions.node) {
|
|
// node
|
|
return fs.readFile(file);
|
|
} else {
|
|
// browser
|
|
//
|
|
// use "retry" to workaround the error "TypeError: Failed to fetch" in some test environments
|
|
return retry(async () => {
|
|
const response = await fetch(file);
|
|
return new Uint8Array(await response.arrayBuffer());
|
|
});
|
|
}
|
|
}
|
|
|
|
export async function readJsonFile(file: string): Promise<any> {
|
|
const content = await readFile(file);
|
|
return JSON.parse(new TextDecoder().decode(content));
|
|
}
|
|
|
|
/**
|
|
* create a single-node graph for unit test purpose
|
|
*/
|
|
export function createMockGraph(opType: string, attributes: Attribute): Graph {
|
|
const node: Graph.Node = {name: '', opType, inputs: [], outputs: [], attributes};
|
|
return {
|
|
getInputIndices: () => [],
|
|
getInputNames: () => [],
|
|
getOutputIndices: () => [],
|
|
getOutputNames: () => [],
|
|
getNodes: () => [node],
|
|
getValues: () => []
|
|
};
|
|
}
|