defensive coding when using cypher to encrypt mail messages

2amigos · Aug 25, 2016 · 1970853 · 1970853
1 parent 3a79e64
commit 1970853
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Queue/MailQueue.php b/src/Queue/MailQueue.php
@@ -2,6 +2,7 @@
 
 namespace Da\Mailer\Queue;
 
+use Da\Mailer\Model\MailMessage;
 use Da\Mailer\Queue\Backend\AbstractQueueStoreConnection;
 use Da\Mailer\Queue\Backend\MailJobInterface;
 use Da\Mailer\Queue\Backend\QueueStoreAdapterInterface;
@@ -56,8 +57,9 @@ public function getCypher()
      */
     public function enqueue(MailJobInterface $mailJob)
     {
-        if (null !== $this->getCypher()) {
-            $mailJob->setMessage($this->getCypher()->encodeMailMessage($mailJob->getMessage()));
+        $message = $mailJob->getMessage();
+        if (null !== $this->getCypher() && $message instanceof MailMessage) {
+            $mailJob->setMessage($this->getCypher()->encodeMailMessage($message));
         }
 
         return $this->adapter->enqueue($mailJob);