Icard/angular-clarity-master(work.../node_modules/eventemitter-asyncresource/test/test.ts

120 lines
3.2 KiB
TypeScript
Raw Normal View History

2024-07-16 14:55:36 +00:00
import EventEmitterAsyncResource from '..';
import { createHook, executionAsyncId, executionAsyncResource } from 'async_hooks';
import { test } from 'tap';
import { promisify } from 'util';
const tick = promisify(setImmediate);
interface InitAsyncEvent {
name : 'init',
type : string,
triggerAsyncId : number,
resource: object
}
interface OtherAsyncEvent {
name : 'before' | 'after' | 'destroy'
}
type AsyncEvent = InitAsyncEvent | OtherAsyncEvent;
function makeHook (trackedTypes : string[]) {
const eventMap = new Map<number, AsyncEvent[]>();
function log (asyncId : number, name : 'before' | 'after' | 'destroy') {
const entry = eventMap.get(asyncId);
if (entry !== undefined) entry.push({ name });
}
const hook = createHook({
init (asyncId, type, triggerAsyncId, resource) {
if (trackedTypes.includes(type)) {
eventMap.set(asyncId, [
{ name: 'init', type, triggerAsyncId, resource }
]);
}
},
before (asyncId) { log(asyncId, 'before'); },
after (asyncId) { log(asyncId, 'after'); },
destroy (asyncId) { log(asyncId, 'destroy'); }
}).enable();
return {
done () {
hook.disable();
return new Set(eventMap.values());
},
ids () {
return new Set(eventMap.keys());
}
};
}
test('tracks emit() calls correctly using async_hooks', async ({ is, isDeep }) => {
const tracer = makeHook(['Foo']);
class Foo extends EventEmitterAsyncResource {}
const origExecutionAsyncId = executionAsyncId();
const foo = new Foo();
let called = false;
foo.on('someEvent', () => {
if (typeof executionAsyncResource === 'function') { // Node.js 12+ only
is(executionAsyncResource(), foo.asyncResource);
}
called = true;
});
foo.emit('someEvent');
is(called, true);
isDeep([foo.asyncId()], [...tracer.ids()]);
is(foo.triggerAsyncId(), origExecutionAsyncId);
is(foo.asyncResource.eventEmitter, foo);
foo.emitDestroy();
await tick();
isDeep(tracer.done(), new Set<AsyncEvent[]>([
[
{ name: 'init', type: 'Foo', triggerAsyncId: origExecutionAsyncId, resource: foo.asyncResource },
{ name: 'before' },
{ name: 'after' },
{ name: 'destroy' }
]
]));
});
test('can explicitly specify name as positional arg', async ({ isDeep }) => {
const tracer = makeHook(['ResourceName']);
const origExecutionAsyncId = executionAsyncId();
class Foo extends EventEmitterAsyncResource {}
const foo = new Foo('ResourceName');
isDeep(tracer.done(), new Set<AsyncEvent[]>([
[
{ name: 'init', type: 'ResourceName', triggerAsyncId: origExecutionAsyncId, resource: foo.asyncResource }
]
]));
});
test('can explicitly specify name as option', async ({ isDeep }) => {
const tracer = makeHook(['ResourceName']);
const origExecutionAsyncId = executionAsyncId();
class Foo extends EventEmitterAsyncResource {}
const foo = new Foo({ name: 'ResourceName' });
isDeep(tracer.done(), new Set<AsyncEvent[]>([
[
{ name: 'init', type: 'ResourceName', triggerAsyncId: origExecutionAsyncId, resource: foo.asyncResource }
]
]));
});
test('is provided as export on itself', async ({ is }) => {
is(EventEmitterAsyncResource.EventEmitterAsyncResource, EventEmitterAsyncResource);
});