| // Copyright Joyent, Inc. and other Node contributors. |
| // |
| // Permission is hereby granted, free of charge, to any person obtaining a |
| // copy of this software and associated documentation files (the |
| // "Software"), to deal in the Software without restriction, including |
| // without limitation the rights to use, copy, modify, merge, publish, |
| // distribute, sublicense, and/or sell copies of the Software, and to permit |
| // persons to whom the Software is furnished to do so, subject to the |
| // following conditions: |
| // |
| // The above copyright notice and this permission notice shall be included |
| // in all copies or substantial portions of the Software. |
| // |
| // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS |
| // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN |
| // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, |
| // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
| // USE OR OTHER DEALINGS IN THE SOFTWARE. |
| |
| 'use strict'; |
| |
| // This test requires the program 'wrk'. |
| const common = require('../common'); |
| |
| const child_process = require('child_process'); |
| const result = child_process.spawnSync('wrk', ['-h']); |
| if (result.error && result.error.code === 'ENOENT') |
| common.skip('test requires `wrk` to be installed first'); |
| |
| const assert = require('assert'); |
| const http = require('http'); |
| const url = require('url'); |
| |
| const body = 'hello world\n'; |
| const server = http.createServer((req, res) => { |
| res.writeHead(200, { |
| 'Content-Length': body.length, |
| 'Content-Type': 'text/plain' |
| }); |
| res.write(body); |
| res.end(); |
| }); |
| |
| let keepAliveReqSec = 0; |
| let normalReqSec = 0; |
| |
| |
| const runAb = (opts, callback) => { |
| const args = [ |
| '-c', opts.concurrent || 50, |
| '-t', opts.threads || 2, |
| '-d', opts.duration || '5s', |
| ]; |
| |
| if (!opts.keepalive) { |
| args.push('-H'); |
| args.push('Connection: close'); |
| } |
| |
| args.push(url.format({ hostname: '127.0.0.1', |
| port: opts.port, protocol: 'http' })); |
| |
| const child = child_process.spawn('wrk', args); |
| child.stderr.pipe(process.stderr); |
| child.stdout.setEncoding('utf8'); |
| |
| let stdout; |
| |
| child.stdout.on('data', (data) => stdout += data); |
| |
| child.on('close', (code, signal) => { |
| if (code) { |
| console.error(code, signal); |
| process.exit(code); |
| return; |
| } |
| |
| let matches = /Requests\/sec:\s*(\d+)\./i.exec(stdout); |
| const reqSec = parseInt(matches[1]); |
| |
| matches = /Keep-Alive requests:\s*(\d+)/i.exec(stdout); |
| let keepAliveRequests; |
| if (matches) { |
| keepAliveRequests = parseInt(matches[1]); |
| } else { |
| keepAliveRequests = 0; |
| } |
| |
| callback(reqSec, keepAliveRequests); |
| }); |
| }; |
| |
| server.listen(0, () => { |
| const port = server.address().port; |
| runAb({ keepalive: true, port: port }, (reqSec) => { |
| keepAliveReqSec = reqSec; |
| |
| runAb({ keepalive: false, port: port }, (reqSec) => { |
| normalReqSec = reqSec; |
| server.close(); |
| }); |
| }); |
| }); |
| |
| process.on('exit', () => { |
| assert.strictEqual( |
| normalReqSec > 50, |
| true, |
| `normalReqSec should be greater than 50, but got ${normalReqSec}` |
| ); |
| assert.strictEqual( |
| keepAliveReqSec > 50, |
| true, |
| `keepAliveReqSec should be greater than 50, but got ${keepAliveReqSec}` |
| ); |
| assert.strictEqual( |
| normalReqSec < keepAliveReqSec, |
| true, |
| 'normalReqSec should be less than keepAliveReqSec, ' + |
| `but ${normalReqSec} is greater than ${keepAliveReqSec}` |
| ); |
| }); |