diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-21 20:56:19 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-21 20:56:19 +0000 |
commit | 0b6210cd37b68b94252cb798598b12974a20e1c1 (patch) | |
tree | e371686554a877842d95aa94f100bee552ff2a8e /test/abort-controller.js | |
parent | Initial commit. (diff) | |
download | node-undici-0b6210cd37b68b94252cb798598b12974a20e1c1.tar.xz node-undici-0b6210cd37b68b94252cb798598b12974a20e1c1.zip |
Adding upstream version 5.28.2+dfsg1+~cs23.11.12.3.upstream/5.28.2+dfsg1+_cs23.11.12.3upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/abort-controller.js')
-rw-r--r-- | test/abort-controller.js | 238 |
1 files changed, 238 insertions, 0 deletions
diff --git a/test/abort-controller.js b/test/abort-controller.js new file mode 100644 index 0000000..4658686 --- /dev/null +++ b/test/abort-controller.js @@ -0,0 +1,238 @@ +'use strict' + +const { test } = require('tap') +const { AbortController: NPMAbortController } = require('abort-controller') +const { Client, errors } = require('..') +const { createServer } = require('http') +const { createReadStream } = require('fs') +const { wrapWithAsyncIterable } = require('./utils/async-iterators') + +const controllers = [{ + AbortControllerImpl: NPMAbortController, + controllerName: 'npm-abortcontroller-shim' +}] + +if (global.AbortController) { + controllers.push({ + AbortControllerImpl: global.AbortController, + controllerName: 'native-abortcontroller' + }) +} +for (const { AbortControllerImpl, controllerName } of controllers) { + test(`Abort ${controllerName} before creating request`, (t) => { + t.plan(1) + + const server = createServer((req, res) => { + t.fail() + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + const abortController = new AbortControllerImpl() + t.teardown(client.destroy.bind(client)) + + abortController.abort() + + client.request({ path: '/', method: 'GET', signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + + test(`Abort ${controllerName} before sending request (no body)`, (t) => { + t.plan(3) + + let count = 0 + const server = createServer((req, res) => { + if (count === 1) { + t.fail('The second request should never be executed') + } + count += 1 + res.end('hello') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + const abortController = new AbortControllerImpl() + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'GET' }, (err, response) => { + t.error(err) + const bufs = [] + response.body.on('data', (buf) => { + bufs.push(buf) + }) + response.body.on('end', () => { + t.equal('hello', Buffer.concat(bufs).toString('utf8')) + }) + }) + + client.request({ path: '/', method: 'GET', signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + + abortController.abort() + }) + }) + + test(`Abort ${controllerName} while waiting response (no body)`, (t) => { + t.plan(1) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + abortController.abort() + res.setHeader('content-type', 'text/plain') + res.end('hello world') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'GET', signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + + test(`Abort ${controllerName} while waiting response (write headers started) (no body)`, (t) => { + t.plan(1) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + res.writeHead(200, { 'content-type': 'text/plain' }) + res.flushHeaders() + abortController.abort() + res.end('hello world') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'GET', signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + + test(`Abort ${controllerName} while waiting response (write headers and write body started) (no body)`, (t) => { + t.plan(2) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + res.writeHead(200, { 'content-type': 'text/plain' }) + res.write('hello') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'GET', signal: abortController.signal }, (err, response) => { + t.error(err) + response.body.on('data', () => { + abortController.abort() + }) + response.body.on('error', err => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + }) + + function waitingWithBody (body, type) { // eslint-disable-line + test(`Abort ${controllerName} while waiting response (with body ${type})`, (t) => { + t.plan(1) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + abortController.abort() + res.setHeader('content-type', 'text/plain') + res.end('hello world') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'POST', body, signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + } + + waitingWithBody('hello', 'string') + waitingWithBody(createReadStream(__filename), 'stream') + waitingWithBody(new Uint8Array([42]), 'Uint8Array') + waitingWithBody(wrapWithAsyncIterable(createReadStream(__filename)), 'async-iterator') + + function writeHeadersStartedWithBody (body, type) { // eslint-disable-line + test(`Abort ${controllerName} while waiting response (write headers started) (with body ${type})`, (t) => { + t.plan(1) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + res.writeHead(200, { 'content-type': 'text/plain' }) + res.flushHeaders() + abortController.abort() + res.end('hello world') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'POST', body, signal: abortController.signal }, (err, response) => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + } + + writeHeadersStartedWithBody('hello', 'string') + writeHeadersStartedWithBody(createReadStream(__filename), 'stream') + writeHeadersStartedWithBody(new Uint8Array([42]), 'Uint8Array') + writeHeadersStartedWithBody(wrapWithAsyncIterable(createReadStream(__filename)), 'async-iterator') + + function writeBodyStartedWithBody (body, type) { // eslint-disable-line + test(`Abort ${controllerName} while waiting response (write headers and write body started) (with body ${type})`, (t) => { + t.plan(2) + + const abortController = new AbortControllerImpl() + const server = createServer((req, res) => { + res.writeHead(200, { 'content-type': 'text/plain' }) + res.write('hello') + }) + t.teardown(server.close.bind(server)) + + server.listen(0, () => { + const client = new Client(`http://localhost:${server.address().port}`) + t.teardown(client.destroy.bind(client)) + + client.request({ path: '/', method: 'POST', body, signal: abortController.signal }, (err, response) => { + t.error(err) + response.body.on('data', () => { + abortController.abort() + }) + response.body.on('error', err => { + t.type(err, errors.RequestAbortedError) + }) + }) + }) + }) + } + + writeBodyStartedWithBody('hello', 'string') + writeBodyStartedWithBody(createReadStream(__filename), 'stream') + writeBodyStartedWithBody(new Uint8Array([42]), 'Uint8Array') + writeBodyStartedWithBody(wrapWithAsyncIterable(createReadStream(__filename), 'async-iterator')) +} |