[Console][Messenger] Fix: Allow UnrecoverableExceptionInterface to bypass retry in RunCommandMessageHandler

Co-authored-by: Kevin Bond <[email protected]>

Author: santysisi

src/Symfony/Component/Console/Messenger/RunCommandMessageHandler.php

@@ -16,6 +16,8 @@
 use Symfony\Component\Console\Exception\RunCommandFailedException;
 use Symfony\Component\Console\Input\StringInput;
 use Symfony\Component\Console\Output\BufferedOutput;
+use Symfony\Component\Messenger\Exception\RecoverableExceptionInterface;
+use Symfony\Component\Messenger\Exception\UnrecoverableExceptionInterface;
 
 /**
  * @author Kevin Bond <[email protected]>
@@ -35,6 +37,8 @@ public function __invoke(RunCommandMessage $message): RunCommandContext
 
         try {
             $exitCode = $this->application->run($input, $output);
+        } catch (UnrecoverableExceptionInterface|RecoverableExceptionInterface $e) {
+            throw $e;
         } catch (\Throwable $e) {
             throw new RunCommandFailedException($e, new RunCommandContext($message, Command::FAILURE, $output->fetch()));
         }

src/Symfony/Component/Console/Tests/Messenger/RunCommandMessageHandlerTest.php

@@ -20,6 +20,10 @@
 use Symfony\Component\Console\Messenger\RunCommandMessage;
 use Symfony\Component\Console\Messenger\RunCommandMessageHandler;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Messenger\Exception\RecoverableExceptionInterface;
+use Symfony\Component\Messenger\Exception\RecoverableMessageHandlingException;
+use Symfony\Component\Messenger\Exception\UnrecoverableExceptionInterface;
+use Symfony\Component\Messenger\Exception\UnrecoverableMessageHandlingException;
 
 /**
  * @author Kevin Bond <[email protected]>
@@ -81,6 +85,38 @@ public function testThrowOnNonSuccess()
         $this->fail('Exception not thrown.');
     }
 
+    public function testExecutesCommandThatThrownUnrecoverableException()
+    {
+        $handler = new RunCommandMessageHandler($this->createApplicationWithCommand());
+
+        try {
+            $handler(new RunCommandMessage('test:command --throw-unrecoverable'));
+        } catch (UnrecoverableExceptionInterface $e) {
+            $this->assertSame('Unrecoverable exception message', $e->getMessage());
+            $this->assertNull($e->getPrevious());
+
+            return;
+        }
+
+        $this->fail('Exception not thrown.');
+    }
+
+    public function testExecutesCommandThatThrownRecoverableException()
+    {
+        $handler = new RunCommandMessageHandler($this->createApplicationWithCommand());
+
+        try {
+            $handler(new RunCommandMessage('test:command --throw-recoverable'));
+        } catch (RecoverableExceptionInterface $e) {
+            $this->assertSame('Recoverable exception message', $e->getMessage());
+            $this->assertNull($e->getPrevious());
+
+            return;
+        }
+
+        $this->fail('Exception not thrown.');
+    }
+
     private function createApplicationWithCommand(): Application
     {
         $application = new Application();
@@ -92,6 +128,8 @@ public function configure(): void
                     $this
                         ->setName('test:command')
                         ->addOption('throw')
+                        ->addOption('throw-unrecoverable')
+                        ->addOption('throw-recoverable')
                         ->addOption('exit', null, InputOption::VALUE_REQUIRED, 0)
                     ;
                 }
@@ -100,6 +138,14 @@ protected function execute(InputInterface $input, OutputInterface $output): int
                 {
                     $output->write('some message');
 
+                    if ($input->getOption('throw-unrecoverable')) {
+                        throw new UnrecoverableMessageHandlingException('Unrecoverable exception message');
+                    }
+
+                    if ($input->getOption('throw-recoverable')) {
+                        throw new RecoverableMessageHandlingException('Recoverable exception message');
+                    }
+
                     if ($input->getOption('throw')) {
                         throw new \RuntimeException('exception message');
                     }