node/test/pummel/test-keep-alive.js
Johan Bergström a45d4f8fd6 build: remove tools/wrk from the tree
wrk is an optional tool that some of the http benchmarks uses. The removal
doesn't affect any users. Developers are assumed to install it before running
the tests.

This change reduces the tarball by 5%

PR-URL: https://github.com/iojs/io.js/pull/982
Reviewed-By: Rod Vagg <rod@vagg.org>
2015-03-04 20:25:41 -08:00

93 lines
2.2 KiB
JavaScript

if (process.platform === 'win32') {
console.log('skipping this test because there is no wrk on windows');
process.exit(0);
}
// This test requires the program 'wrk'
var common = require('../common');
var assert = require('assert');
var spawn = require('child_process').spawn;
var http = require('http');
var path = require('path');
var url = require('url');
var body = 'hello world\n';
var server = http.createServer(function(req, res) {
res.writeHead(200, {
'Content-Length': body.length,
'Content-Type': 'text/plain'
});
res.write(body);
res.end();
});
var keepAliveReqSec = 0;
var normalReqSec = 0;
function runAb(opts, callback) {
var args = [
'-c', opts.concurrent || 100,
'-t', opts.threads || 2,
'-d', opts.duration || '10s',
];
if (!opts.keepalive) {
args.push('-H');
args.push('Connection: close');
}
args.push(url.format({ hostname: '127.0.0.1', port: common.PORT, protocol: 'http'}));
//console.log(comm, args.join(' '));
var child = spawn('wrk', args);
child.stderr.pipe(process.stderr);
child.stdout.setEncoding('utf8');
var stdout;
child.stdout.on('data', function(data) {
stdout += data;
});
child.on('close', function(code, signal) {
if (code) {
console.error(code, signal);
process.exit(code);
return;
}
var matches = /Requests\/sec:\s*(\d+)\./mi.exec(stdout);
var reqSec = parseInt(matches[1]);
matches = /Keep-Alive requests:\s*(\d+)/mi.exec(stdout);
var keepAliveRequests;
if (matches) {
keepAliveRequests = parseInt(matches[1]);
} else {
keepAliveRequests = 0;
}
callback(reqSec, keepAliveRequests);
});
}
server.listen(common.PORT, function() {
runAb({ keepalive: true }, function(reqSec) {
keepAliveReqSec = reqSec;
console.log('keep-alive:', keepAliveReqSec, 'req/sec');
runAb({ keepalive: false }, function(reqSec) {
normalReqSec = reqSec;
console.log('normal:' + normalReqSec + ' req/sec');
server.close();
});
});
});
process.on('exit', function() {
assert.equal(true, normalReqSec > 50);
assert.equal(true, keepAliveReqSec > 50);
assert.equal(true, normalReqSec < keepAliveReqSec);
});