mirror of
https://github.com/nodejs/node.git
synced 2024-11-21 10:59:27 +00:00
fe751df537
PR-URL: https://github.com/nodejs/node/pull/44943 Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Benjamin Gruenbaum <benjamingr@gmail.com> Reviewed-By: Chengzhong Wu <legendecas@gmail.com> Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Gerhard Stöbich <deb2001-github@yahoo.de> Reviewed-By: Rafael Gonzaga <rafael.nunu@hotmail.com> Reviewed-By: Bryan English <bryan@bryanenglish.com>
109 lines
3.1 KiB
JavaScript
109 lines
3.1 KiB
JavaScript
'use strict';
|
|
|
|
const common = require('../common');
|
|
const assert = require('assert');
|
|
const dc = require('diagnostics_channel');
|
|
const { AsyncLocalStorage } = require('async_hooks');
|
|
|
|
let n = 0;
|
|
const thisArg = new Date();
|
|
const inputs = [
|
|
{ foo: 'bar' },
|
|
{ baz: 'buz' },
|
|
];
|
|
|
|
const channel = dc.channel('test');
|
|
|
|
// Bind a storage directly to published data
|
|
const store1 = new AsyncLocalStorage();
|
|
channel.bindStore(store1);
|
|
let store1bound = true;
|
|
|
|
// Bind a store with transformation of published data
|
|
const store2 = new AsyncLocalStorage();
|
|
channel.bindStore(store2, common.mustCall((data) => {
|
|
assert.strictEqual(data, inputs[n]);
|
|
return { data };
|
|
}, 4));
|
|
|
|
// Regular subscribers should see publishes from runStores calls
|
|
channel.subscribe(common.mustCall((data) => {
|
|
if (store1bound) {
|
|
assert.deepStrictEqual(data, store1.getStore());
|
|
}
|
|
assert.deepStrictEqual({ data }, store2.getStore());
|
|
assert.strictEqual(data, inputs[n]);
|
|
}, 4));
|
|
|
|
// Verify stores are empty before run
|
|
assert.strictEqual(store1.getStore(), undefined);
|
|
assert.strictEqual(store2.getStore(), undefined);
|
|
|
|
channel.runStores(inputs[n], common.mustCall(function(a, b) {
|
|
// Verify this and argument forwarding
|
|
assert.strictEqual(this, thisArg);
|
|
assert.strictEqual(a, 1);
|
|
assert.strictEqual(b, 2);
|
|
|
|
// Verify store 1 state matches input
|
|
assert.strictEqual(store1.getStore(), inputs[n]);
|
|
|
|
// Verify store 2 state has expected transformation
|
|
assert.deepStrictEqual(store2.getStore(), { data: inputs[n] });
|
|
|
|
// Should support nested contexts
|
|
n++;
|
|
channel.runStores(inputs[n], common.mustCall(function() {
|
|
// Verify this and argument forwarding
|
|
assert.strictEqual(this, undefined);
|
|
|
|
// Verify store 1 state matches input
|
|
assert.strictEqual(store1.getStore(), inputs[n]);
|
|
|
|
// Verify store 2 state has expected transformation
|
|
assert.deepStrictEqual(store2.getStore(), { data: inputs[n] });
|
|
}));
|
|
n--;
|
|
|
|
// Verify store 1 state matches input
|
|
assert.strictEqual(store1.getStore(), inputs[n]);
|
|
|
|
// Verify store 2 state has expected transformation
|
|
assert.deepStrictEqual(store2.getStore(), { data: inputs[n] });
|
|
}), thisArg, 1, 2);
|
|
|
|
// Verify stores are empty after run
|
|
assert.strictEqual(store1.getStore(), undefined);
|
|
assert.strictEqual(store2.getStore(), undefined);
|
|
|
|
// Verify unbinding works
|
|
assert.ok(channel.unbindStore(store1));
|
|
store1bound = false;
|
|
|
|
// Verify unbinding a store that is not bound returns false
|
|
assert.ok(!channel.unbindStore(store1));
|
|
|
|
n++;
|
|
channel.runStores(inputs[n], common.mustCall(() => {
|
|
// Verify after unbinding store 1 will remain undefined
|
|
assert.strictEqual(store1.getStore(), undefined);
|
|
|
|
// Verify still bound store 2 receives expected data
|
|
assert.deepStrictEqual(store2.getStore(), { data: inputs[n] });
|
|
}));
|
|
|
|
// Contain transformer errors and emit on next tick
|
|
const fail = new Error('fail');
|
|
channel.bindStore(store1, () => {
|
|
throw fail;
|
|
});
|
|
|
|
let calledRunStores = false;
|
|
process.once('uncaughtException', common.mustCall((err) => {
|
|
assert.strictEqual(calledRunStores, true);
|
|
assert.strictEqual(err, fail);
|
|
}));
|
|
|
|
channel.runStores(inputs[n], common.mustCall());
|
|
calledRunStores = true;
|