1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-12-25 00:29:09 -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:
Bartek Iwańczuk 2023-03-17 13:53:57 -04:00 committed by GitHub
parent e6fe1630f0
commit 0bc6bf5d33
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 88 additions and 1 deletions

View 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);
});

View file

@ -84,7 +84,7 @@ class AsyncContextFrame {
const propagate = (parent: AsyncContextFrame) => { const propagate = (parent: AsyncContextFrame) => {
parent.storage = parent.storage.filter((entry) => !entry.key.isDead()); 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) { if (maybeStorageEntry) {
const existingEntry = this.storage.find((entry) => const existingEntry = this.storage.find((entry) =>
@ -250,6 +250,10 @@ class StorageEntry {
this.key = key; this.key = key;
this.value = value; this.value = value;
} }
clone() {
return new StorageEntry(this.key, this.value);
}
} }
class StorageKey { class StorageKey {