mirror of
https://github.com/zebrajr/node.git
synced 2025-12-06 12:20:27 +01:00
13c5a1629c
This avoids the need to wrap every promise in an AsyncWrap and also makes it easier to skip the machinery to track destroy events when there's no destroy listener. Co-authored-by: Andrey Pechkurov <apechkurov@gmail.com> PR-URL: https://github.com/nodejs/node/pull/32891 Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Andrey Pechkurov <apechkurov@gmail.com> Reviewed-By: Anna Henningsen <anna@addaleax.net> Reviewed-By: Gerhard Stöbich <deb2001-github@yahoo.de> Reviewed-By: Chengzhong Wu <legendecas@gmail.com>
105 lines
3.0 KiB
JavaScript
105 lines
3.0 KiB
JavaScript
'use strict';
|
|
// Flags: --expose-internals
|
|
|
|
const common = require('../../common');
|
|
const assert = require('assert');
|
|
const async_hooks = require('async_hooks');
|
|
const { async_id_symbol,
|
|
trigger_async_id_symbol } = require('internal/async_hooks').symbols;
|
|
const binding = require(`./build/${common.buildType}/binding`);
|
|
|
|
if (process.env.NODE_TEST_WITH_ASYNC_HOOKS) {
|
|
common.skip('cannot test with env var NODE_TEST_WITH_ASYNC_HOOKS');
|
|
return;
|
|
}
|
|
|
|
// Baseline to make sure the internal field isn't being set.
|
|
assert.strictEqual(
|
|
binding.getPromiseField(Promise.resolve(1)),
|
|
0);
|
|
|
|
const emptyHook = async_hooks.createHook({}).enable();
|
|
|
|
// Check that no PromiseWrap is created when there are no hook callbacks.
|
|
assert.strictEqual(
|
|
binding.getPromiseField(Promise.resolve(1)),
|
|
0);
|
|
|
|
emptyHook.disable();
|
|
|
|
let lastResource;
|
|
let lastAsyncId;
|
|
let lastTriggerAsyncId;
|
|
const initOnlyHook = async_hooks.createHook({
|
|
init(asyncId, type, triggerAsyncId, resource) {
|
|
lastAsyncId = asyncId;
|
|
lastTriggerAsyncId = triggerAsyncId;
|
|
lastResource = resource;
|
|
}
|
|
}).enable();
|
|
|
|
// Check that no PromiseWrap is created when only using an init hook.
|
|
{
|
|
const promise = Promise.resolve(1);
|
|
assert.strictEqual(binding.getPromiseField(promise), 0);
|
|
assert.strictEqual(lastResource, promise);
|
|
assert.strictEqual(lastAsyncId, promise[async_id_symbol]);
|
|
assert.strictEqual(lastTriggerAsyncId, promise[trigger_async_id_symbol]);
|
|
}
|
|
|
|
initOnlyHook.disable();
|
|
|
|
lastResource = null;
|
|
const hookWithDestroy = async_hooks.createHook({
|
|
init(asyncId, type, triggerAsyncId, resource) {
|
|
lastAsyncId = asyncId;
|
|
lastTriggerAsyncId = triggerAsyncId;
|
|
lastResource = resource;
|
|
},
|
|
|
|
destroy() {
|
|
|
|
}
|
|
}).enable();
|
|
|
|
// Check that the internal field returns the same PromiseWrap passed to init().
|
|
{
|
|
const promise = Promise.resolve(1);
|
|
const promiseWrap = binding.getPromiseField(promise);
|
|
assert.strictEqual(lastResource, promiseWrap);
|
|
assert.strictEqual(lastAsyncId, promiseWrap[async_id_symbol]);
|
|
assert.strictEqual(lastTriggerAsyncId, promiseWrap[trigger_async_id_symbol]);
|
|
}
|
|
|
|
hookWithDestroy.disable();
|
|
|
|
// Check that internal fields are no longer being set. This needs to be delayed
|
|
// a bit because the `disable()` call only schedules disabling the hook in a
|
|
// future microtask.
|
|
setImmediate(() => {
|
|
assert.strictEqual(
|
|
binding.getPromiseField(Promise.resolve(1)),
|
|
0);
|
|
|
|
const noDestroyHook = async_hooks.createHook({
|
|
init(asyncId, type, triggerAsyncId, resource) {
|
|
lastAsyncId = asyncId;
|
|
lastTriggerAsyncId = triggerAsyncId;
|
|
lastResource = resource;
|
|
},
|
|
|
|
before() {},
|
|
after() {},
|
|
resolve() {}
|
|
}).enable();
|
|
|
|
// Check that no PromiseWrap is created when there is no destroy hook.
|
|
const promise = Promise.resolve(1);
|
|
assert.strictEqual(binding.getPromiseField(promise), 0);
|
|
assert.strictEqual(lastResource, promise);
|
|
assert.strictEqual(lastAsyncId, promise[async_id_symbol]);
|
|
assert.strictEqual(lastTriggerAsyncId, promise[trigger_async_id_symbol]);
|
|
|
|
noDestroyHook.disable();
|
|
});
|