Fix tests

Author: jeremyFreeAgent

src/Symfony/Component/Notifier/Tests/Transport/TransportsTest.php

@@ -15,6 +15,7 @@
 use Symfony\Component\Notifier\Exception\InvalidArgumentException;
 use Symfony\Component\Notifier\Exception\LogicException;
 use Symfony\Component\Notifier\Message\ChatMessage;
+use Symfony\Component\Notifier\Message\SentMessage;
 use Symfony\Component\Notifier\Transport\TransportInterface;
 use Symfony\Component\Notifier\Transport\Transports;
 
@@ -30,9 +31,12 @@ public function testSendToTransportDefinedByMessage(): void
 
         $one->method('supports')->with($message)->willReturn(true);
 
-        $one->expects($this->once())->method('send');
+        $one->expects($this->once())->method('send')->willReturn(new SentMessage($message, 'one'));
 
-        $transports->send($message);
+        $sentMessage = $transports->send($message);
+
+        $this->assertSame($message, $sentMessage->getOriginalMessage());
+        $this->assertSame('one', $sentMessage->getTransport());
     }
 
     public function testSendToFirstSupportedTransportIfMessageDoesNotDefineATransport(): void
@@ -47,10 +51,16 @@ public function testSendToFirstSupportedTransportIfMessageDoesNotDefineATranspor
         $one->method('supports')->with($message)->willReturn(false);
         $two->method('supports')->with($message)->willReturn(true);
 
+        $one->method('send')->with($message)->willReturn(new SentMessage($message, 'one'));
+        $two->method('send')->with($message)->willReturn(new SentMessage($message, 'two'));
+
         $one->expects($this->never())->method('send');
-        $two->expects($this->once())->method('send');
+        $two->expects($this->once())->method('send')->willReturn(new SentMessage($message, 'two'));
 
-        $transports->send($message);
+        $sentMessage = $transports->send($message);
+
+        $this->assertSame($message, $sentMessage->getOriginalMessage());
+        $this->assertSame('two', $sentMessage->getTransport());
     }
 
     public function testThrowExceptionIfNoSupportedTransportWasFound(): void