connect_test.js

'use strict';

var grunt = require('grunt');
var http = require('http');
var https = require('https');
var http2 = require('node-http2');

function get(url, done) {
  var client = http;
  if (typeof url === 'object' && url.scheme === 'http2') {
    client = http2;
    delete url.scheme;
  }
  else if (typeof url === 'string' && url.toLowerCase().indexOf('https') === 0 ||
    typeof url === 'object' && url.port === 443 ||
    typeof url === 'object' && url.scheme === 'https') {
    client = https;
    delete url.scheme;
  }
  client.get(url, function(res) {
    var body = '';
    res.on('data', function(chunk) {
      body += chunk;
    }).on('end', function() {
      done(res, body);
    });
  });
}

exports.connect = {
  custom_base: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8000,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  custom_port: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8001,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  custom_http2: function(test) {
    if (!grunt.config.data.connect.http2) {
      test.done();
    }
    else {
      test.expect(3);
      get({
        scheme: 'http2',
        rejectUnauthorized: false,
        hostname: 'localhost',
        port: 8017,
        path: '/fixtures/hello.txt',
        headers: {
          accept: 'text/plain'
        }
      }, function(res, body) {
        test.equal(res.httpVersion, '2.0', 'should return HTTP/2 response');
        test.equal(res.statusCode, 200, 'should return 200');
        test.equal(body, 'Hello world', 'should return static page');
        test.done();
      });
    }
  },
  custom_https: function(test) {
    test.expect(2);
    get({
      scheme: 'https',
      /* per http://nodejs.org/api/https.html#https_https_request_options_callback
         allow certs signed by an untrusted CA */
      rejectUnauthorized: false,
      hostname: 'localhost',
      port: 8002,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  custom_https_certs: function(test) {
    test.expect(2);
    get({
      scheme: 'https',
      rejectUnauthorized: false,
      hostname: 'localhost',
      port: 8003,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  custom_base_with_options: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8014,
      path: '/',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  multiple_base: function(test) {
    test.expect(4);
    get({
      hostname: 'localhost',
      port: 8004,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(res.headers['content-type'], 'text/plain; charset=UTF-8', 'should return plaintext content type');
      get('http://localhost:8004/connect-examples.md', function(res, body) {
        test.equal(res.headers['content-type'], 'text/markdown; charset=UTF-8', 'should return markdown content type');
        test.equal(res.statusCode, 200, 'should return 200');
        test.done();
      });
    });
  },
  multiple_base_with_options: function(test) {
    test.expect(5);
    get({
      hostname: 'localhost',
      port: 8015,
      path: '/',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.ok(res.headers['cache-control'].indexOf('max-age=0') > 0);
      get('http://localhost:8015/connect-examples.md', function(res, body) {
        test.ok(res.headers['cache-control'].indexOf('max-age=300') > 0);
        test.equal(res.statusCode, 200, 'should return 200');
        test.done();
      });
    });
  },
  multiple_base_directory: function(test) {
    test.expect(4);
    get({
      hostname: 'localhost',
      port: 8005,
      path: '/',
      headers: {
        accept: 'text/html'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.ok(body.indexOf('hello.txt') !== -1, 'Listing should contain hello.txt');
      get('http://localhost:8005/fixtures/hello.txt', function(res, body) {
        test.equal(res.statusCode, 200, 'should return 200');
        test.equal(body, 'Hello world', 'Should display contents of /fixtures/hello.txt');
        test.done();
      });
    });
  },
  livereload: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8006,
      path: '/livereload.html',
      headers: {
        accept: 'text/html'
      }
    }, function(res, body) {
      test.ok(body.indexOf('35729/livereload.js') !== -1, 'Should contain livereload snippet.');

      // check if livereload works with params
      get({
        hostname: 'localhost',
        port: 8006,
        path: '/livereload.html?a=1&b=2#id',
        headers: {
          accept: 'text/html'
        }
      }, function(res, body) {
        test.ok(body.indexOf('35729/livereload.js') !== -1, 'Should contain livereload snippet.');
        test.done();
      });
    });
  },
  livereload_port: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8018,
      path: '/livereload.html',
      headers: {
        accept: 'text/html'
      }
    }, function(res, body) {
      test.ok(body.indexOf('12345/livereload.js') !== -1, 'Should contain livereload snippet with configured port.');

      // check if livereload works with params
      get({
        hostname: 'localhost',
        port: 8018,
        path: '/livereload.html?a=1&b=2#id',
        headers: {
          accept: 'text/html'
        }
      }, function(res, body) {
        test.ok(body.indexOf('12345/livereload.js') !== -1, 'Should contain livereload snippet with configured port.');
        test.done();
      });
    });
  },
  livereload_general: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8019,
      path: '/livereload.html',
      headers: {
        accept: 'text/html'
      }
    }, function(res, body) {
      test.ok(body.indexOf('https://example.org:54321/livereload.js?snipver=1') !== -1, 'Should contain livereload snippet with configured src.');

      // check if livereload works with params
      get({
        hostname: 'localhost',
        port: 8019,
        path: '/livereload.html?a=1&b=2#id',
        headers: {
          accept: 'text/html'
        }
      }, function(res, body) {
        test.ok(body.indexOf('https://example.org:54321/livereload.js?snipver=1') !== -1, 'Should contain livereload snippet with configured src.');
        test.done();
      });
    });
  },
  custom_middleware: function(test) {
    var PORT = 8007;
    test.expect(4);
    get({
      hostname: 'localhost',
      port: PORT,
      path: '/hello/world',
      headers: {
        accept: 'text/plain'
      },
      middleware: []
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello from port ' + PORT, 'should return a dynamic response');
      get({
        hostname: 'localhost',
        port: PORT,
        path: '/fixtures/hello.txt',
        headers: {
          accept: 'text/plain'
        }
      }, function(res, body) {
        test.equal(res.statusCode, 200, 'should return 200');
        test.equal(body, 'Hello world', 'should return static page');
        test.done();
      });
    });
  },
  null_middleware_should_use_default_middleware: function(test) {
    test.expect(2);
    get({
      hostname: 'localhost',
      port: 8008,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello world', 'should return static page');
      test.done();
    });
  },
  empty_middleware_should_404_everything: function(test) {
    test.expect(1);
    get({
      hostname: 'localhost',
      port: 8009,
      path: '/fixtures/hello.txt',
      headers: {
        accept: 'text/plain'
      }
    }, function(res, body) {
      test.equal(res.statusCode, 404, 'should return 404');
      test.done();
    });
  },
  custom_middleware_can_patch_default_middleware: function(test) {
    var PORT = 8010;
    test.expect(4);
    get({
      hostname: 'localhost',
      port: PORT,
      path: '/hello/world',
      headers: {
        accept: 'text/plain'
      },
      middleware: []
    }, function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      test.equal(body, 'Hello, world from port #' + PORT + '!', 'should return a dynamic response');

      get({
        hostname: 'localhost',
        port: PORT,
        path: '/fixtures/hello.txt',
        headers: {
          accept: 'text/plain'
        }
      }, function(res, body) {
        test.equal(res.statusCode, 200, 'should return 200');
        test.equal(body, 'Hello world', 'should return static page');
        test.done();
      });
    });
  },
  allHostname: function(test) {
    test.expect(3);

    get('http://localhost:8012/fixtures/hello.txt', function(res, body) {
      test.equal(res.statusCode, 200, 'should return 200');
      get('http://127.0.0.1:8012/fixtures/hello.txt', function(res, body) {
        test.equal(res.statusCode, 200, 'should return 200');
        if (process.platform === 'win32') {
          // there is a discrepancy with hostnames. Skip the test until further changes.
          test.ok(true);
          test.done();
        } else {
          get('http://0.0.0.0:8012/fixtures/hello.txt', function(res, body) {
            test.equal(res.statusCode, 200, 'should return 200');
            test.done();
          });
        }
      });
    });
  },
  onCreateServer: function(test) {
    test.expect(1);

    get('http://localhost:8013/hello', function(res, body) {
      test.ok(grunt.config.data.connect.onCreateServer.test, 'should set configuration object on request');
      test.done();
    });
  },
  routedMiddleware: function(test) {
    test.expect(1);

    get('http://localhost:8016/mung', function(res, body) {
      test.equal(body, 'Yay', 'should return a string at /mung');
      test.done();
    });
  }
};