test/parallel/test-http-after-connect.js

   1 // Copyright Joyent, Inc. and other Node contributors.
   2 //
   3 // Permission is hereby granted, free of charge, to any person obtaining a
   4 // copy of this software and associated documentation files (the
   5 // "Software"), to deal in the Software without restriction, including
   6 // without limitation the rights to use, copy, modify, merge, publish,
   7 // distribute, sublicense, and/or sell copies of the Software, and to permit
   8 // persons to whom the Software is furnished to do so, subject to the
   9 // following conditions:
  10 //
  11 // The above copyright notice and this permission notice shall be included
  12 // in all copies or substantial portions of the Software.
  13 //
  14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  15 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
  17 // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
  18 // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  19 // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  20 // USE OR OTHER DEALINGS IN THE SOFTWARE.
  21
  22 var common = require('../common');
  23 var assert = require('assert');
  24 var http = require('http');
  25
  26 var serverConnected = false;
  27 var serverRequests = 0;
  28 var clientResponses = 0;
  29
  30 var server = http.createServer(function(req, res) {
  31   common.debug('Server got GET request');
  32   req.resume();
  33   ++serverRequests;
  34   res.writeHead(200);
  35   res.write('');
  36   setTimeout(function() {
  37     res.end(req.url);
  38   }, 50);
  39 });
  40 server.on('connect', function(req, socket, firstBodyChunk) {
  41   common.debug('Server got CONNECT request');
  42   serverConnected = true;
  43   socket.write('HTTP/1.1 200 Connection established\r\n\r\n');
  44   socket.resume();
  45   socket.on('end', function() {
  46     socket.end();
  47   });
  48 });
  49 server.listen(common.PORT, function() {
  50   var req = http.request({
  51     port: common.PORT,
  52     method: 'CONNECT',
  53     path: 'google.com:80'
  54   });
  55   req.on('connect', function(res, socket, firstBodyChunk) {
  56     common.debug('Client got CONNECT response');
  57     socket.end();
  58     socket.on('end', function() {
  59       doRequest(0);
  60       doRequest(1);
  61     });
  62     socket.resume();
  63   });
  64   req.end();
  65 });
  66
  67 function doRequest(i) {
  68   var req = http.get({
  69     port: common.PORT,
  70     path: '/request' + i
  71   }, function(res) {
  72     common.debug('Client got GET response');
  73     var data = '';
  74     res.setEncoding('utf8');
  75     res.on('data', function(chunk) {
  76       data += chunk;
  77     });
  78     res.on('end', function() {
  79       assert.equal(data, '/request' + i);
  80       ++clientResponses;
  81       if (clientResponses === 2) {
  82         server.close();
  83       }
  84     });
  85   });
  86 }
  87
  88 process.on('exit', function() {
  89   assert(serverConnected);
  90   assert.equal(serverRequests, 2);
  91   assert.equal(clientResponses, 2);
  92 });