drupal-media / media

src/Tests/DocumentBundleTest.php

<?php

namespace Drupal\media\Tests;

use Drupal\simpletest\WebTestBase;

/**
 * Ensures that media bundle for document can be created.
 *
 * @group media
 */
class DocumentBundleTest extends WebTestBase {
  /**
   * Exempt from strict schema checking.
   *
   * @see \Drupal\Core\Config\Testing\ConfigSchemaChecker
   *
   * @var bool
   */
  protected $strictConfigSchema = FALSE;

  /**
   * Modules to enable.
   *
   * @var array
   */
  public static $modules = [
    'media',
    'media_entity',
    'media_entity_document',
    'node',
    'editor',
  ];

  /**
   * The test media bundle.
   *
   * @var \Drupal\media_entity\MediaBundleInterface
   */
  protected $testBundle;

  /**
   * {@inheritdoc}
   */
  protected function setUp() {
    parent::setUp();
    $this->testBundle = $this->container->get('entity_type.manager')->getStorage('media_bundle')->load('document');

    $adminUser = $this->drupalCreateUser([
      'view media',
      'create media',
      'update media',
      'update any media',
      'delete media',
      'delete any media',
      'access media overview',
    ]);
    $this->drupalLogin($adminUser);
  }

  /**
   * Tests document media bundle creation from config files.
   */
  public function testMediaBundleCreationFromModule() {

This method seems to be duplicated in your project.

    $type_configuration = [
      'source_field' => 'field_document',
    ];
    $field_map = [
      'mime' => 'field_mime_type',
      'size' => 'field_document_size',
    ];

    $this->assertTrue((bool) $this->testBundle, 'The media bundle from default configuration has been created in the database.');
    $this->assertEqual($this->testBundle->get('label'), 'Document', 'Correct label detected.');
    $this->assertEqual($this->testBundle->get('description'), 'Use Document for uploading document files such as PDF.', 'Correct description detected.');
    $this->assertEqual($this->testBundle->get('type'), 'document', 'Correct plugin ID detected.');
    $this->assertEqual($this->testBundle->get('type_configuration'), $type_configuration, 'Type configuration correct.');
    $this->assertEqual($this->testBundle->get('field_map'), $field_map, 'Correct field map detected.');
  }

  /**
   * Tests thumbnails of the document items.
   */
  public function testDocumentItemThumbnail() {
    // Array of test files and corresponding file icons.
    $files = [
      'Test.pdf' => 'public://media-icons/generic/application-pdf.png',
      'Test.doc' => 'public://media-icons/generic/application-msword.png',
      'Test.docx' => 'public://media-icons/generic/application-vnd.openxmlformats-officedocument.wordprocessingml.document.png',
      'Test.ods' => 'public://media-icons/generic/application-vnd.oasis.opendocument.spreadsheet.png',
      'Test.odt' => 'public://media-icons/generic/application-vnd.oasis.opendocument.text.png',
      'Test.ott' => 'public://media-icons/generic/application-vnd.oasis.opendocument.text-template.png',
      'Test.ppt' => 'public://media-icons/generic/application-vnd.ms-powerpoint.png',
      'Test.pptx' => 'public://media-icons/generic/application-vnd.openxmlformats-officedocument.presentationml.presentation.png',
      'Test.rtf' => 'public://media-icons/generic/application-rtf.png',
      'Test.txt' => 'public://media-icons/generic/text-plain.png',
      'Test.xls' => 'public://media-icons/generic/application-vnd.ms-excel.png',
      'Test.xlsx' => 'public://media-icons/generic/application-vnd.openxmlformats-officedocument.spreadsheetml.sheet.png',
    ];

    foreach ($files as $fileName => $thumbnail) {
      $file = drupal_get_path('module', 'media') . '/files/' . $fileName;
      $name = $this->randomMachineName();
      $this->drupalGet('media/add/document');
      $edit = [
        'files[field_document_0]' => $file,
      ];
      $this->drupalPostAjaxForm(NULL, $edit, "field_document_0_upload_button");
      $fid = (string) current($this->xpath('//input[@data-drupal-selector="edit-field-document-0-fids"]/@value'));
      $edit = [
        'name[0][value]' => $name,
        'form_id' => 'media_document_form',
        'field_document[0][fids]' => $fid,
        'field_document[0][display]' => 1,
      ];
      $this->drupalPostForm(NULL, $edit, t('Save and publish'));
      $recentThumbnail = $this->getMostRecentThumbnail();
      $this->assertEqual($thumbnail, $recentThumbnail, "Correct thumbnail detected for " . $fileName);
    }
  }

  /**
   * Returns the thumbnail of the most recent document.
   *
   * @return string
   *   Path of the thumbnail.
   */
  public function getMostRecentThumbnail() {
    $document_id = $this->container->get('entity.query')->get('media')->condition('bundle', 'document')->sort('created', 'DESC')->execute();
    $item = $this->container->get('entity_type.manager')
      ->getStorage('media')
      ->loadUnchanged(reset($document_id));
    return $item->getType()->thumbnail($item);
  }

}