From 7f1d779676e8c2c414b9c12b0f55ab9da13d2ed8 Mon Sep 17 00:00:00 2001
From: Bernardo Guerreiro <39738771+bernardobridge@users.noreply.github.com>
Date: Tue, 2 Jan 2024 16:22:55 +0000
Subject: [PATCH] feat(cli): expose --variables flag in fargate (#2391)

* feat(cli): expose --variables flag in fargate

* refactor(cli): move --variables to common flags
---
 packages/artillery/lib/cli/common-flags.js                   | 5 +++++
 packages/artillery/lib/cmds/run-lambda.js                    | 5 -----
 packages/artillery/lib/cmds/run.js                           | 5 -----
 .../artillery/lib/platform/aws-ecs/legacy/run-cluster.js     | 3 +++
 4 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/packages/artillery/lib/cli/common-flags.js b/packages/artillery/lib/cli/common-flags.js
index 226f979a15..b589a7d1ca 100644
--- a/packages/artillery/lib/cli/common-flags.js
+++ b/packages/artillery/lib/cli/common-flags.js
@@ -25,6 +25,11 @@ const CommonRunFlags = {
   dotenv: Flags.string({
     description: 'Path to a dotenv file to load environment variables from'
   }),
+  variables: Flags.string({
+    char: 'v',
+    description:
+      'Set variables available to vusers during the test; a JSON object'
+  }),
   overrides: Flags.string({
     description: 'Dynamically override values in the test script; a JSON object'
   }),
diff --git a/packages/artillery/lib/cmds/run-lambda.js b/packages/artillery/lib/cmds/run-lambda.js
index 1a74f49d5c..5bfc7b4d51 100644
--- a/packages/artillery/lib/cmds/run-lambda.js
+++ b/packages/artillery/lib/cmds/run-lambda.js
@@ -52,11 +52,6 @@ Examples:
 `;
 RunLambdaCommand.flags = {
   ...CommonRunFlags,
-  variables: Flags.string({
-    char: 'v',
-    description:
-      'Set variables available to vusers during the test; a JSON object'
-  }),
   payload: Flags.string({
     char: 'p',
     description: 'Specify a CSV file for dynamic data'
diff --git a/packages/artillery/lib/cmds/run.js b/packages/artillery/lib/cmds/run.js
index 87cda4df85..e0fff882fa 100644
--- a/packages/artillery/lib/cmds/run.js
+++ b/packages/artillery/lib/cmds/run.js
@@ -80,11 +80,6 @@ RunCommand.flags = {
     char: 'q',
     description: 'Quiet mode'
   }),
-  variables: Flags.string({
-    char: 'v',
-    description:
-      'Set variables available to vusers during the test; a JSON object'
-  }),
   // TODO: Deprecation notices for commands below:
   payload: Flags.string({
     char: 'p',
diff --git a/packages/artillery/lib/platform/aws-ecs/legacy/run-cluster.js b/packages/artillery/lib/platform/aws-ecs/legacy/run-cluster.js
index 8b415912da..217eaffbe7 100644
--- a/packages/artillery/lib/platform/aws-ecs/legacy/run-cluster.js
+++ b/packages/artillery/lib/platform/aws-ecs/legacy/run-cluster.js
@@ -1365,6 +1365,9 @@ async function generateTaskOverrides(context) {
     context.cliOptions.overrides
       ? ['--overrides', context.cliOptions.overrides]
       : [],
+    context.cliOptions.variables
+      ? ['--variables', context.cliOptions.variables]
+      : [],
     context.configPath ? ['--config', context.configPath] : []
   );
   // NOTE: This MUST come last: