Improving the average function reusability for stress tests

Author: feliun

stress-test/stress-test.js

@@ -11,27 +11,28 @@ try {
     worksmith.use("lodash", worksmith.createAdapter(require('lodash')))
     var workflow = worksmith("./stress-test/stress-workflows/workflow1.js")
     console.log('About to execute workflow1')
-    calculateAverageExecTime(function(average) {
+    calculateAverageExecTime(workflow, function(average) {
         console.log('The average execution time is: ', average, ' ms')
     })
 } catch(e) {
     console.error('There was an error setting up a workflow: ', e)
 }
 
-function calculateAverageExecTime(next) {
+function calculateAverageExecTime(workflow, next) {
     var average = counter = 0
     async.whilst(
         function () { return counter < SAMPLE_SIZE },
         function (cb) {
             counter++;
             var ctx = {}
             workflow(ctx, function(err, results) {
-                if (err) console.error('There was an error running a workflow: ', err.toString())
+                if (err) return cb(err)
                 average += ctx.duration
                 cb()
             })
         },
         function (err) {
+            if (err) console.error('There was an error running a workflow: ', err.toString())
             next(average / SAMPLE_SIZE)
         }
     )