Fix up async_hooks polyfill

1 files changed, 30 insertions, 0 deletions

diff --git a/test/bun.js/disabled-module.test.js b/test/bun.js/disabled-module.test.js
index 43b08731d..c12676959 100644
--- a/test/bun.js/disabled-module.test.js
+++ b/test/bun.js/disabled-module.test.js
@@ -6,3 +6,33 @@ test("not implemented yet module masquerades as undefined and throws an error",
   expect(typeof worker_threads).toBe("undefined");
   expect(typeof worker_threads.getEnvironmentData).toBe("undefined");
 });
+
+test("AsyncLocalStorage polyfill", () => {
+  const { AsyncLocalStorage } = import.meta.require("async_hooks");
+
+  const store = new AsyncLocalStorage();
+  var called = false;
+  expect(store.getStore()).toBe(null);
+  store.run({ foo: "bar" }, () => {
+    expect(store.getStore()).toEqual({ foo: "bar" });
+    called = true;
+  });
+  expect(store.getStore()).toBe(null);
+  expect(called).toBe(true);
+});
+
+test("AsyncResource polyfill", () => {
+  const { AsyncResource } = import.meta.require("async_hooks");
+
+  const resource = new AsyncResource("test");
+  var called = false;
+  resource.runInAsyncScope(
+    () => {
+      called = true;
+    },
+    null,
+    "foo",
+    "bar",
+  );
+  expect(called).toBe(true);
+});