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
|
import { AsyncLocalStorage, AsyncResource } from "async_hooks";
import assert from "assert";
test("node async_hooks.AsyncLocalStorage enable disable", async done => {
const asyncLocalStorage = new AsyncLocalStorage<Map<string, any>>();
asyncLocalStorage.run(new Map(), () => {
asyncLocalStorage.getStore()!.set("foo", "bar");
process.nextTick(() => {
assert.strictEqual(asyncLocalStorage.getStore()!.get("foo"), "bar");
process.nextTick(() => {
assert.strictEqual(asyncLocalStorage.getStore(), undefined);
});
asyncLocalStorage.disable();
assert.strictEqual(asyncLocalStorage.getStore(), undefined);
// Calls to exit() should not mess with enabled status
asyncLocalStorage.exit(() => {
assert.strictEqual(asyncLocalStorage.getStore(), undefined);
});
assert.strictEqual(asyncLocalStorage.getStore(), undefined);
process.nextTick(() => {
assert.strictEqual(asyncLocalStorage.getStore(), undefined);
asyncLocalStorage.run(new Map().set("bar", "foo"), () => {
assert.strictEqual(asyncLocalStorage.getStore()!.get("bar"), "foo");
done();
});
});
});
});
});
test("AsyncResource.prototype.bind", () => {
const localStorage = new AsyncLocalStorage<true>();
let ar!: AsyncResource;
localStorage.run(true, () => {
ar = new AsyncResource("test");
});
expect(ar.bind(() => localStorage.getStore())()).toBe(true);
});
test("AsyncResource.bind", () => {
const localStorage = new AsyncLocalStorage<true>();
let fn!: () => true | undefined;
localStorage.run(true, () => {
fn = AsyncResource.bind(() => localStorage.getStore());
});
expect(fn()).toBe(true);
});
|
