<?php
namespace MailPoet\Test\Newsletter;

use MailPoet\Models\CustomField;
use MailPoet\Newsletter\Shortcodes\ShortcodesHelper;

class ShortcodesHelperTest extends \MailPoetTest {
  function testGetsShortcodes() {
    $shortcodes = ShortcodesHelper::getShortcodes();
    expect(array_keys($shortcodes))->equals(
      [
        'Subscriber',
        'Newsletter',
        'Post Notifications',
        'Date',
        'Links',
      ]
    );
  }

  function testItGetsCustomShortShortcodes() {
    $shortcodes = ShortcodesHelper::getShortcodes();
    expect(count($shortcodes['Subscriber']))->equals(5);
    $custom_field = CustomField::create();
    $custom_field->name = 'name';
    $custom_field->type = 'type';
    $custom_field->save();
    $shortcodes = ShortcodesHelper::getShortcodes();
    expect(count($shortcodes['Subscriber']))->equals(6);
    $custom_subscriber_shortcode = end($shortcodes['Subscriber']);
    expect($custom_subscriber_shortcode['text'])->equals($custom_field->name);
    expect($custom_subscriber_shortcode['shortcode'])
      ->equals('[subscriber:cf_' . $custom_field->id . ']');
  }

  function _after() {
    \ORM::raw_execute('TRUNCATE ' . CustomField::$_table);
  }
}