class MailHandlerTest

class MailHandlerTest extends UnitTestCase {
    
    /**
     * Language manager service.
     *
     * @var \Drupal\Core\Language\LanguageManagerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $languageManager;
    
    /**
     * Logger service.
     *
     * @var \Psr\Log\LoggerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $logger;
    
    /**
     * Mail manager service.
     *
     * @var \Drupal\Core\Mail\MailManagerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $mailManager;
    
    /**
     * Contact mail messages service.
     *
     * @var \Drupal\contact\MailHandlerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $contactMailHandler;
    
    /**
     * The contact form entity.
     *
     * @var \Drupal\contact\ContactFormInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $contactForm;
    
    /**
     * The entity type manager.
     *
     * @var \Drupal\Core\Entity\EntityTypeManagerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $entityTypeManager;
    
    /**
     * The user storage handler.
     *
     * @var \Drupal\Core\Entity\EntityStorageInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $userStorage;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->mailManager = $this->createMock('\\Drupal\\Core\\Mail\\MailManagerInterface');
        $this->languageManager = $this->createMock('\\Drupal\\Core\\Language\\LanguageManagerInterface');
        $this->logger = $this->createMock('\\Psr\\Log\\LoggerInterface');
        $this->entityTypeManager = $this->createMock(EntityTypeManagerInterface::class);
        $this->userStorage = $this->createMock('\\Drupal\\Core\\Entity\\EntityStorageInterface');
        $this->entityTypeManager
            ->expects($this->any())
            ->method('getStorage')
            ->with('user')
            ->willReturn($this->userStorage);
        $string_translation = $this->getStringTranslationStub();
        $this->contactMailHandler = new MailHandler($this->mailManager, $this->languageManager, $this->logger, $string_translation, $this->entityTypeManager);
        $language = new Language([
            'id' => 'en',
        ]);
        $this->languageManager
            ->expects($this->any())
            ->method('getDefaultLanguage')
            ->willReturn($language);
        $this->languageManager
            ->expects($this->any())
            ->method('getCurrentLanguage')
            ->willReturn($language);
    }
    
    /**
     * Tests the children() method with an invalid key.
     *
     * @covers ::sendMailMessages
     */
    public function testInvalidRecipient() : void {
        $message = $this->createMock('\\Drupal\\contact\\MessageInterface');
        $message->expects($this->once())
            ->method('isPersonal')
            ->willReturn(TRUE);
        $message->expects($this->once())
            ->method('getPersonalRecipient')
            ->willReturn(NULL);
        $message->expects($this->once())
            ->method('getContactForm')
            ->willReturn($this->createMock('\\Drupal\\contact\\ContactFormInterface'));
        $sender = $this->createMock('\\Drupal\\Core\\Session\\AccountInterface');
        $this->userStorage
            ->expects($this->any())
            ->method('load')
            ->willReturn($sender);
        // User IDs 1 and 0 have special implications, use 3 instead.
        $sender->expects($this->any())
            ->method('id')
            ->willReturn(3);
        $sender->expects($this->once())
            ->method('isAnonymous')
            ->willReturn(FALSE);
        $this->expectException(MailHandlerException::class);
        $this->expectExceptionMessage('Unable to determine message recipient');
        $this->contactMailHandler
            ->sendMailMessages($message, $sender);
    }
    
    /**
     * Tests the sendMailMessages method.
     *
     * @dataProvider getSendMailMessages
     *
     * @covers ::sendMailMessages
     */
    public function testSendMailMessages(bool $anonymous, ?bool $auto_reply, bool $copy_sender, array $results) : void {
        if ($anonymous) {
            $message = $this->getAnonymousMockMessage(explode(', ', $results[0]['to']), $auto_reply, $copy_sender);
            $sender = $this->getMockSender();
        }
        else {
            $message = $this->getAuthenticatedMockMessage($copy_sender);
            $sender = $this->getMockSender(FALSE, 'user@drupal.org');
        }
        $expected_params['contact_message'] = $message;
        $expected_params['sender'] = $sender;
        if ($anonymous) {
            $expected_params['contact_form'] = $message->getContactForm();
        }
        else {
            $expected_params['recipient'] = $message->getPersonalRecipient();
        }
        $this->logger
            ->expects($this->once())
            ->method('info');
        $this->mailManager
            ->expects($this->any())
            ->method('mail')
            ->willReturnCallback(function ($module, $key, $to, $langcode, $params, $from) use (&$results, $expected_params) {
            $result = array_shift($results);
            $this->assertEquals($module, $result['module']);
            $this->assertEquals($key, $result['key']);
            $this->assertEquals($to, $result['to']);
            $this->assertEquals($langcode, $result['langcode']);
            $this->assertEquals($params, $expected_params);
            $this->assertEquals($from, $result['from']);
        });
        $this->userStorage
            ->expects($this->any())
            ->method('load')
            ->willReturn(clone $sender);
        $this->contactMailHandler
            ->sendMailMessages($message, $sender);
    }
    
    /**
     * Data provider for ::testSendMailMessages.
     */
    public static function getSendMailMessages() {
        $default_result = [
            'module' => 'contact',
            'key' => 'page_mail',
            'to' => 'admin@drupal.org, user@drupal.org',
            'langcode' => 'en',
            'params' => [],
            'from' => 'anonymous@drupal.org',
        ];
        $autoreply_result = [
            'key' => 'page_autoreply',
            'to' => 'anonymous@drupal.org',
            'from' => NULL,
        ] + $default_result;
        $copy_result = [
            'key' => 'page_copy',
            'to' => 'anonymous@drupal.org',
        ] + $default_result;
        (yield 'anonymous, no auto reply, no copy sender' => [
            TRUE,
            FALSE,
            FALSE,
            [
                $default_result,
            ],
        ]);
        (yield 'anonymous, auto reply, no copy sender' => [
            TRUE,
            TRUE,
            FALSE,
            [
                $default_result,
                $autoreply_result,
            ],
        ]);
        (yield 'anonymous, no auto reply, copy sender' => [
            TRUE,
            FALSE,
            TRUE,
            [
                $default_result,
                $copy_result,
            ],
        ]);
        (yield 'anonymous, auto reply, copy sender' => [
            TRUE,
            TRUE,
            TRUE,
            [
                $default_result,
                $copy_result,
                $autoreply_result,
            ],
        ]);
        // For authenticated user.
        $default_result = [
            'module' => 'contact',
            'key' => 'user_mail',
            'to' => 'user2@drupal.org',
            'langcode' => 'en',
            'from' => 'user@drupal.org',
        ];
        $copy_result = [
            'key' => 'user_copy',
            'to' => 'user@drupal.org',
        ] + $default_result;
        (yield 'authenticated, no copy sender' => [
            FALSE,
            NULL,
            FALSE,
            [
                $default_result,
            ],
        ]);
        (yield 'authenticated, copy sender' => [
            FALSE,
            NULL,
            TRUE,
            [
                $default_result,
                $copy_result,
            ],
        ]);
    }
    
    /**
     * Builds a mock sender on given scenario.
     *
     * @param bool $anonymous
     *   TRUE if the sender is anonymous.
     * @param string $mail_address
     *   The mail address of the user.
     *
     * @return \Drupal\Core\Session\AccountInterface|\PHPUnit\Framework\MockObject\MockObject
     *   Mock sender for testing.
     */
    protected function getMockSender($anonymous = TRUE, $mail_address = 'anonymous@drupal.org') {
        $sender = $this->createMock(User::class);
        $sender->expects($this->once())
            ->method('isAnonymous')
            ->willReturn($anonymous);
        $sender->expects($this->any())
            ->method('getEmail')
            ->willReturn($mail_address);
        $sender->expects($this->any())
            ->method('getDisplayName')
            ->willReturn('user');
        // User ID 1 has special implications, use 3 instead.
        $sender->expects($this->any())
            ->method('id')
            ->willReturn($anonymous ? 0 : 3);
        if ($anonymous) {
            // Anonymous user values set in params include updated values for name and
            // mail.
            $sender->name = 'Anonymous (not verified)';
            $sender->mail = 'anonymous@drupal.org';
        }
        return $sender;
    }
    
    /**
     * Builds a mock message from anonymous user.
     *
     * @param array $recipients
     *   An array of recipient email addresses.
     * @param bool $auto_reply
     *   TRUE if auto reply is enable.
     * @param bool $copy_sender
     *   TRUE if a copy should be sent, FALSE if not.
     *
     * @return \Drupal\contact\MessageInterface|\PHPUnit\Framework\MockObject\MockObject
     *   Mock message for testing.
     */
    protected function getAnonymousMockMessage($recipients, $auto_reply, $copy_sender = FALSE) {
        $message = $this->createMock('\\Drupal\\contact\\MessageInterface');
        $message->expects($this->any())
            ->method('getSenderName')
            ->willReturn('Anonymous');
        $message->expects($this->once())
            ->method('getSenderMail')
            ->willReturn('anonymous@drupal.org');
        $message->expects($this->any())
            ->method('isPersonal')
            ->willReturn(FALSE);
        $message->expects($this->once())
            ->method('copySender')
            ->willReturn($copy_sender);
        $message->expects($this->any())
            ->method('getContactForm')
            ->willReturn($this->getMockContactForm($recipients, $auto_reply ? 'reply' : ''));
        return $message;
    }
    
    /**
     * Builds a mock message from authenticated user.
     *
     * @param bool $copy_sender
     *   TRUE if a copy should be sent, FALSE if not.
     *
     * @return \Drupal\contact\MessageInterface|\PHPUnit\Framework\MockObject\MockObject
     *   Mock message for testing.
     */
    protected function getAuthenticatedMockMessage($copy_sender = FALSE) {
        $message = $this->createMock('\\Drupal\\contact\\MessageInterface');
        $message->expects($this->any())
            ->method('isPersonal')
            ->willReturn(TRUE);
        $message->expects($this->once())
            ->method('copySender')
            ->willReturn($copy_sender);
        $recipient = $this->createMock('\\Drupal\\user\\UserInterface');
        $recipient->expects($this->once())
            ->method('getEmail')
            ->willReturn('user2@drupal.org');
        $recipient->expects($this->any())
            ->method('getDisplayName')
            ->willReturn('user2');
        $recipient->expects($this->once())
            ->method('getPreferredLangcode')
            ->willReturn('en');
        $message->expects($this->any())
            ->method('getPersonalRecipient')
            ->willReturn($recipient);
        return $message;
    }
    
    /**
     * Builds a mock message on given scenario.
     *
     * @param array $recipients
     *   An array of recipient email addresses.
     * @param string $auto_reply
     *   An auto-reply message to send to the message author.
     *
     * @return \Drupal\contact\ContactFormInterface|\PHPUnit\Framework\MockObject\MockObject
     *   Mock message for testing.
     */
    protected function getMockContactForm($recipients, $auto_reply) {
        $contact_form = $this->createMock('\\Drupal\\contact\\ContactFormInterface');
        $contact_form->expects($this->once())
            ->method('getRecipients')
            ->willReturn($recipients);
        $contact_form->expects($this->once())
            ->method('getReply')
            ->willReturn($auto_reply);
        return $contact_form;
    }

}