test-simple-test-https-finish.js

// 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.

var common = require('../common');
var assert = require('assert');

var assert = require('assert');
var https = require('https');
var stream = require('stream');
var fs = require('fs');
var options = {
  key: fs.readFileSync(common.fixturesDir + '/keys/agent1-key.pem'),
  cert: fs.readFileSync(common.fixturesDir + '/keys/agent1-cert.pem')
};

var url = require('url');

var chunks = 5;

var responseBuffer = new Buffer(1024 * 1024 * 16);
responseBuffer.fill('x');
responseBuffer = responseBuffer.toString();

var expect = chunks * responseBuffer.length;

var server = https.createServer(options, function(req, res) {
  console.error('request');
  var src = new stream.Readable();
  var reads = 0;
  src._read = function(n) {
    src.push(reads++ === chunks ? null : responseBuffer);
    console.error('pushing', reads);
  };

  res.setHeader('connection', 'close');
  res.setHeader('x-id', req.socket.id);

  // 'finish' should mean fully flushed.
  res.once('finish', function() {
    console.error('about to destroy');
    req.destroy();
  });

  src.pipe(res);
});

var socketId = 1;
server.on('secureConnection', function(sock) {
  sock.id = socketId++;
});

server.listen(common.PORT, function() {
  makeReq();
//  makeReq();
//  makeReq();
//  makeReq();
//  makeReq();
//  makeReq();
//  makeReq();
//  makeReq();
});

var reqs = 0;

function makeReq() {
  reqs++;
  var r = url.parse('https://127.0.0.1:' + common.PORT);
  r.method = 'PUT';
  r.rejectUnauthorized = false;
  // r.agent = false;
  r = https.request(r, function(res) {
    console.error('response', res.headers);
    var received = 0;
    res.on('data', function(chunk) {
      received += chunk.length;
      console.error('received', res.headers['x-id'], received);
      this.socket.read(0);
      this.socket.pair.encrypted.read(0);
      this.socket.pair.cleartext.read(0);
    });

    res.once('end', function() {
      assert.equal(received, expect, 'got ' + received + ' wanted ' + expect + ' diff=' + (expect-received));
      console.log('ok - socket id #%d', res.headers['x-id']);
      if (--reqs === 0) {
        console.log('ok - got all requests');
        server.close();
      }
    });
  });
  r.end();
}