forked from nodejs/node
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test: update test-abortsignal-cloneable to use node:test
PR-URL: nodejs#54581 Reviewed-By: Yagiz Nizipli <yagiz@nizipli.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
- Loading branch information
Showing
1 changed file
with
50 additions
and
42 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,83 +1,91 @@ | ||
'use strict'; | ||
|
||
const common = require('../common'); | ||
require('../common'); | ||
const { ok, strictEqual } = require('assert'); | ||
const { setImmediate: pause } = require('timers/promises'); | ||
const { setImmediate: sleep } = require('timers/promises'); | ||
const { | ||
transferableAbortSignal, | ||
transferableAbortController, | ||
} = require('util'); | ||
const { | ||
test, | ||
mock, | ||
} = require('node:test'); | ||
|
||
|
||
function deferred() { | ||
let res; | ||
const promise = new Promise((resolve) => res = resolve); | ||
return { res, promise }; | ||
} | ||
|
||
(async () => { | ||
test('Can create a transferable abort controller', async () => { | ||
const ac = transferableAbortController(); | ||
const mc = new MessageChannel(); | ||
|
||
const deferred1 = deferred(); | ||
const deferred2 = deferred(); | ||
const resolvers = [deferred1, deferred2]; | ||
const setup1 = Promise.withResolvers(); | ||
const setup2 = Promise.withResolvers(); | ||
const setupResolvers = [setup1, setup2]; | ||
|
||
mc.port1.onmessage = common.mustCall(({ data }) => { | ||
data.addEventListener('abort', common.mustCall(() => { | ||
const abort1 = Promise.withResolvers(); | ||
const abort2 = Promise.withResolvers(); | ||
const abort3 = Promise.withResolvers(); | ||
const abortResolvers = [abort1, abort2, abort3]; | ||
|
||
mc.port1.onmessage = ({ data }) => { | ||
data.addEventListener('abort', () => { | ||
strictEqual(data.reason, 'boom'); | ||
})); | ||
resolvers.shift().res(); | ||
}, 2); | ||
abortResolvers.shift().resolve(); | ||
}); | ||
setupResolvers.shift().resolve(); | ||
}; | ||
|
||
mc.port2.postMessage(ac.signal, [ac.signal]); | ||
|
||
// Can be cloned/transferd multiple times and they all still work | ||
mc.port2.postMessage(ac.signal, [ac.signal]); | ||
|
||
mc.port2.close(); | ||
|
||
// Although we're using transfer semantics, the local AbortSignal | ||
// is still usable locally. | ||
ac.signal.addEventListener('abort', common.mustCall(() => { | ||
ac.signal.addEventListener('abort', () => { | ||
strictEqual(ac.signal.reason, 'boom'); | ||
})); | ||
abortResolvers.shift().resolve(); | ||
}); | ||
|
||
await Promise.all([ deferred1.promise, deferred2.promise ]); | ||
await Promise.all([ setup1.promise, setup2.promise ]); | ||
|
||
ac.abort('boom'); | ||
|
||
// Because the postMessage used by the underlying AbortSignal | ||
// takes at least one turn of the event loop to be processed, | ||
// and because it is unref'd, it won't, by itself, keep the | ||
// event loop open long enough for the test to complete, so | ||
// we schedule two back to back turns of the event to ensure | ||
// the loop runs long enough for the test to complete. | ||
await pause(); | ||
await pause(); | ||
await Promise.all([ abort1.promise, abort2.promise, abort3.promise ]); | ||
|
||
mc.port2.close(); | ||
|
||
})().then(common.mustCall()); | ||
}); | ||
|
||
{ | ||
test('Can create a transferable abort signal', async () => { | ||
const signal = transferableAbortSignal(AbortSignal.abort('boom')); | ||
ok(signal.aborted); | ||
strictEqual(signal.reason, 'boom'); | ||
const mc = new MessageChannel(); | ||
mc.port1.onmessage = common.mustCall(({ data }) => { | ||
const { promise, resolve } = Promise.withResolvers(); | ||
mc.port1.onmessage = ({ data }) => { | ||
ok(data instanceof AbortSignal); | ||
ok(data.aborted); | ||
strictEqual(data.reason, 'boom'); | ||
mc.port1.close(); | ||
}); | ||
resolve(); | ||
}; | ||
mc.port2.postMessage(signal, [signal]); | ||
} | ||
await promise; | ||
mc.port1.close(); | ||
}); | ||
|
||
{ | ||
// The cloned AbortSignal does not keep the event loop open | ||
// waiting for the abort to be triggered. | ||
test('A cloned AbortSignal does not keep the event loop open', async () => { | ||
const ac = transferableAbortController(); | ||
const mc = new MessageChannel(); | ||
mc.port1.onmessage = common.mustCall(); | ||
const fn = mock.fn(); | ||
mc.port1.onmessage = fn; | ||
mc.port2.postMessage(ac.signal, [ac.signal]); | ||
// Because the postMessage used by the underlying AbortSignal | ||
// takes at least one turn of the event loop to be processed, | ||
// and because it is unref'd, it won't, by itself, keep the | ||
// event loop open long enough for the test to complete, so | ||
// we schedule two back to back turns of the event to ensure | ||
// the loop runs long enough for the test to complete. | ||
await sleep(); | ||
await sleep(); | ||
strictEqual(fn.mock.calls.length, 1); | ||
mc.port2.close(); | ||
} | ||
}); |