fix: set appropriate status when test run is cancelled

artilleryio · Jul 19, 2023 · 48a4a91 · 48a4a91
1 parent 77f6423
commit 48a4a91
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/packages/artillery/lib/cmds/run.js b/packages/artillery/lib/cmds/run.js
@@ -360,8 +360,12 @@ RunCommand.runCommandImplementation = async function (flags, argv, args) {
 
     let finalReport = {};
     let shuttingDown = false;
-    process.once('SIGINT', gracefulShutdown);
-    process.once('SIGTERM', gracefulShutdown);
+    process.on('SIGINT', async () => {
+      gracefulShutdown({ earlyStop: true });
+    });
+    process.on('SIGTERM', async () => {
+      gracefulShutdown({ earlyStop: true });
+    });
 
     async function gracefulShutdown(opts = { exitCode: 0 }) {
       debug('shutting down 🦑');

diff --git a/packages/artillery/lib/platform/cloud/cloud.js b/packages/artillery/lib/platform/cloud/cloud.js
@@ -107,9 +107,11 @@ class ArtilleryCloudPlugin {
     // output has been captured and sent to the dashboard.
     global.artillery.ext({
       ext: 'onShutdown',
-      method: async () => {
+      method: async (opts) => {
         await this._event('testrun:end', { ts: testEndInfo.endTime });
-        await this._event('testrun:changestatus', { status: 'COMPLETED' });
+        await this._event('testrun:changestatus', {
+          status: opts.earlyStop ? 'EARLY_STOP' : 'COMPLETED'
+        });
       }
     });