mirror of
https://github.com/denoland/deno.git
synced 2024-12-23 15:49:44 -05:00
chore: add tests for node:async_hooks (#18004)
Closes https://github.com/denoland/deno/issues/17878 --------- Co-authored-by: crowlkats <crowlkats@toaxl.com>
This commit is contained in:
parent
e6fe1630f0
commit
0bc6bf5d33
2 changed files with 88 additions and 1 deletions
83
cli/tests/unit_node/async_hooks_test.ts
Normal file
83
cli/tests/unit_node/async_hooks_test.ts
Normal file
|
@ -0,0 +1,83 @@
|
|||
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
|
||||
import { AsyncLocalStorage, AsyncResource } from "node:async_hooks";
|
||||
import {
|
||||
assert,
|
||||
assertEquals,
|
||||
} from "../../../test_util/std/testing/asserts.ts";
|
||||
import { deferred } from "../../../test_util/std/async/deferred.ts";
|
||||
|
||||
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(() => {
|
||||
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"
|
||||
});
|
||||
}, {
|
||||
signal: ac.signal,
|
||||
port: 4000,
|
||||
});
|
||||
|
||||
const res = await fetch("http://localhost:4000");
|
||||
assertEquals(await res.text(), "Hello World");
|
||||
ac.abort();
|
||||
await server;
|
||||
assert(differentScopeDone);
|
||||
});
|
||||
|
||||
Deno.test(async function nested() {
|
||||
const als = new AsyncLocalStorage();
|
||||
const promise = deferred();
|
||||
const promise1 = deferred();
|
||||
|
||||
als.run(null, () => {
|
||||
als.run({ x: 1 }, () => {
|
||||
promise.resolve(als.getStore());
|
||||
});
|
||||
promise1.resolve(als.getStore());
|
||||
});
|
||||
|
||||
assertEquals(await promise, { x: 1 });
|
||||
assertEquals(await promise1, null);
|
||||
});
|
|
@ -84,7 +84,7 @@ class AsyncContextFrame {
|
|||
|
||||
const propagate = (parent: AsyncContextFrame) => {
|
||||
parent.storage = parent.storage.filter((entry) => !entry.key.isDead());
|
||||
parent.storage.forEach((entry) => this.storage.push(entry));
|
||||
parent.storage.forEach((entry) => this.storage.push(entry.clone()));
|
||||
|
||||
if (maybeStorageEntry) {
|
||||
const existingEntry = this.storage.find((entry) =>
|
||||
|
@ -250,6 +250,10 @@ class StorageEntry {
|
|||
this.key = key;
|
||||
this.value = value;
|
||||
}
|
||||
|
||||
clone() {
|
||||
return new StorageEntry(this.key, this.value);
|
||||
}
|
||||
}
|
||||
|
||||
class StorageKey {
|
||||
|
|
Loading…
Reference in a new issue