electron/spec/lib/events-helpers.ts
Milan Burda d78f37ec8f
refactor: use node scheme imports in spec (#38487)
Co-authored-by: Milan Burda <miburda@microsoft.com>
2023-06-15 10:42:27 -04:00

23 lines
734 B
TypeScript

/**
* @fileoverview A set of helper functions to make it easier to work
* with events in async/await manner.
*/
import { on } from 'node:events';
export const emittedNTimes = async (emitter: NodeJS.EventEmitter, eventName: string, times: number, trigger?: () => void) => {
const events: any[][] = [];
const iter = on(emitter, eventName);
if (trigger) await Promise.resolve(trigger());
for await (const args of iter) {
events.push(args);
if (events.length === times) { break; }
}
return events;
};
export const emittedUntil = async (emitter: NodeJS.EventEmitter, eventName: string, untilFn: Function) => {
for await (const args of on(emitter, eventName)) {
if (untilFn(...args)) { return args; }
}
};