TransportBuilderMock.php

Go to the documentation of this file.

<?php
namespace Magento\TestFramework\Mail\Template;

class TransportBuilderMock extends \Magento\Framework\Mail\Template\TransportBuilder
{
    protected $_sentMessage;

    protected function reset()
    {
        $this->_sentMessage = $this->message;
        parent::reset();
    }

    public function getSentMessage()
    {
        return $this->_sentMessage;
    }

    public function getTransport()
    {
        $this->prepareMessage();
        $this->reset();
        return new \Magento\TestFramework\Mail\TransportInterfaceMock();
    }
}