/** * Copyright (c) Microsoft Corporation. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ import util from 'util'; import path from 'path'; import type { TestError, Location } from './types'; import { default as minimatch } from 'minimatch'; export class DeadlineRunner { private _timer: NodeJS.Timer | undefined; private _done = false; private _fulfill!: (t: { result?: T, timedOut?: boolean }) => void; private _reject!: (error: any) => void; readonly result: Promise<{ result?: T, timedOut?: boolean }>; constructor(promise: Promise, deadline: number | undefined) { this.result = new Promise((f, r) => { this._fulfill = f; this._reject = r; }); promise.then(result => { this._finish({ result }); }).catch(e => { this._finish(undefined, e); }); this.setDeadline(deadline); } private _finish(success?: { result?: T, timedOut?: boolean }, error?: any) { if (this._done) return; this.setDeadline(undefined); if (success) this._fulfill(success); else this._reject(error); } setDeadline(deadline: number | undefined) { if (this._timer) { clearTimeout(this._timer); this._timer = undefined; } if (deadline === undefined) return; const timeout = deadline - monotonicTime(); if (timeout <= 0) this._finish({ timedOut: true }); else this._timer = setTimeout(() => this._finish({ timedOut: true }), timeout); } } export async function raceAgainstDeadline(promise: Promise, deadline: number | undefined): Promise<{ result?: T, timedOut?: boolean }> { return (new DeadlineRunner(promise, deadline)).result; } export function serializeError(error: Error | any): TestError { if (error instanceof Error) { return { message: error.message, stack: error.stack }; } return { value: util.inspect(error) }; } export function monotonicTime(): number { const [seconds, nanoseconds] = process.hrtime(); return seconds * 1000 + (nanoseconds / 1000000 | 0); } export function isRegExp(e: any): e is RegExp { return e && typeof e === 'object' && (e instanceof RegExp || Object.prototype.toString.call(e) === '[object RegExp]'); } export type Matcher = (value: string) => boolean; export type FilePatternFilter = { re: RegExp; line: number | null; }; export function createMatcher(patterns: string | RegExp | (string | RegExp)[]): Matcher { const reList: RegExp[] = []; const filePatterns: string[] = []; for (const pattern of Array.isArray(patterns) ? patterns : [patterns]) { if (isRegExp(pattern)) { reList.push(pattern); } else { if (!pattern.startsWith('**/') && !pattern.startsWith('**/')) filePatterns.push('**/' + pattern); else filePatterns.push(pattern); } } return (value: string) => { for (const re of reList) { re.lastIndex = 0; if (re.test(value)) return true; } for (const pattern of filePatterns) { if (minimatch(value, pattern, { nocase: true, })) return true; } return false; }; } export function mergeObjects(a: A | undefined | void, b: B | undefined | void): A & B { const result = { ...a } as any; if (!Object.is(b, undefined)) { for (const [name, value] of Object.entries(b as B)) { if (!Object.is(value, undefined)) result[name] = value; } } return result as any; } export async function wrapInPromise(value: any) { return value; } export function forceRegExp(pattern: string): RegExp { const match = pattern.match(/^\/(.*)\/([gi]*)$/); if (match) return new RegExp(match[1], match[2]); return new RegExp(pattern, 'g'); } export function relativeFilePath(file: string): string { if (!path.isAbsolute(file)) return file; return path.relative(process.cwd(), file); } export function formatLocation(location: Location) { return relativeFilePath(location.file) + ':' + location.line + ':' + location.column; } export function errorWithFile(file: string, message: string) { return new Error(`${relativeFilePath(file)}: ${message}`); } export function errorWithLocation(location: Location, message: string) { return new Error(`${formatLocation(location)}: ${message}`); }