-
Notifications
You must be signed in to change notification settings - Fork 57
/
server_timeout.test.js
78 lines (72 loc) · 2.1 KB
/
server_timeout.test.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
'use strict';
const assert = require('assert');
const http = require('http');
const Agent = require('..');
describe('test/server_timeout.test.js', () => {
let port;
let server;
let timer;
before(done => {
server = http.createServer((req, res) => {
if (server.keepAliveTimeout) {
res.setHeader('Keep-Alive', `timeout=${parseInt(server.keepAliveTimeout / 1000)}`);
}
res.end('Hello World, ' + req.connection.remotePort);
});
server.on('clientError', (err, socket) => {
socket.end('HTTP/1.1 400 Bad Request\r\n\r\n');
});
server.keepAliveTimeout = 1000;
server.listen(0, err => {
port = server.address().port;
done(err);
});
});
after(() => {
clearInterval(timer);
});
it('should handle Keep-Alive header and not throw reset error', done => {
const keepaliveAgent = new Agent({
keepAlive: true,
});
let count = 0;
function request() {
count++;
const req = http.request({
method: 'GET',
port,
path: '/',
agent: keepaliveAgent,
}, res => {
assert(res.statusCode === 200);
const chunks = [];
res.on('data', data => {
chunks.push(data);
});
res.on('end', () => {
const text = Buffer.concat(chunks).toString();
console.log('[%s] status: %s, text: %s, headers: %j', count, text, res.statusCode, res.headers);
assert(res.headers.connection === 'keep-alive');
assert(res.headers['keep-alive'] === 'timeout=1');
const m = /^timeout=(\d+?)/.exec(res.headers['keep-alive']);
if (m) {
const keepAliveTimeout = parseInt(m[1]) * 1000 - 500;
if (keepAliveTimeout > 0) {
res.socket.freeSocketKeepAliveTimeout = keepAliveTimeout;
}
}
if (count > 5) {
done();
}
});
});
req.on('error', err => {
console.error('[%s] error: %s', count, err);
done(err);
});
req.end();
}
timer = setInterval(request, server.keepAliveTimeout);
request();
});
});