wip: addressing PR comments

n8n-io · Dec 19, 2024 · 9cec618 · 9cec618
1 parent 755ea2c
commit 9cec618
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/packages/cli/src/evaluation/test-runner/test-runner.service.ee.ts b/packages/cli/src/evaluation/test-runner/test-runner.service.ee.ts
@@ -6,7 +6,7 @@ import type {
 	IRunExecutionData,
 	IWorkflowExecutionDataProcess,
 } from 'n8n-workflow';
-import { NodeConnectionType, Workflow } from 'n8n-workflow';
+import { ExecutionCancelledError, NodeConnectionType, Workflow } from 'n8n-workflow';
 import assert from 'node:assert';
 import { Service } from 'typedi';
 
@@ -306,6 +306,12 @@ export class TestRunnerService {
 				const aggregatedMetrics = metrics.getAggregatedMetrics();
 				await this.testRunRepository.markAsCompleted(testRun.id, aggregatedMetrics);
 			}
+		} catch (e) {
+			if (e instanceof ExecutionCancelledError) {
+				await this.testRunRepository.markAsCancelled(testRun.id);
+			} else {
+				throw e;
+			}
 		} finally {
 			// Clean up abort controller
 			this.abortControllers.delete(testRun.id);