test: deflake test-http-remove-header-stays-removed

Instead of relying on a timer, verify that `socket.end()` is called when
the `'finish'` event is emitted on the `ServerResponse` object.

PR-URL: https://github.com/nodejs/node/pull/55004
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Jake Yuesong Li <jake.yuesong@gmail.com>
This commit is contained in:
Luigi Pinca 2024-09-26 04:59:29 +02:00 committed by GitHub
parent afe56aa58b
commit 9bb6a1a790
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -25,7 +25,9 @@ const assert = require('assert');
const http = require('http'); const http = require('http');
const server = http.createServer(function(request, response) { const server = http.createServer(common.mustCall(function(request, response) {
const socket = response.socket;
// Removed headers should stay removed, even if node automatically adds them // Removed headers should stay removed, even if node automatically adds them
// to the output: // to the output:
response.removeHeader('connection'); response.removeHeader('connection');
@ -36,32 +38,29 @@ const server = http.createServer(function(request, response) {
response.removeHeader('date'); response.removeHeader('date');
response.setHeader('date', 'coffee o clock'); response.setHeader('date', 'coffee o clock');
response.on('finish', common.mustCall(function() {
// The socket should be closed immediately, with no keep-alive, because
// no content-length or transfer-encoding are used.
assert.strictEqual(socket.writableEnded, true);
}));
response.end('beep boop\n'); response.end('beep boop\n');
}); }));
let response = '';
process.on('exit', function() {
assert.strictEqual(response, 'beep boop\n');
console.log('ok');
});
server.listen(0, function() { server.listen(0, function() {
http.get({ port: this.address().port }, function(res) { http.get({ port: this.address().port }, function(res) {
assert.strictEqual(res.statusCode, 200); assert.strictEqual(res.statusCode, 200);
assert.deepStrictEqual(res.headers, { date: 'coffee o clock' }); assert.deepStrictEqual(res.headers, { date: 'coffee o clock' });
let response = '';
res.setEncoding('ascii'); res.setEncoding('ascii');
res.on('data', function(chunk) { res.on('data', function(chunk) {
response += chunk; response += chunk;
if (response === 'beep boop\n') { });
setTimeout(function() {
// The socket should be closed immediately, with no keep-alive, because res.on('end', function() {
// no content-length or transfer-encoding are used: assert.strictEqual(response, 'beep boop\n');
assert.strictEqual(res.socket.closed, true); server.close();
server.close();
}, common.platformTimeout(15));
}
}); });
}); });
}); });