fix failing test

Signed-off-by: MarioRadu <[email protected]>
dotkernel · Nov 29, 2024 · 44910ae · 44910ae
1 parent 5224a41
commit 44910ae
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/Unit/Contact/Service/MessageServiceTest.php b/test/Unit/Contact/Service/MessageServiceTest.php
@@ -4,6 +4,7 @@
 
 namespace FrontendTest\Unit\Contact\Service;
 
+use Dot\Mail\Email;
 use Dot\Mail\Exception\MailException;
 use Dot\Mail\Result\ResultInterface;
 use Dot\Mail\Service\MailServiceInterface;
@@ -41,10 +42,12 @@ public function testProcessMessage(): void
         $mailService       = $this->createMock(MailServiceInterface::class);
         $template          = $this->createMock(TemplateRendererInterface::class);
         $result            = $this->createMock(ResultInterface::class);
+        $mail             = $this->createMock(Email::class);
 
+        $mail->expects($this->once())->method('addFrom')->willReturn($mail);
         $result->expects($this->once())->method('isValid')->willReturn(true);
         $mailService->expects($this->once())->method('send')->willReturn($result);
-
+        $mailService->expects($this->any())->method('getMessage')->willReturn($mail);
         $service = new MessageService(
             $messageRepository,
             $mailService,