From 3167fabf8cd6a397668400f4283718451f435790 Mon Sep 17 00:00:00 2001
From: Karan Raina <karanraina1996@gmail.com>
Date: Mon, 14 Oct 2024 22:28:52 +0530
Subject: [PATCH] assume onPort if -- node is set (#530)

* assume onPort if -- node is set

* doc: onPort assumed to be true is node script is provided

* dropped support for node v14.x and v16.x

---------

Co-authored-by: Matteo Collina <hello@matteocollina.com>
---
 README.md           |  1 +
 autocannon.js       |  5 +++++
 test/onPort.test.js | 54 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 60 insertions(+)

diff --git a/README.md b/README.md
index dac38ed0..144f5e6e 100644
--- a/README.md
+++ b/README.md
@@ -62,6 +62,7 @@ Available options:
         Start the command listed after -- on the command line. When it starts listening on a port,
         start sending requests to that port. A URL is still required to send requests to
         the correct path. The hostname can be omitted, `localhost` will be used by default.
+        If the command after -- is `node <script>`, this flag is optional and assumed to be `true`.
   -m/--method METHOD
         The HTTP method to use. default: 'GET'.
   -t/--timeout NUM
diff --git a/autocannon.js b/autocannon.js
index 8d67285a..d50078ce 100755
--- a/autocannon.js
+++ b/autocannon.js
@@ -115,6 +115,11 @@ function parseArguments (argvs) {
 
   argv.url = argv._.length > 1 ? argv._ : argv._[0]
 
+  // Assume onPort if `-- node` is provided
+  if (argv['--'][0] === 'node') {
+    argv.onPort = true
+  }
+
   if (argv.onPort) {
     argv.spawn = argv['--']
   }
diff --git a/test/onPort.test.js b/test/onPort.test.js
index 86a7222a..d3b86f1f 100644
--- a/test/onPort.test.js
+++ b/test/onPort.test.js
@@ -59,3 +59,57 @@ test('--on-port flag', { skip: !hasAsyncHooks() }, (t) => {
       t.ok(regexp.test(line), 'line matches ' + regexp)
     })
 })
+
+test('assume --on-port flag if -- node is set', { skip: !hasAsyncHooks() }, (t) => {
+  const lines = [
+    /Running 1s test @ .*$/,
+    /10 connections.*$/,
+    /$/,
+    /.*/,
+    /$/,
+    /Stat.*2\.5%.*50%.*97\.5%.*99%.*Avg.*Stdev.*Max.*$/,
+    /.*/,
+    /Latency.*$/,
+    /$/,
+    /.*/,
+    /Stat.*1%.*2\.5%.*50%.*97\.5%.*Avg.*Stdev.*Min.*$/,
+    /.*/,
+    /Req\/Sec.*$/,
+    /$/,
+    /Bytes\/Sec.*$/,
+    /.*/,
+    /$/,
+    /Req\/Bytes counts sampled once per second.*$/,
+    /# of samples: 10*$/,
+    /$/,
+    /.* requests in ([0-9]|\.)+s, .* read/
+  ]
+
+  t.plan(lines.length * 2)
+
+  const child = spawn(process.execPath, [
+    path.join(__dirname, '..'),
+    '-c', '10',
+    '-d', '1',
+    '/',
+    '--', 'node', path.join(__dirname, './targetProcess')
+  ], {
+    cwd: __dirname,
+    env: process.env,
+    stdio: ['ignore', 'pipe', 'pipe'],
+    detached: false
+  })
+
+  t.teardown(() => {
+    child.kill()
+  })
+
+  child
+    .stderr
+    .pipe(split())
+    .on('data', (line) => {
+      const regexp = lines.shift()
+      t.ok(regexp, 'we are expecting this line')
+      t.ok(regexp.test(line), 'line matches ' + regexp + `actual: ${line}  expected: ${regexp}`)
+    })
+})