fix(runner): stop runner if test time exceeds the duration given (#53)

Zooz · Dec 26, 2020 · 387b552 · 387b552 · roychri · Jun 13, 2021
1 parent e16a596
commit 387b552
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/app/app.js b/app/app.js
@@ -11,6 +11,8 @@ const runner = require('./models/runner'),
     errorHandler = require('./handler/errorHandler'),
     { version: PREDATOR_RUNNER_VERSION } = require('../package.json');
 
+const RUNNER_TIMEOUT_GRACE_MS = 300;
+
 const getContainerId = () => {
     let containerId = uuid();
     if (process.env.MARATHON_APP_ID) {
@@ -45,7 +47,18 @@ let start = async () => {
             });
             process.exit(1);
         });
+        process.on('SIGUSR1', async function() {
+            logger.info('Runner exceeded test duration, sending DONE status and existing');
+            await reporterConnector.postStats(jobConfig, {
+                phase_status: 'done',
+                data: JSON.stringify({ message: 'Test Finished' })
+            });
+            process.exit(1);
+        });
         verifyPredatorVersion();
+        setTimeout(function() {
+            process.kill(process.pid, 'SIGUSR1');
+        }, (jobConfig.duration * 1000) + jobConfig.delayRunnerMs + RUNNER_TIMEOUT_GRACE_MS);
         await runner.runTest(jobConfig);
         logger.info('Finished running test successfully');
         process.exit(0);