prevent mailer from sending messages marked as "draft"

kbond · kbond · commit 48ef6e2e5598 · 2021-12-20T12:06:38.000-05:00
diff --git a/src/Symfony/Component/Mailer/Mailer.php b/src/Symfony/Component/Mailer/Mailer.php
@@ -13,9 +13,11 @@
 
 use Psr\EventDispatcher\EventDispatcherInterface;
 use Symfony\Component\Mailer\Event\MessageEvent;
+use Symfony\Component\Mailer\Exception\LogicException;
 use Symfony\Component\Mailer\Messenger\SendEmailMessage;
 use Symfony\Component\Mailer\Transport\TransportInterface;
 use Symfony\Component\Messenger\MessageBusInterface;
+use Symfony\Component\Mime\Message;
 use Symfony\Component\Mime\RawMessage;
 
 /**
@@ -36,6 +38,10 @@ public function __construct(TransportInterface $transport, MessageBusInterface $
 
     public function send(RawMessage $message, Envelope $envelope = null): void
     {
+        if ($message instanceof Message && '1' === $message->getHeaders()->getHeaderBody('X-Unsent')) {
+            throw new LogicException('Cannot send messages marked as "draft".');
+        }
+
         if (null === $this->bus) {
             $this->transport->send($message, $envelope);
 
diff --git a/src/Symfony/Component/Mailer/Tests/MailerTest.php b/src/Symfony/Component/Mailer/Tests/MailerTest.php
@@ -69,4 +69,16 @@ public function dispatch($message, array $stamps = []): Envelope
         self::assertCount(1, $bus->messages);
         self::assertSame($email, $bus->messages[0]->getMessage());
     }
+
+    public function testCannotSendDraftEmails()
+    {
+        $transport = new Mailer($this->createMock(TransportInterface::class), $this->createMock(MessageBusInterface::class), $this->createMock(EventDispatcherInterface::class));
+
+        $this->expectException(LogicException::class);
+
+        $email = new Email();
+        $email->getHeaders()->addTextHeader('X-Unsent', '1');
+
+        $transport->send($email);
+    }
 }
