summaryrefslogtreecommitdiff
path: root/test/parallel/test-async-hooks-correctly-switch-promise-hook.js
blob: 73127f1ebaf94cfed474604e0aca9be136568ac2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
'use strict';
require('../common');
const assert = require('assert');
const async_hooks = require('async_hooks');

// Regression test for:
// - https://github.com/nodejs/node/issues/38814
// - https://github.com/nodejs/node/issues/38815

const layers = new Map();

// Only init to start context-based promise hook
async_hooks.createHook({
  init(asyncId, type) {
    layers.set(asyncId, {
      type,
      init: true,
      before: false,
      after: false,
      promiseResolve: false
    });
  },
  before(asyncId) {
    if (layers.has(asyncId)) {
      layers.get(asyncId).before = true;
    }
  },
  after(asyncId) {
    if (layers.has(asyncId)) {
      layers.get(asyncId).after = true;
    }
  },
  promiseResolve(asyncId) {
    if (layers.has(asyncId)) {
      layers.get(asyncId).promiseResolve = true;
    }
  }
}).enable();

// With destroy, this should switch to native
// and disable context - based promise hook
async_hooks.createHook({
  init() { },
  destroy() { }
}).enable();

async function main() {
  return Promise.resolve();
}

main();

process.on('exit', () => {
  assert.deepStrictEqual(Array.from(layers.values()), [
    {
      type: 'PROMISE',
      init: true,
      before: true,
      after: true,
      promiseResolve: true
    },
    {
      type: 'PROMISE',
      init: true,
      before: false,
      after: false,
      promiseResolve: true
    },
    {
      type: 'PROMISE',
      init: true,
      before: true,
      after: true,
      promiseResolve: true
    },
  ]);
});