@artilleryio/int-core 2.1.0-d39abda → 2.1.0-da56158

package/lib/engine_http.js

@@ -167,7 +167,7 @@ HttpEngine.prototype.createScenario = function (scenarioSpec, ee) {
     });
   });
 
-  return self.compile(tasks, scenarioSpec.flow, ee);
+  return self.compile(tasks, scenarioSpec, ee);
 };
 
 HttpEngine.prototype.step = function step(requestSpec, ee, opts) {
@@ -868,6 +868,7 @@ HttpEngine.prototype.compile = function compile(tasks, scenarioSpec, ee) {
 
   return async function scenario(initialContext, callback) {
     initialContext = self.setInitialContext(initialContext);
+
     ee.emit('started');
     let context = initialContext;
     for (const task of tasks) {

package/lib/runner.js

@@ -200,7 +200,13 @@ function run(script, ee, options, runState, contextVars) {
     if (runState.pendingScenarios >= spec.maxVusers) {
       metrics.counter('vusers.skipped', 1);
     } else {
-      scenarioContext = runScenario(script, metrics, runState, contextVars);
+      scenarioContext = runScenario(
+        script,
+        metrics,
+        runState,
+        contextVars,
+        options
+      );
     }
   });
   phaser.on('phaseStarted', function (spec) {
@@ -242,7 +248,7 @@ function run(script, ee, options, runState, contextVars) {
   phaser.run();
 }
 
-function runScenario(script, metrics, runState, contextVars) {
+function runScenario(script, metrics, runState, contextVars, options) {
   const start = process.hrtime();
 
   //
@@ -263,7 +269,7 @@ function runScenario(script, metrics, runState, contextVars) {
         const w = engineUtil.template(scenario.weight, {
           vars: variableValues
         });
-        scenario.weight = isNaN(parseInt(w)) ? 0 : parseInt(w);
+        scenario.weight = isNaN(parseInt(w)) ? 0 : parseInt(w); //eslint-disable-line radix
         debug(
           `scenario ${scenario.name} weight has been set to ${scenario.weight}`
         );
@@ -299,16 +305,45 @@ function runScenario(script, metrics, runState, contextVars) {
 
     runState.compiledScenarios = _.map(
       script.scenarios,
-      function (scenarioSpec) {
+      function (scenarioSpec, scenarioIndex) {
         const name = scenarioSpec.engine || script.config.engine || 'http';
         const engine = runState.engines.find((e) => e.__name === name);
+
+        if (typeof engine === 'undefined') {
+          const scenarioNameOrIndex = scenarioSpec.name || scenarioIndex;
+          throw new Error(
+            `Failed to run scenario "${scenarioNameOrIndex}": unknown engine "${name}". Did you forget to include it in "config.engines.${name}"?`
+          );
+        }
+
         return engine.createScenario(scenarioSpec, runState.scenarioEvents);
       }
     );
   }
 
+  //default to weighted picked scenario
   let i = runState.picker()[0];
 
+  if (options.scenarioName) {
+    let foundIndex;
+    const foundScenario = script.scenarios.filter((scenario, index) => {
+      foundIndex = index;
+      return new RegExp(options.scenarioName).test(scenario.name);
+    });
+
+    if (foundScenario?.length === 0) {
+      throw new Error(
+        `Scenario ${options.scenarioName} not found in script. Make sure your chosen scenario matches the one in your script exactly.`
+      );
+    } else if (foundScenario.length > 1) {
+      throw new Error(
+        `Multiple scenarios for ${options.scenarioName} found in script. Make sure you give unique names to your scenarios in your script.`
+      );
+    } else {
+      debug(`Scenario ${options.scenarioName} found in script. running it!`);
+      i = foundIndex;
+    }
+  }
   debug(
     'picking scenario %s (%s) weight = %s',
     i,
@@ -320,7 +355,9 @@ function runScenario(script, metrics, runState, contextVars) {
   metrics.counter('vusers.created', 1);
 
   const scenarioStartedAt = process.hrtime();
-  const scenarioContext = createContext(script, contextVars);
+  const scenarioContext = createContext(script, contextVars, {
+    scenario: script.scenarios[i]
+  });
 
   const finish = process.hrtime(start);
   const runScenarioDelta = finish[0] * 1e9 + finish[1];
@@ -383,7 +420,13 @@ function inlineVariables(script) {
 /**
  * Create initial context for a scenario.
  */
-function createContext(script, contextVars) {
+function createContext(script, contextVars, additionalProperties = {}) {
+  //allow for additional properties to be passed in, but not override vars and funcs
+  const additionalPropertiesWithoutOverride = _.omit(additionalProperties, [
+    'vars',
+    'funcs'
+  ]);
+
   const INITIAL_CONTEXT = {
     vars: Object.assign(
       {
@@ -398,7 +441,8 @@ function createContext(script, contextVars) {
       $randomNumber: $randomNumber,
       $randomString: $randomString,
       $template: (input) => engineUtil.template(input, { vars: result.vars })
-    }
+    },
+    ...additionalPropertiesWithoutOverride
   };
 
   let result = INITIAL_CONTEXT;
@@ -445,7 +489,10 @@ function handleScriptHook(hook, script, hookEvents, contextVars = {}) {
     const name = script[hook].engine || 'http';
     const engine = engines.find((e) => e.__name === name);
     const hookScenario = engine.createScenario(script[hook], ee);
-    const hookContext = createContext(script, contextVars);
+    const hookContext = createContext(script, contextVars, {
+      scenario: script[hook]
+    });
+
     hookScenario(hookContext, function (err, context) {
       if (err) {
         debug(err);

package/lib/ssms.js

@@ -626,6 +626,7 @@ function summarizeHistogram(h) {
     min: round(h.min, 1),
     max: round(h.max, 1),
     count: h.count,
+    mean: round(h.sum/h.count, 1),
     p50: round(h.getValueAtQuantile(0.5), 1),
     median: round(h.getValueAtQuantile(0.5), 1), // Here for compatibility
     p75: round(h.getValueAtQuantile(0.75), 1),

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "@artilleryio/int-core",
-  "version": "2.1.0-d39abda",
+  "version": "2.1.0-da56158",
   "main": "./index.js",
+  "license": "MPL-2.0",
   "dependencies": {
-    "@artilleryio/int-commons": "2.0.1-d39abda",
+    "@artilleryio/int-commons": "2.0.1-da56158",
     "@artilleryio/sketches-js": "^2.1.1",
     "agentkeepalive": "^4.1.0",
     "arrivals": "^2.1.2",