app/bundles/ReportBundle/Tests/Scheduler/Model/SendScheduleTest.php

<?php

namespace Mautic\ReportBundle\Tests\Scheduler\Model;

use Mautic\EmailBundle\Helper\MailHelper;
use Mautic\ReportBundle\Entity\Report;
use Mautic\ReportBundle\Entity\Scheduler;
use Mautic\ReportBundle\Exception\FileTooBigException;
use Mautic\ReportBundle\Scheduler\Model\FileHandler;
use Mautic\ReportBundle\Scheduler\Model\MessageSchedule;
use Mautic\ReportBundle\Scheduler\Model\SendSchedule;
use PHPUnit\Framework\MockObject\MockObject;
use Symfony\Component\EventDispatcher\EventDispatcher;

class SendScheduleTest extends \PHPUnit\Framework\TestCase
{
    private Report $report;

    private Scheduler $scheduler;

    private SendSchedule $sendSchedule;

    /**
     * @var MockObject|MailHelper
     */
    private MockObject $mailHelperMock;

    /**
     * @var MockObject|MessageSchedule
     */
    private MockObject $messageSchedule;

    /**
     * @var MockObject|FileHandler
     */
    private MockObject $fileHandler;

    private \PHPUnit\Framework\MockObject\MockObject|EventDispatcher $eventDispatcher;

    protected function setUp(): void
    {
        parent::setUp();

        $this->report          = new Report();
        $this->scheduler       = new Scheduler($this->report, new \DateTime());
        $this->mailHelperMock  = $this->createMock(MailHelper::class);
        $this->messageSchedule = $this->createMock(MessageSchedule::class);
        $this->fileHandler     = $this->createMock(FileHandler::class);
        $this->eventDispatcher = $this->createMock(EventDispatcher::class);

        $this->mailHelperMock->expects($this->once())
            ->method('getMailer')
            ->willReturnSelf();

        $this->sendSchedule = new SendSchedule(
            $this->mailHelperMock,
            $this->messageSchedule,
            $this->fileHandler,
            $this->eventDispatcher
        );
    }

    public function testSendScheduleWithCsvFile(): void
    {
        $this->report->setToAddress('[email protected], [email protected]');

        $this->messageSchedule->expects($this->once())
            ->method('getSubject')
            ->with($this->report)
            ->willReturn('Subject');

        $this->messageSchedule->expects($this->once())
            ->method('getMessageForAttachedFile')
            ->with($this->report)
            ->willReturn('Message');

        $this->fileHandler->expects($this->once())
            ->method('fileCanBeAttached')
            ->with('/path/to/report.csv');

        $this->mailHelperMock->expects($this->once())
            ->method('setTo')
            ->with(['[email protected]', '[email protected]']);

        $this->mailHelperMock->expects($this->once())
            ->method('setSubject')
            ->with('Subject');

        $this->mailHelperMock->expects($this->once())
            ->method('setBody')
            ->with('Message');

        $this->mailHelperMock->expects($this->once())
            ->method('parsePlainText')
            ->with('Message');

        $this->mailHelperMock->expects($this->once())
            ->method('attachFile')
            ->with('/path/to/report.csv', 'report.csv', 'text/csv');

        $this->mailHelperMock->expects($this->once())
            ->method('send')
            ->with(true);

        $this->sendSchedule->send($this->scheduler, '/path/to/report.csv');
    }

    public function testSendScheduleWithZipFile(): void
    {
        $this->report->setToAddress('[email protected], [email protected]');

        $this->messageSchedule->expects($this->once())
            ->method('getSubject')
            ->with($this->report)
            ->willReturn('Subject');

        $this->messageSchedule->expects($this->once())
            ->method('getMessageForAttachedFile')
            ->with($this->report)
            ->willReturn('Message');

        $this->fileHandler->expects($this->exactly(2))
            ->method('fileCanBeAttached')
            ->withConsecutive(
                ['/path/to/report.csv'],
                ['/path/to/report.zip']
            )
            ->will($this->onConsecutiveCalls(
                $this->throwException(new FileTooBigException()),
                null
            ));

        $this->fileHandler->expects($this->once())
            ->method('zipIt')
            ->with('/path/to/report.csv')
            ->willReturn('/path/to/report.zip');

        $this->mailHelperMock->expects($this->once())
            ->method('setTo')
            ->with(['[email protected]', '[email protected]']);

        $this->mailHelperMock->expects($this->once())
            ->method('setSubject')
            ->with('Subject');

        $this->mailHelperMock->expects($this->once())
            ->method('setBody')
            ->with('Message');

        $this->mailHelperMock->expects($this->once())
            ->method('parsePlainText')
            ->with('Message');

        $this->mailHelperMock->expects($this->once())
            ->method('attachFile')
            ->with('/path/to/report.zip', 'report.zip', 'application/zip');

        $this->mailHelperMock->expects($this->once())
            ->method('send')
            ->with(true);

        $this->sendSchedule->send($this->scheduler, '/path/to/report.csv');
    }

    public function testSendScheduleWithoutFile(): void
    {
        $this->report->setToAddress('[email protected], [email protected]');

        $this->messageSchedule->expects($this->once())
            ->method('getSubject')
            ->with($this->report)
            ->willReturn('Subject');

        $this->messageSchedule->expects($this->once())
            ->method('getMessageForLinkedFile')
            ->with($this->report)
            ->willReturn('Message');

        $this->fileHandler->expects($this->once())
            ->method('zipIt')
            ->with('path-to-a-file')
            ->willReturn('path-to-a-zip-file');

        $this->fileHandler->expects($this->exactly(2))
            ->method('fileCanBeAttached')
            ->withConsecutive(['path-to-a-file'], ['path-to-a-zip-file'])
            ->will($this->throwException(new FileTooBigException()));

        $this->mailHelperMock->expects($this->once())
            ->method('setTo')
            ->with(['[email protected]', '[email protected]']);

        $this->mailHelperMock->expects($this->once())
            ->method('setSubject')
            ->with('Subject');

        $this->mailHelperMock->expects($this->once())
            ->method('setBody')
            ->with('Message');

        $this->mailHelperMock->expects($this->once())
            ->method('parsePlainText')
            ->with('Message');

        $this->mailHelperMock->expects($this->never())
            ->method('attachFile');

        $this->mailHelperMock->expects($this->once())
            ->method('send')
            ->with(true);

        $this->sendSchedule->send($this->scheduler, 'path-to-a-file');
    }
}