test: issue #2026

test/fetch/issue-2026.js

@@ -0,0 +1,75 @@
+'use strict'
+
+const { cpus } = require('os')
+const { isMainThread, Worker, parentPort } = require('worker_threads')
+const { test } = require('tap')
+const { fetch } = require('../..')
+
+const THREADS = cpus().length - 1
+const ROUNDS = 50
+const PARALLEL_REQUESTS = 6
+
+if (isMainThread) {
+ test('https:/nodejs/undici/issues/2026', async (t) => {
+ let stuckWorkers = 0
+ const timeout = setTimeout(() => {
+ if (stuckWorkers > 0) {
+ t.fail(
+ `Not finished in 40s. ${stuckWorkers} workers finished fetch() calls but cannot exit.`
+ )
+ }
+ }, 40_000)
+
+ async function task (workerId) {
+ const worker = new Worker(__filename)
+
+ await new Promise((resolve) =>
+ worker.on(
+ 'message',
+ (message) => message === 'FETCH_FINISHED' && resolve()
+ )
+ )
+
+ const timer = setTimeout(() => {
+ stuckWorkers++
+ console.error(`Unable to terminate worker #${workerId}`)
+ }, 10_000)
+ await worker.terminate()
+ clearTimeout(timer)
+ }
+
+ const pool = Array(ROUNDS).fill(task)
+
+ async function execute () {
+ const task = pool.shift()
+
+ if (task) {
+ await task(ROUNDS - pool.length)
+ return execute()
+ }
+ }
+
+ await Promise.all(
+ Array(THREADS)
+ .fill(execute)
+ .map((task) => task())
+ )
+
+ clearTimeout(timeout)
+ })
+} else {
+ const urls = [
+ new URL('/v4/starlink', 'https://api.spacexdata.com'),
+ new URL(
+ '/v4/starlink/5eed770f096e59000698560d',
+ 'https://api.spacexdata.com'
+ )
+ ]
+
+ const count = Math.round(PARALLEL_REQUESTS / urls.length)
+ const requests = Array(count)
+ .fill(urls.map((url) => fetch(url).then((r) => r.json())))
+ .flat()
+
+ Promise.all(requests).then(() => parentPort.postMessage('FETCH_FINISHED'))
+}