piratepoet/tests/integration/WooCommerce/TransactionalEmailHooksTest.php

<?php

namespace MailPoet\WooCommerce;

use Codeception\Stub;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Models\Newsletter;
use MailPoet\Newsletter\Editor\LayoutHelper as L;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Settings\SettingsController;
use MailPoet\WooCommerce\TransactionalEmails\Renderer;
use MailPoet\WP\Functions as WPFunctions;

class TransactionalEmailHooksTest extends \MailPoetTest {

  /** @var SettingsController */
  private $settings;

  /** @var array */
  private $originalWcSettings;

  /** @var NewslettersRepository */
  private $newslettersRepository;

  /** @var WPFunctions */
  private $wp;

  public function _before() {
    $this->settings = $this->diContainer->get(SettingsController::class);
    $this->originalWcSettings = $this->settings->get('woocommerce');
    $this->newslettersRepository = $this->diContainer->get(NewslettersRepository::class);
    $this->wp = $this->diContainer->get(WPFunctions::class);
  }

  public function testItCanReplaceWoocommerceEmailStyles() {
    $newsletter = new NewsletterEntity;
    $newsletter->setType(Newsletter::TYPE_WC_TRANSACTIONAL_EMAIL);
    $newsletter->setSubject('WooCommerce Transactional Email');
    $newsletter->setBody([
      'globalStyles' => [
        'text' =>
          [
            'fontColor' => '#111111',
            'fontFamily' => 'Arial',
            'fontSize' => '14px',
            'lineHeight' => '1.6',
          ],
        'h1' =>
          [
            'fontColor' => '#222222',
            'fontFamily' => 'Source Sans Pro',
            'fontSize' => '36px',
            'lineHeight' => '1.6',
          ],
        'h2' =>
          [
            'fontColor' => '#333333',
            'fontFamily' => 'Verdana',
            'fontSize' => '24px',
            'lineHeight' => '1.6',
          ],
        'h3' =>
          [
            'fontColor' => '#444444',
            'fontFamily' => 'Trebuchet MS',
            'fontSize' => '22px',
            'lineHeight' => '1.6',
          ],
        'link' =>
          [
            'fontColor' => '#555555',
            'textDecoration' => 'underline',
          ],
        'wrapper' =>
          [
            'backgroundColor' => '#666666',
          ],
        'body' =>
          [
            'backgroundColor' => '#777777',
          ],
        'woocommerce' =>
          [
            'brandingColor' => '#888888',
            'headingFontColor' => '#999999',
          ],
      ],
    ]);
    $this->newslettersRepository->persist($newsletter);
    $this->newslettersRepository->flush();
    $this->settings->set(TransactionalEmails::SETTING_EMAIL_ID, $newsletter->getId());

    // Set woo options
    $this->wp->updateOption('woocommerce_email_background_color', 'white');
    $this->wp->updateOption('woocommerce_email_base_color', 'red');
    $this->wp->updateOption('woocommerce_email_body_background_color', 'blue');
    $this->wp->updateOption('woocommerce_email_text_color', 'black');

    expect($this->wp->getOption('woocommerce_email_background_color'))->equals('white');
    expect($this->wp->getOption('woocommerce_email_base_color'))->equals('red');
    expect($this->wp->getOption('woocommerce_email_body_background_color'))->equals('blue');
    expect($this->wp->getOption('woocommerce_email_text_color'))->equals('black');

    $transactionalEmails = $this->diContainer->get(TransactionalEmailHooks::class);
    $transactionalEmails->overrideStylesForWooEmails();

    expect($this->wp->getOption('woocommerce_email_background_color'))->equals('#777777');
    expect($this->wp->getOption('woocommerce_email_base_color'))->equals('#888888');
    expect($this->wp->getOption('woocommerce_email_body_background_color'))->equals('#666666');
    expect($this->wp->getOption('woocommerce_email_text_color'))->equals('#111111');
  }

  public function testItDoesntReplaceWoocommerceEmailStylesIfEmailIsNotSet() {
    $this->settings->set(TransactionalEmails::SETTING_EMAIL_ID, null);
    // Set woo options
    $this->wp->updateOption('woocommerce_email_background_color', 'white');
    $this->wp->updateOption('woocommerce_email_base_color', 'red');
    $this->wp->updateOption('woocommerce_email_body_background_color', 'blue');
    $this->wp->updateOption('woocommerce_email_text_color', 'black');

    $transactionalEmails = $this->diContainer->get(TransactionalEmailHooks::class);
    $transactionalEmails->overrideStylesForWooEmails();

    expect($this->wp->getOption('woocommerce_email_background_color'))->equals('white');
    expect($this->wp->getOption('woocommerce_email_base_color'))->equals('red');
    expect($this->wp->getOption('woocommerce_email_body_background_color'))->equals('blue');
    expect($this->wp->getOption('woocommerce_email_text_color'))->equals('black');
  }

  public function testUseTemplateForWCEmails() {
    $addedActions = [];
    $removedActions = [];
    $newsletter = new NewsletterEntity;
    $newsletter->setType(Newsletter::TYPE_WC_TRANSACTIONAL_EMAIL);
    $newsletter->setSubject('WooCommerce Transactional Email');
    $newsletter->setBody([
      'content' => L::col([
        L::row([L::col([['type' => 'text', 'text' => 'Some text before heading']])]),
        ['type' => 'woocommerceHeading'],
        L::row([L::col([['type' => 'text', 'text' => 'Some text between heading and content']])]),
        ['type' => 'woocommerceContent'],
        L::row([L::col([['type' => 'text', 'text' => 'Some text after content']])]),
      ]),
    ]);
    $this->newslettersRepository->persist($newsletter);
    $this->newslettersRepository->flush();
    $this->settings->set(TransactionalEmails::SETTING_EMAIL_ID, $newsletter->getId());
    $wp = Stub::make(new WPFunctions, [
      'getOption' => function($name) {
        return '';
      },
      'addAction' => function ($name, $action) use(&$addedActions) {
        $addedActions[$name] = $action;
      },
      'removeAction' => function ($name, $action) use(&$removedActions) {
        $removedActions[$name] = $action;
      },
    ]);
    $renderer = Stub::make(Renderer::class, [
      'render' => function($email) use(&$newsletter) {
        expect($email->id)->equals($newsletter->getId());
      },
      'getHTMLBeforeContent' => function($headingText) {
        return 'HTML before content with ' . $headingText;
      },
      'getHTMLAfterContent' => function() {
        return 'HTML after content';
      },
      'prefixCss' => function($css) {
        return 'prefixed ' . $css;
      },
    ]);
    $wcEmails = $this->makeEmpty("\WC_Emails");

    $transactionalEmails = new TransactionalEmailHooks(
      $wp,
      $this->settings,
      $renderer,
      $this->diContainer->get(NewslettersRepository::class),
      $this->diContainer->get(TransactionalEmails::class)
    );
    $transactionalEmails->useTemplateForWoocommerceEmails();
    expect($addedActions)->count(1);
    expect($addedActions['woocommerce_email'])->callable();
    $addedActions['woocommerce_email']($wcEmails);
    expect($removedActions)->count(2);
    expect($addedActions)->count(4);
    expect($addedActions['woocommerce_email_header'])->callable();
    ob_start();
    $addedActions['woocommerce_email_header']('heading text');
    expect(ob_get_clean())->equals('HTML before content with heading text');
    expect($addedActions['woocommerce_email_footer'])->callable();
    ob_start();
    $addedActions['woocommerce_email_footer']();
    expect(ob_get_clean())->equals('HTML after content');
    expect($addedActions['woocommerce_email_styles'])->callable();
    expect($addedActions['woocommerce_email_styles']('some css'))->equals('prefixed some css');
  }

  public function testUseTemplateForWCEmailsWorksWithNoEmail() {
    $addedActions = [];
    $removedActions = [];

    $this->settings->set(TransactionalEmails::SETTING_EMAIL_ID, 2);
    $wp = Stub::make(new WPFunctions, [
      'getOption' => function($name) {
        return '';
      },
      'addAction' => function ($name, $action) use(&$addedActions) {
        $addedActions[$name] = $action;
      },
      'removeAction' => function ($name, $action) use(&$removedActions) {
        $removedActions[$name] = $action;
      },
    ]);

    new TransactionalEmailHooks(
      $wp,
      $this->settings,
      $this->diContainer->get(Renderer::class),
      $this->diContainer->get(NewslettersRepository::class),
      $this->diContainer->get(TransactionalEmails::class)
    );

  }

  public function _after() {
    $this->entityManager
      ->createQueryBuilder()
      ->delete()
      ->from(NewsletterEntity::class, 'n')
      ->getQuery()
      ->execute();
    $this->settings->set('woocommerce', $this->originalWcSettings);
  }
}