Support http2.

.travis.yml

@@ -3,3 +3,10 @@ node_js:
   - 6
   - 8
   - 9
+
+matrix:
+  include:
+    - node_js: "9"
+      env: HTTP2_TEST=1
+    - node_js: "8"
+      env: HTTP2_TEST=1

index.js

@@ -4,6 +4,12 @@
 var methods = require('methods');
 var Test = require('./lib/test');
 var http = require('http');
+var http2;
+try {
+  http2 = require('http2'); // eslint-disable-line global-require
+} catch (_) {
+  // eslint-disable-line no-empty
+}
 
 /**
  * Test against the given `app`,
@@ -17,12 +23,20 @@ module.exports = function(app) {
   var obj = {};
 
   if (typeof app === 'function') {
-    app = http.createServer(app); // eslint-disable-line no-param-reassign
+    if (module.exports.http2) {
+      app = http2.createServer(app); // eslint-disable-line no-param-reassign
+    } else {
+      app = http.createServer(app); // eslint-disable-line no-param-reassign
+    }
   }
 
   methods.forEach(function(method) {
     obj[method] = function(url) {
-      return new Test(app, method, url);
+      var test = new Test(app, method, url);
+      if (module.exports.http2) {
+        test.http2();
+      }
+      return test;
     };
   });
 
@@ -37,6 +51,11 @@ module.exports = function(app) {
  */
 module.exports.Test = Test;
 
+/**
+ * Expose enable http2 property
+ */
+module.exports.http2 = false;
+
 /**
  * Expose the agent function
  */

lib/test.js

@@ -5,8 +5,8 @@
 var request = require('superagent');
 var util = require('util');
 var http = require('http');
-var https = require('https');
 var assert = require('assert');
+var tls = require('tls');
 var Request = request.Request;
 
 /**
@@ -58,7 +58,8 @@ Test.prototype.serverAddress = function(app, path, host) {
 
   if (!addr) this._server = app.listen(0);
   port = app.address().port;
-  protocol = app instanceof https.Server ? 'https' : 'http';
+
+  protocol = app instanceof tls.Server ? 'https' : 'http';
   return protocol + '://' + (host || '127.0.0.1') + ':' + port + path;
 };