Add test for exceptionHandler property on RoundRobinTransport

jnoordsij · jnoordsij · commit ebdb82411995 · 2025-04-01T14:25:55.000+02:00
diff --git a/src/Symfony/Component/Mailer/Tests/Transport/RoundRobinTransportTest.php b/src/Symfony/Component/Mailer/Tests/Transport/RoundRobinTransportTest.php
@@ -144,6 +144,37 @@ public function testSendOneDeadAndRecoveryWithinRetryPeriod()
         $this->assertTransports($t, 1, []);
     }
 
+    public function testExceptionHandlerReceivesExceptions()
+    {
+        $t1 = $this->createMock(TransportInterface::class);
+        $t1->expects($this->exactly(2))->method('send');
+
+        $t2 = $this->createMock(TransportInterface::class);
+        $t2->expects($this->exactly(1))
+            ->method('send')
+            ->willReturnCallback(function () {
+                throw new TransportException();
+            });
+
+        $eh = $this
+            ->getMockBuilder(\stdClass::class)
+            ->addMethods(['__invoke'])
+            ->getMock();
+        $eh->expects($this->exactly(1))
+            ->method('__invoke')
+            ->willReturnCallback(function (TransportExceptionInterface $e) {
+                $this->assertInstanceOf(TransportException::class, $e);
+            });
+
+        $t = new RoundRobinTransport([$t1, $t2], exceptionHandler: $eh);
+        $p = new \ReflectionProperty($t, 'cursor');
+        $p->setValue($t, 0);
+        $t->send(new RawMessage(''));
+        $this->assertTransports($t, 1, []);
+        $t->send(new RawMessage(''));
+        $this->assertTransports($t, 1, [$t2]);
+    }
+
     public function testFailureDebugInformation()
     {
         $t1 = $this->createMock(TransportInterface::class);
