interrupt signal is converted to InterruptException and handled by CliTester

allows to interrupt watch mode

Author: dg

src/Runner/CliTester.php

@@ -66,6 +66,8 @@ public function run(): ?int
 		$runner->setEnvironmentVariable(Environment::RUNNER, '1');
 		$runner->setEnvironmentVariable(Environment::COLORS, (string) (int) Environment::$useColors);
 
+		$this->installInterruptHandler();
+
 		if ($this->options['--coverage']) {
 			$coverageFile = $this->prepareCodeCoverage($runner);
 		}
@@ -360,7 +362,9 @@ private function setupErrors(): void
 		});
 
 		set_exception_handler(function (\Throwable $e) {
-			$this->displayException($e);
+			if (!$e instanceof InterruptException) {
+				$this->displayException($e);
+			}
 			exit(2);
 		});
 	}
@@ -374,4 +378,21 @@ private function displayException(\Throwable $e): void
 			: Dumper::color('white/red', 'Error: ' . $e->getMessage());
 		echo "\n";
 	}
+
+
+	private function installInterruptHandler(): void
+	{
+		if (function_exists('pcntl_signal')) {
+			pcntl_signal(SIGINT, function (): void {
+				pcntl_signal(SIGINT, SIG_DFL);
+				throw new InterruptException;
+			});
+			pcntl_async_signals(true);
+
+		} elseif (function_exists('sapi_windows_set_ctrl_handler') && PHP_SAPI === 'cli') {
+			sapi_windows_set_ctrl_handler(function (): void {
+				throw new InterruptException;
+			});
+		}
+	}
 }

src/Runner/Runner.php

@@ -125,36 +125,37 @@ public function run(): bool
 
 		$threads = range(1, $this->threadCount);
 
-		$this->installInterruptHandler();
 		$async = $this->threadCount > 1 && count($this->jobs) > 1;
 
-		while (($this->jobs || $running) && !$this->isInterrupted()) {
-			while ($threads && $this->jobs) {
-				$running[] = $job = array_shift($this->jobs);
-				$job->setEnvironmentVariable(Environment::THREAD, (string) array_shift($threads));
-				$job->run($async ? $job::RUN_ASYNC : 0);
-			}
-
-			if ($async) {
-				usleep(Job::RUN_USLEEP); // stream_select() doesn't work with proc_open()
-			}
+		try {
+			while (($this->jobs || $running) && !$this->interrupted) {
+				while ($threads && $this->jobs) {
+					$running[] = $job = array_shift($this->jobs);
+					$job->setEnvironmentVariable(Environment::THREAD, (string) array_shift($threads));
+					$job->run($async ? $job::RUN_ASYNC : 0);
+				}
 
-			foreach ($running as $key => $job) {
-				if ($this->isInterrupted()) {
-					break 2;
+				if ($async) {
+					usleep(Job::RUN_USLEEP); // stream_select() doesn't work with proc_open()
 				}
 
-				if (!$job->isRunning()) {
-					$threads[] = $job->getEnvironmentVariable(Environment::THREAD);
-					$this->testHandler->assess($job);
-					unset($running[$key]);
+				foreach ($running as $key => $job) {
+					if ($this->interrupted) {
+						break 2;
+					}
+
+					if (!$job->isRunning()) {
+						$threads[] = $job->getEnvironmentVariable(Environment::THREAD);
+						$this->testHandler->assess($job);
+						unset($running[$key]);
+					}
 				}
 			}
-		}
-		$this->removeInterruptHandler();
 
-		foreach ($this->outputHandlers as $handler) {
-			$handler->end();
+		} finally {
+			foreach ($this->outputHandlers as $handler) {
+				$handler->end();
+			}
 		}
 
 		return $this->result;
@@ -235,41 +236,6 @@ public function getInterpreter(): PhpInterpreter
 	}
 
 
-	private function installInterruptHandler(): void
-	{
-		if (function_exists('pcntl_signal')) {
-			pcntl_signal(SIGINT, function (): void {
-				pcntl_signal(SIGINT, SIG_DFL);
-				$this->interrupted = true;
-			});
-		} elseif (function_exists('sapi_windows_set_ctrl_handler') && PHP_SAPI === 'cli') {
-			sapi_windows_set_ctrl_handler(function () {
-				$this->interrupted = true;
-			});
-		}
-	}
-
-
-	private function removeInterruptHandler(): void
-	{
-		if (function_exists('pcntl_signal')) {
-			pcntl_signal(SIGINT, SIG_DFL);
-		} elseif (function_exists('sapi_windows_set_ctrl_handler') && PHP_SAPI === 'cli') {
-			sapi_windows_set_ctrl_handler(null);
-		}
-	}
-
-
-	private function isInterrupted(): bool
-	{
-		if (function_exists('pcntl_signal_dispatch')) {
-			pcntl_signal_dispatch();
-		}
-
-		return $this->interrupted;
-	}
-
-
 	private function getLastResult(Test $test): int
 	{
 		$signature = $test->getSignature();

src/Runner/exceptions.php

@@ -0,0 +1,15 @@
+<?php
+
+/**
+ * This file is part of the Nette Tester.
+ * Copyright (c) 2009 David Grudl (https://davidgrudl.com)
+ */
+
+declare(strict_types=1);
+
+namespace Tester\Runner;
+
+
+class InterruptException extends \Exception
+{
+}

src/tester.php

@@ -7,6 +7,7 @@
 
 declare(strict_types=1);
 
+require __DIR__ . '/Runner/exceptions.php';
 require __DIR__ . '/Runner/Test.php';
 require __DIR__ . '/Runner/PhpInterpreter.php';
 require __DIR__ . '/Runner/Runner.php';