1 files changed, 23 insertions, 5 deletions
diff --git a/test/js/node/async_hooks/async_hooks.node.test.ts b/test/js/node/async_hooks/async_hooks.node.test.ts
index 3d6183948..5fc56a39b 100644
--- a/test/js/node/async_hooks/async_hooks.node.test.ts
+++ b/test/js/node/async_hooks/async_hooks.node.test.ts
@@ -1,13 +1,13 @@
-import { AsyncLocalStorage } from "async_hooks";
+import { AsyncLocalStorage, AsyncResource } from "async_hooks";
 import assert from "assert";
 
 test("node async_hooks.AsyncLocalStorage enable disable", async done => {
-  const asyncLocalStorage = new AsyncLocalStorage();
+  const asyncLocalStorage = new AsyncLocalStorage<Map<string, any>>();
 
   asyncLocalStorage.run(new Map(), () => {
-    asyncLocalStorage.getStore().set("foo", "bar");
+    asyncLocalStorage.getStore()!.set("foo", "bar");
     process.nextTick(() => {
-      assert.strictEqual(asyncLocalStorage.getStore().get("foo"), "bar");
+      assert.strictEqual(asyncLocalStorage.getStore()!.get("foo"), "bar");
       process.nextTick(() => {
         assert.strictEqual(asyncLocalStorage.getStore(), undefined);
       });
@@ -24,7 +24,7 @@ test("node async_hooks.AsyncLocalStorage enable disable", async done => {
       process.nextTick(() => {
         assert.strictEqual(asyncLocalStorage.getStore(), undefined);
         asyncLocalStorage.run(new Map().set("bar", "foo"), () => {
-          assert.strictEqual(asyncLocalStorage.getStore().get("bar"), "foo");
+          assert.strictEqual(asyncLocalStorage.getStore()!.get("bar"), "foo");
 
           done();
         });
@@ -32,3 +32,21 @@ test("node async_hooks.AsyncLocalStorage enable disable", async 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);
+});