Club Drupal

/**

 * Defines base class for media_wysiwyg_view_mode test cases.

 */

class MediaWYSIWYGViewModeTestHelper extends MediaWYSIWYGTestHelper {

  function setUp() {

    parent::setUp();

    $web_user = $this->drupalCreateUser(array('access administration pages', 'administer file types', 'view files', 'use media wysiwyg'));

    $this->drupalLogin($web_user);

  }

}

/**

 * Test configuring view modes available on the format form.

 */

class FormatFormViewModesTest extends MediaWYSIWYGViewModeTestHelper {

  public static function getInfo() {

    return array(

      'name' => 'Format Form WYSIWYG View Modes',

      'description' => 'Test configuring view modes available on the format form.',

      'group' => 'Media WYSIWYG',

    );

  }

  function setUp() {

    parent::setUp();

  }

  /**

   * Configure format form view mode restrictions and ensure that they are followed.

   */

  function testAllowedFormatFormViewModes() {

    // Create an image file to test with.

    $files = $this->drupalGetTestFiles('image');

    $files[0]->status = FILE_STATUS_PERMANENT;

    $file = file_save($files[0]);

    $fid = $file->fid;

    // Teaser view mode should be selectable.

    $this->drupalGet('media/' . $fid . '/format-form');

    $this->assertResponse(200);

    $this->assertOption('edit-format', 'teaser');

    // Restrict the use of the default view mode.

    $this->drupalGet('admin/structure/file-types/manage/image/file-display/teaser');

    $this->assertResponse(200);

    $edit = array(

      'restrict_wysiwyg' => 1,

    );

    $this->drupalPost(NULL, $edit, t('Save configuration'));

    $this->assertResponse(200);

    // Teaser view mode should be restricted.

    $this->drupalGet('media/' . $fid . '/format-form');

    $this->assertResponse(200);

    $this->assertNoOption('edit-format', 'teaser');

  }

  /**

   * Assert that a select option in the current page exists.

   *

   * @param $id

   *   id of select field to assert.

   * @param $option

   *   Option to assert.

   */

  function assertOption($id, $option, $message = '', $group = 'Browser') {

    $options = $this->xpath('//select[@id=:id]/option[@value=:option]', array(':id' => $id, ':option' => $option));

    return $this->assertTrue(isset($options[0]), $message ? $message : t('Option @option for field @id exists.', array('@option' => $option, '@id' => $id)));

  }

  /**

   * Assert that a select option in the current page does not exist.

   *

   * @param $id

   *   id of select field to assert.

   * @param $option

   *   Option to assert.

   */

  function assertNoOption($id, $option, $message = '', $group = 'Browser') {

    $options = $this->xpath('//select[@id=:id]/option[@value=:option]', array(':id' => $id, ':option' => $option));

    return $this->assertFalse(isset($options[0]), $message ? $message : t('Option @option for field @id does not exist.', array('@option' => $option, '@id' => $id)));

  }

}