feat: add --no-assign-public-ip flag for Fargate (#3643)

Author: hassy

packages/artillery/lib/cmds/run-fargate.js

@@ -177,6 +177,11 @@ RunCommand.flags = {
   }),
   'max-duration': Flags.string({
     description: 'Maximum duration of the test run'
+  }),
+  'no-assign-public-ip': Flags.boolean({
+    description:
+      'Turn off the default behavior of assigning public IPs to Fargate worker tasks. When this option is used you must make sure tasks have a route to the internet, i.e. via a NAT gateway attached to a private subnet',
+    default: false
   })
 };
 

packages/artillery/lib/platform/aws-ecs/legacy/run-cluster.js

@@ -1600,6 +1600,10 @@ async function setupDefaultECSParams(context) {
         subnets: context.fargatePublicSubnetIds
       }
     };
+
+    if (context.cliOptions.noAssignPublicIp) {
+      defaultParams.awsvpcConfiguration.assignPublicIp = 'DISABLED';
+    }
   } else {
     defaultParams.launchType = 'EC2';
   }