Merge pull request #2 from godfreydoo/get-test-data-auto

Test data is now automatically retrieved and ran

Author: godfreydoo

README.md

@@ -1,6 +1,6 @@
 # retrieve-loaderio-test-data
 
-This repo helps you perform your load testing on [Loader.io](https://loader.io/) by
+This application helps you perform your load testing on [Loader.io](https://loader.io/) by
 1. sequentially executing prepared tests with the same duration,
 2. extracting test results from current run, and
 3. writing to a CSV file for further analysis.
@@ -9,7 +9,6 @@ This repo helps you perform your load testing on [Loader.io](https://loader.io/)
 ## Limitations
 - assumes all tests have the same duration
 - includes a buffer time between running the next test (you could change the core functions to check for the status before running subsequent tests)
-- test names must be unique in the `testSuite` object
 
 ## Installation
 1. Fork and clone this repository.
@@ -21,15 +20,8 @@ $ npm install
 4. Update `index.js` file with your test configurations.
 - `duration` is the duration of your test cases.
 - `buffer` is a buffer time between when the next test case runs. This is important, as Loader.io does not allow for more than one running test at once.
-- `testSuite` is an object with the key as the name of your test, and value as your test id. Test names must be unique.
-
-To find the test id for your tests, you can run this command. Replace `API_KEY` with your API key. 
-```bash
-curl -H 'loaderio-auth: API_KEY' https://api.loader.io/v2/tests | json_pp
-```
-
 5. Ensure your application is running as Loader.io will need to verify your token.
-6. Run to start tests
+6. Run to start tests. Application will retrieve all your tests for you so you don't have to input your test name and test id.
 ```
 node index.js
 ```

index.js

@@ -5,15 +5,10 @@ require('dotenv').config();
 
 
 
-// Define your test ids, test names, and test parameters
-var duration = 30000; // in ms, ie. 30 seconds
-var buffer = 5000; // in ms, i.e. 5 second buffer (I found this to be a good buffer)
-var url = 'https://api.loader.io/v2/tests'
-var testSuite = {
-  '<your test name>': '<your test id>',
-  '<your test name>': '<your test id>'
-};
+// Define your test duration
 
+var duration = 30000; // in ms, ie. 60 seconds
+var buffer = 5000; // in ms, i.e. 5 second buffer (I found this to be a good buffer)
 
 
 
@@ -22,65 +17,88 @@ var testSuite = {
 
 
 
+// Core functions -- do not change anything below!
 
+var url = 'https://api.loader.io/v2/tests'
 
-// Core functions -- do not change anything below!
-function LoaderTests() {
+function LoaderIOTests() {
+  this._tests = {};
   this._testResults = [];
-  this._testSuite = [];
 }
 
-LoaderTests.prototype.runTest = function (testName, testidx){
-  console.log(`${testName} test is running...`);
-  return new Promise((resolve, reject) => {
-    const config = {
-      method: 'put',
-      url: `${url}/${testidx}/run`,
-      headers: {
-        'loaderio-auth': process.env.LOADER_API_KEY
-      }
-    };
-    axios(config)
-      .then(response => {
-        const obj =
-        this._testSuite.push({
-          testName: testName,
-          testidx: testidx,
-          resultidx: response.data.result_id
-        });
-        resolve();
-      })
-      .catch(error => {
-        console.log(`Something went wrong with running testidx: ${testidx}`);
-        console.log(error);
-        reject();
-      })
-  })
+LoaderIOTests.prototype.getTests = async function() {
+  const config = {
+    url: url,
+    headers: { 'loaderio-auth': process.env.LOADER_API_KEY }
+  }
+  try {
+    let { data } = await axios(config);
+    let extractedData = this.extractNameAndTotal(data);
+    this._tests = extractedData;
+  } catch (error) {
+    console.log(`Something went wrong with fetching test results for resultidx: ${resultidx}`);
+    console.error(error);
+  }
+}
+
+
+LoaderIOTests.prototype.extractNameAndTotal = function(testData) {
+  var results = {};
+  for (var i = 0; i < testData.length; i++) {
+    var testid = testData[i].test_id;
+    var details = {};
+    details.name = testData[i].name;
+    details.total_clients = testData[i].total;
+    details.duration = testData[i].duration;
+    details.testidx = testid;
+
+    results[testid] = details;
+  }
+  return results;
+}
+
+LoaderIOTests.prototype.runTestAndSaveResultId = async function(testidx, name) {
+  console.log(`${name} is running...`);
+
+  const config = {
+    method: 'put',
+    url: `${url}/${testidx}/run`,
+    headers: { 'loaderio-auth': process.env.LOADER_API_KEY }
+  };
+
+  try {
+    let { data } = await axios(config);
+    this._tests[testidx].resultidx = data.result_id;
+  } catch (error) {
+    if (error.response.status === 422) {
+      console.log('Error 422 received. Try increasing the buffer time between tests. This error is most likely due to running more than one test at once or could be some lag between requests and responses; however, the script will continue to run.');
+    } else {
+      console.log(`Something went wrong with running testidx: ${testidx}`);
+      console.error(error);
+    }
+  }
 }
 
-LoaderTests.prototype.getTestResults = async function (testName, testidx, resultidx, datetime){
-  var testName = testName;
-  var resultidx = resultidx;
+LoaderIOTests.prototype.getTestResults = async function(testidx, resultidx, name, duration, totalClients) {
   const config = {
     url: `${url}/${testidx}/results/${resultidx}`,
     headers: {
       'loaderio-auth': process.env.LOADER_API_KEY
     }
   }
-
-  await axios(config)
-    .then(response => {
-      response.data['test_name'] = testName;
-      this.write2CSV(datetime);
-      this._testResults.push(response.data);
-    })
-    .catch(error => {
-      console.log(`Something went wrong with fetching test results for resultidx: ${resultidx}`);
-      console.log(error);
-    });
+  try {
+    let { data } = await axios(config);
+    data.name = name;
+    data.duration = duration;
+    data.total_clients = totalClients;
+    this._testResults.push(data);
+  } catch (error) {
+    console.log(`Something went wrong with fetching test results for resultidx: ${resultidx}`);
+    console.error(error);
+  }
 }
 
-LoaderTests.prototype.write2CSV = async function (datetime) {
+LoaderIOTests.prototype.write2CSV = async function(datetime) {
   const writeCSVStream = fs.createWriteStream(`results_${datetime}.csv`);
   await converter.json2csv(this._testResults, (err, csv) => {
     if (err) {
@@ -91,27 +109,31 @@ LoaderTests.prototype.write2CSV = async function (datetime) {
   })
 }
 
-function sleep(ms) {
+function sleep(fn) {
   return new Promise(resolve => setTimeout(resolve, duration + buffer));
 }
 
 async function run() {
-  var newRun = new LoaderTests;
+  var newRun = new LoaderIOTests;
   try {
+    console.log('Retrieving your test data...');
+    await newRun.getTests();
     console.log('Starting to run tests...');
-    for (var key in testSuite) {
-      await sleep(newRun.runTest(key, testSuite[key]))
+    for (var key in newRun._tests) {
+      await sleep(newRun.runTestAndSaveResultId(key, newRun._tests[key].name));
     }
     console.log('All tests ran...');
     console.log('Fetching data and writing to CSV file...');
-    const datetime = Date.now();
-    await newRun._testSuite.forEach(value => {
-       newRun.getTestResults(value.testName, value.testidx, value.resultidx, datetime)
-    });
+    for (var key in newRun._tests) {
+      var test = newRun._tests[key];
+      await newRun.getTestResults(key, test.resultidx, test.name, test.duration, test.total_clients)
+    }
+    const datetime = new Date();
+    newRun.write2CSV(datetime);
     console.log('results.csv file should be created / updated soon');
-  } catch (err) {
-    console.log('Double check testSuite has right data, and that the duration + buffer is accurate for your test settings');
-    console.log(err);
+  } catch (error) {
+    console.log('Double check the duration + buffer is accurate for your test settings');
+    console.error(error);
   }
 }