Post request support for run test

Author: max-ostapenko

lib/helper.js

@@ -9,7 +9,8 @@ var xml2js   = require('xml2js'),
     url      = require('url'),
     os       = require('os'),
     csv      = require('csv'),
-    entities = require('entities');
+    entities = require('entities')
+    qs       = require('querystring');
 
 var parser = new xml2js.Parser({explicitArray: false, mergeAttrs: true});
 
@@ -155,19 +156,32 @@ function scriptToString(data) {
 }
 
 // Build the RESTful API url call only
-function dryRun(config, path) {
+function dryRun(config, path, form) {
   path = url.parse(path, true);
 
-  return {
-    url: url.format({
-      protocol: config.protocol,
-      hostname: config.hostname,
-      port: (config.port !== 80 && config.port !== 443 ?
-        config.port : undefined),
-      pathname: path.pathname,
-      query: path.query
-    })
-  };
+  if (config.method == "POST") {
+    return {
+      url: url.format({
+        protocol: config.protocol,
+        hostname: config.hostname,
+        port: (config.port !== 80 && config.port !== 443 ?
+          config.port : undefined),
+        pathname: path.pathname,
+      }),
+      form: qs.stringify(form)
+    };
+  } else {
+    return {
+      url: url.format({
+        protocol: config.protocol,
+        hostname: config.hostname,
+        port: (config.port !== 80 && config.port !== 443 ?
+          config.port : undefined),
+        pathname: path.pathname,
+        query: path.query
+      })
+    };
+  }
 }
 
 // Normalize server config

lib/webpagetest.js

@@ -13,7 +13,8 @@ var http = require("http"),
   specs = require("./specs"),
   helper = require("./helper"),
   server = require("./server"),
-  mapping = require("./mapping");
+  mapping = require("./mapping"),
+  qs = require('querystring');
 
 var reSpace = /\s/,
   reConnectivity =
@@ -57,8 +58,8 @@ var filenames = {
   cached: "_Cached",
 };
 
-// GET helper function
-function get(config, pathname, proxy, agent, callback, encoding) {
+// GET/POST helper function
+function get(config, pathname, data, proxy, agent, callback, encoding) {
   var protocol, options;
 
   if (proxy) {
@@ -79,6 +80,7 @@ function get(config, pathname, proxy, agent, callback, encoding) {
       headers: {
         Host: config.hostname,
       },
+      method: config.method
     };
   } else {
     protocol = config.protocol === "https:" ? https : http;
@@ -87,10 +89,15 @@ function get(config, pathname, proxy, agent, callback, encoding) {
       host: config.hostname,
       auth: config.auth,
       port: config.port,
+      method: config.method,
       headers: {},
     };
   }
 
+  if (options.method == "POST") {
+    options.headers['Content-Type'] = 'application/x-www-form-urlencoded';
+  }
+
   if (encoding !== "binary") {
     options.headers["X-WPT-API-KEY"] = this.config.key;
     options.headers["accept-encoding"] = "gzip,deflate";
@@ -101,8 +108,8 @@ function get(config, pathname, proxy, agent, callback, encoding) {
     options.agent = agent;
   }
 
-  return protocol
-    .get(options, function getResponse(res) {
+  var request = protocol
+    .request(options, function getResponse(res) {
       var data,
         length,
         statusCode = res.statusCode;
@@ -159,6 +166,13 @@ function get(config, pathname, proxy, agent, callback, encoding) {
     .on("error", function onError(err) {
       callback(err);
     });
+
+  if (options.method == "POST") {
+    return request.end(qs.stringify(data));
+  } else {
+    return request.end();
+  }
+
 }
 
 // execute callback properly normalizing optional args
@@ -186,22 +200,33 @@ function api(pathname, callback, query, options) {
     config = this.config;
   }
 
-  pathname = url.format({
-    pathname: url.resolve(config.pathname, pathname),
+  pathname = url.resolve(config.pathname, pathname);
+
+  config.method = url.format({
+    pathname: pathname,
     query: query,
-  });
+  }).toString().length > 6 * 1024 ? "POST" : "GET";
+
+  if (config.method == "GET") {
+    pathname = url.format({
+      pathname: pathname,
+      query: query,
+    });
+    query = undefined;
+  }
 
   if (options.dryRun) {
     // dry run: return the API url (string) only
     if (typeof callback === "function") {
-      callback.apply(callback, [undefined, helper.dryRun(config, pathname)]);
+      callback.apply(callback, [undefined, helper.dryRun(config, pathname, query)]);
     }
   } else {
     // make the real API call
     get.call(
       this,
       config,
       pathname,
+      query,
       options.proxy,
       options.agent,
       function apiCallback(err, data, info) {

test/edge-cases-test.js

@@ -85,6 +85,19 @@ describe('Edge Cases of', function() {
       });
     });
 
+    it('gets a test with custom metrics then returns API url and payload with custom metrics data present', function (done) {
+      wpt.runTest('http://foobar.com', {
+        dryRun: true,
+        mobile: 1,
+        custom: '[example]\n\\\\' + 'X'.repeat(6 * 1024) + '\nreturn 1;'
+      }, function (err, data) {
+        if (err) return done(err);
+        assert.equal(data.url, wptServer + 'runtest.php');
+        assert.equal(data.form.length, 6233);
+        done();
+      });
+    });
+
   });
 
   describe('WebPageTest localhost helper', function() {