Support requests via the streambed API

scjody · scjody · commit f1b58e5ba14d · 2018-02-13T13:32:04.000-05:00
diff --git a/test/image/load_test b/test/image/load_test
@@ -18,6 +18,10 @@ const REQ_TIMEOUT = 40000;
 const VERBOSE = true;
 // Randomize test ordering (otherwise glob sorting is used):
 const RANDOMIZE = false;
+// Specify a username and API key to make requests via the streambed v2 API,
+// otherwise set falsy to make a direct request.
+const PLOTLY_USERNAME='._-aA1'
+const PLOTLY_API_KEY='testapikey'
 
 /**
  *  Load Tester for imageservers
@@ -170,7 +174,21 @@ function runReq(req, completion) {
         }
     }
 
-    request.post({uri: endpointUrl, body: body, timeout: REQ_TIMEOUT})
+    var postArgs = {uri: endpointUrl, body: body, timeout: REQ_TIMEOUT}
+
+    if (PLOTLY_USERNAME) {
+        postArgs.auth = {
+          user: PLOTLY_USERNAME,
+          pass: PLOTLY_API_KEY,
+          sendImmediately: true
+        }
+        postArgs.headers = {
+          'Content-Type': 'application/json',
+          'Plotly-Client-Platform': 'Load Test'
+        }
+    }
+
+    request.post(postArgs)
         .on('error', onError)
         .on('response', onResponse);
 }
