1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-12-29 18:49:07 -05:00
denoland-deno/tests/unit_node/async_hooks_test.ts

175 lines
4.6 KiB
TypeScript
Raw Normal View History

// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
import { AsyncLocalStorage, AsyncResource } from "node:async_hooks";
import process from "node:process";
import { setImmediate } from "node:timers";
import { assert, assertEquals } from "@std/assert";
Deno.test(async function foo() {
const asyncLocalStorage = new AsyncLocalStorage();
const out: string[] = [];
function logWithId(msg: string) {
const id = asyncLocalStorage.getStore();
out.push(`${id !== undefined ? id : "-"}: ${msg}`);
}
async function exec() {
logWithId("start");
await new Promise((resolve) => setTimeout(resolve, 100));
logWithId("finish");
}
for (const foo of [1, 2, 3]) {
asyncLocalStorage.run(foo, exec);
}
await new Promise((resolve) => setTimeout(resolve, 500));
assertEquals(out, [
"1: start",
"2: start",
"3: start",
"1: finish",
"2: finish",
"3: finish",
]);
});
Deno.test(async function bar() {
let differentScopeDone = false;
const als = new AsyncLocalStorage();
const ac = new AbortController();
const server = Deno.serve({
signal: ac.signal,
port: 4000,
}, () => {
const differentScope = als.run(123, () =>
AsyncResource.bind(() => {
differentScopeDone = true;
}));
return als.run("Hello World", async () => {
// differentScope is attached to a different async context, so
// it will see a different value for als.getStore() (123)
setTimeout(differentScope, 5);
// Some simulated async delay.
await new Promise((res) => setTimeout(res, 10));
return new Response(als.getStore() as string); // "Hello World"
});
});
const res = await fetch("http://localhost:4000");
assertEquals(await res.text(), "Hello World");
ac.abort();
await server.finished;
assert(differentScopeDone);
});
Deno.test(async function nested() {
const als = new AsyncLocalStorage();
const deferred = Promise.withResolvers();
const deferred1 = Promise.withResolvers();
als.run(null, () => {
als.run({ x: 1 }, () => {
deferred.resolve(als.getStore());
});
deferred1.resolve(als.getStore());
});
assertEquals(await deferred.promise, { x: 1 });
assertEquals(await deferred1.promise, null);
});
Deno.test(async function enterWith() {
const als = new AsyncLocalStorage();
const deferred = Promise.withResolvers();
const deferred1 = Promise.withResolvers();
als.run(null, () => {
als.run({ x: 1 }, () => {
als.enterWith({ x: 2 });
deferred.resolve(als.getStore());
});
deferred1.resolve(als.getStore());
});
assertEquals(await deferred.promise, { x: 2 });
assertEquals(await deferred1.promise, null);
});
Deno.test(async function snapshot() {
const als = new AsyncLocalStorage();
const deferred = Promise.withResolvers();
als.run(null, () => {
const snapshot = AsyncLocalStorage.snapshot();
als.run({ x: 1 }, () => {
deferred.resolve(snapshot(() => als.getStore()));
});
});
assertEquals(await deferred.promise, null);
});
Deno.test(async function bind() {
const als = new AsyncLocalStorage();
const deferred = Promise.withResolvers();
const bound = als.run(null, () => {
return AsyncLocalStorage.bind(() => {
deferred.resolve(als.getStore());
});
});
als.run({ x: 1 }, () => {
bound();
});
assertEquals(await deferred.promise, null);
});
Deno.test(function asyncResourceStub() {
const resource = new AsyncResource("dbquery");
assert(typeof resource.asyncId() === "number");
});
Deno.test(function emitDestroyStub() {
const resource = new AsyncResource("foo");
assert(typeof resource.emitDestroy === "function");
});
Deno.test(async function worksWithAsyncAPIs() {
const store = new AsyncLocalStorage();
const test = () => assertEquals(store.getStore(), "data");
await store.run("data", async () => {
test();
queueMicrotask(() => test());
process.nextTick(() => test());
setImmediate(() => test());
setTimeout(() => test(), 0);
const intervalId = setInterval(() => {
test();
clearInterval(intervalId);
}, 0);
store.run("data2", () => {
assertEquals(store.getStore(), "data2");
});
await new Promise((r) => setTimeout(r, 50));
test();
});
});
Deno.test(async function worksWithDynamicImports() {
const store = new AsyncLocalStorage();
// @ts-expect-error implicit any
globalThis.alsDynamicImport = () => store.getStore();
const dataUrl =
`data:application/javascript,export const data = alsDynamicImport()`;
await store.run("data", async () => {
const { data } = await import(dataUrl);
assertEquals(data, "data");
});
});