feat(artillery): add ability to specify scenario to run by name

packages/artillery/lib/cmds/run.js

@@ -155,6 +155,10 @@ RunCommand.flags = {
   }),
   key: Flags.string({
     description: 'API key for Artillery Cloud'
+  }),
+  name: Flags.string({
+    description: 'Name of the test to run',
+    char: 'n'
   })
 };
 
@@ -207,7 +211,8 @@ RunCommand.runCommandImplementation = async function (flags, argv, args) {
       scriptPath: args.script,
       // TODO: This should be an array of files, like inputFiles above
       absoluteScriptPath: path.resolve(process.cwd(), args.script),
-      plugins: []
+      plugins: [],
+      scenarioName: flags.name
     };
 
     // Set "name" tag if not set explicitly

packages/artillery/test/cli/command-run.test.js

@@ -88,6 +88,32 @@ tap.test('Environment specified with -e should be used', async (t) => {
   t.ok(exitCode === 0 && !output.stdout.includes('ECONNREFUSED'));
 });
 
+tap.test('Can specify scenario to run by name', async (t) => {
+  const reportFile = 'report-with-scenario-by-name.json';
+  const reportFilePath = await getRootPath(reportFile);
+
+  const [exitCode, output] = await execute([
+    'run',
+    '-n',
+    'Test Scenario 2',
+    '-o',
+    `${reportFilePath}`,
+    'test/scripts/scenario-named/scenario.yml'
+  ]);
+  console.log(output.stdout);
+
+  // Here if the right environment is not picked up, we'll get ECONNREFUSED errors in the report
+  t.ok(
+    exitCode === 0 && output.stdout.includes('Successfully running scenario 2')
+  );
+  const json = JSON.parse(fs.readFileSync(reportFilePath, 'utf8'));
+
+  t.ok(
+    deleteFile(reportFilePath) &&
+      json.aggregate.counters['vusers.created_by_name.Test Scenario 2'] === 6
+  );
+});
+
 tap.test('Run a script with one payload command line', async (t) => {
   const [, output] = await execute([
     'run',

packages/artillery/test/scripts/scenario-named/scenario.yml

@@ -0,0 +1,13 @@
+config:
+  target: http://asciiart.artillery.io:8080
+  phases:
+    - duration: 3
+      arrivalRate: 2
+
+scenarios:
+  - name: Test Scenario 1
+    flow:
+      - log: "Successfully running scenario 1"
+  - name: Test Scenario 2
+    flow:
+      - log: "Successfully running scenario 2"

packages/core/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}`
         );
@@ -307,8 +313,29 @@ function runScenario(script, metrics, runState, contextVars) {
     );
   }
 
+  //default to randomly 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) {
+      debug(
+        `scenario ${options.scenarioName} not found in script. Choosing random scenario instead.`
+      );
+    } else if (foundScenario.length > 1) {
+      debug(
+        `multiple scenarios found for ${options.scenarioName}. Choosing random scenario instead.`
+      );
+    } else {
+      debug(`scenario ${options.scenarioName} found in script. running it!`);
+      i = foundIndex;
+    }
+  }
   debug(
     'picking scenario %s (%s) weight = %s',
     i,