Override batch notification recipient email

Author: gquemener

Command/BatchCommand.php

@@ -31,7 +31,20 @@ protected function configure()
             ->setDescription('Launch a registered job instance')
             ->addArgument('code', InputArgument::REQUIRED, 'Job instance code')
             ->addArgument('execution', InputArgument::OPTIONAL, 'Job execution id')
-            ->addOption('config', 'c', InputOption::VALUE_REQUIRED, 'Override job configuration (formatted as json. ie: php app/console oro:batch:job -c \'[{"reader":{"filePath":"/tmp/foo.csv"}}]\' acme_product_import)');
+            ->addOption(
+                'config',
+                'c',
+                InputOption::VALUE_REQUIRED,
+                'Override job configuration (formatted as json. ie: ' .
+                'php app/console oro:batch:job -c \'[{"reader":{"filePath":"/tmp/foo.csv"}}]\' ' .
+                'acme_product_import)'
+            )
+            ->addOption(
+                'email',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'The email to notify at the end of the job execution'
+            );
     }
 
     /**
@@ -62,6 +75,13 @@ protected function execute(InputInterface $input, OutputInterface $output)
             );
         }
 
+        // Override mail notifier recipient email
+        if ($email = $input->getOption('email')) {
+            $this
+                ->getMailNotifier()
+                ->setRecipientEmail($email);
+        }
+
         $errors = $this->getValidator()->validate($jobInstance, array('Default', 'Execution'));
         if (count($errors) > 0) {
             throw new \RuntimeException(sprintf('Job "%s" is invalid: %s', $code, $this->getErrorMessages($errors)));
@@ -72,7 +92,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
             $jobExecution = $this->getEntityManager()->getRepository('OroBatchBundle:JobExecution')->find($executionId);
             if (!$jobExecution) {
                 throw new \InvalidArgumentException(sprintf('Could not find job execution "%s".', $id));
-                }
+            }
             if (!$jobExecution->getStatus()->isStarting()) {
                 throw new \RuntimeException(
                     sprintf('Job execution "%s" has invalid status: %s', $executionId, $jobExecution->getStatus())
@@ -124,6 +144,14 @@ protected function getValidator()
         return $this->getContainer()->get('validator');
     }
 
+    /**
+     * @return Validator
+     */
+    protected function getMailNotifier()
+    {
+        return $this->getContainer()->get('oro_batch.mail_notifier');
+    }
+
     /**
      * @return \Oro\Bundle\BatchBundle\Connector\ConnectorRegistry
      */

Notification/MailNotifier.php

@@ -37,6 +37,11 @@ class MailNotifier implements Notifier
      */
     protected $senderEmail;
 
+    /**
+     * @var string $recipientEmail
+     */
+    protected $recipientEmail;
+
     /**
      * @param BatchLogHandler          $logger
      * @param SecurityContextInterface $securityContext
@@ -58,18 +63,30 @@ public function __construct(
         $this->senderEmail     = $senderEmail;
     }
 
+    /**
+     * Set the recipient email
+     *
+     * @param string $recipientEmail
+     *
+     * @return MailNotifier
+     */
+    public function setRecipientEmail($recipientEmail)
+    {
+        $this->recipientEmail = $recipientEmail;
+
+        return $this;
+    }
+
     /**
      * {@inheritdoc}
      */
     public function notify(JobExecution $jobExecution)
     {
-        $user = $this->getUser();
-        if (!$user) {
+        if (null === $email = $this->getEmail()) {
             return;
         }
 
         $parameters = array(
-            'user'         => $user,
             'jobExecution' => $jobExecution,
             'log'          => $this->logger->getFilename(),
         );
@@ -80,7 +97,7 @@ public function notify(JobExecution $jobExecution)
         $message = $this->mailer->createMessage();
         $message->setSubject('Job has been executed');
         $message->setFrom($this->senderEmail);
-        $message->setTo($user->getEmail());
+        $message->setTo($email);
         $message->setBody($txtBody, 'text/plain');
         $message->addPart($htmlBody, 'text/html');
 
@@ -90,10 +107,14 @@ public function notify(JobExecution $jobExecution)
     /**
      * Get the current authenticated user
      *
-     * @return null|UserInterface
+     * @return null|string
      */
-    private function getUser()
+    private function getEmail()
     {
+        if ($this->recipientEmail) {
+            return $this->recipientEmail;
+        }
+
         if (null === $token = $this->securityContext->getToken()) {
             return;
         }
@@ -102,6 +123,6 @@ private function getUser()
             return;
         }
 
-        return $user;
+        return $user->getEmail();
     }
 }

Resources/views/Mails/notification.html.twig

@@ -1,5 +1,3 @@
-<p>{{ user.firstName }} {{ user.lastName }},</p>
-
 <p>
     Akeneo successfully completed your batch {{ jobExecution.jobInstance.type }}.<br />
     <br />

Resources/views/Mails/notification.txt.twig

@@ -1,5 +1,3 @@
-{{ user.firstName }} {{ user.lastName }},
-
 Akeneo successfully completed your batch {{ jobExecution.jobInstance.type }}.
 
 Started on {{ jobExecution.startTime|date("Y-m-d") }} at {{ jobExecution.startTime|date("H:i:s")  }}.
@@ -12,5 +10,5 @@ Results:
     - {{ stepExecution.writeCount }} item(s) written
 {% endfor %}
 
--- 
+--
 Oro Platform

Tests/Unit/Notification/MailNotifierTest.php

@@ -51,7 +51,6 @@ public function testNotifyWithLoggedInUserEmail()
 
         $jobExecution = $this->getDisabledConstructorMock('Oro\Bundle\BatchBundle\Entity\JobExecution');
         $parameters = array(
-            'user'         => $user,
             'jobExecution' => $jobExecution,
             'log'          => '/tmp/foo.log',
         );
@@ -102,6 +101,63 @@ public function testNotifyWithLoggedInUserEmail()
         $this->notifier->notify($jobExecution);
     }
 
+    public function testNotifyIfRecipientEmailIsSet()
+    {
+        $this->handler
+            ->expects($this->once())
+            ->method('getFilename')
+            ->will($this->returnValue('/tmp/foo.log'));
+
+        $jobExecution = $this->getDisabledConstructorMock('Oro\Bundle\BatchBundle\Entity\JobExecution');
+        $parameters = array(
+            'jobExecution' => $jobExecution,
+            'log'          => '/tmp/foo.log',
+        );
+        $this->twig
+            ->expects($this->exactly(2))
+            ->method('render')
+            ->will(
+                $this->returnValueMap(
+                    array(
+                        array('OroBatchBundle:Mails:notification.txt.twig', $parameters, 'notification'),
+                        array('OroBatchBundle:Mails:notification.html.twig', $parameters, '<p>notification</p>'),
+                    )
+                )
+            );
+
+        $message = $this->getDisabledConstructorMock('\Swift_Message');
+        $this->mailer
+            ->expects($this->once())
+            ->method('createMessage')
+            ->will($this->returnValue($message));
+
+        $message->expects($this->once())
+            ->method('setSubject')
+            ->with('Job has been executed');
+
+        $message->expects($this->once())
+            ->method('setFrom')
+            ->with('[email protected]');
+
+        $message->expects($this->once())
+            ->method('setTo')
+            ->with('[email protected]');
+        $message->expects($this->once())
+            ->method('setBody')
+            ->with('notification', 'text/plain');
+        $message->expects($this->once())
+            ->method('addPart')
+            ->with('<p>notification</p>', 'text/html');
+
+        $this->mailer
+            ->expects($this->once())
+            ->method('send')
+            ->with($message);
+
+        $this->notifier->setRecipientEmail('[email protected]');
+        $this->notifier->notify($jobExecution);
+    }
+
     public function testDoNotNotifyIfNoUserLoggedIn()
     {
         $token = $this->getTokenMock(null);