www.aleph1.co.uk Git - yaffs-website/blob - web/modules/contrib/media_entity/src/Tests/MediaTestTrait.php

   1 <?php
   2
   3 namespace Drupal\media_entity\Tests;
   4
   5 use Drupal\media_entity\Entity\MediaBundle;
   6
   7 /**
   8  * Provides common functionality for media entity test classes.
   9  */
  10 trait MediaTestTrait {
  11
  12   /**
  13    * Creates media bundle.
  14    *
  15    * @param array $values
  16    *   The media bundle values.
  17    * @param string $type_name
  18    *   (optional) The media type provider plugin that is responsible for
  19    *   additional logic related to this media).
  20    *
  21    * @return \Drupal\Core\Entity\EntityInterface
  22    *   Returns newly created media bundle.
  23    */
  24   protected function drupalCreateMediaBundle(array $values = [], $type_name = 'generic') {
  25     if (!isset($values['bundle'])) {
  26       $id = strtolower($this->randomMachineName());
  27     }
  28     else {
  29       $id = $values['bundle'];
  30     }
  31     $values += [
  32       'id' => $id,
  33       'label' => $id,
  34       'type' => $type_name,
  35       'type_configuration' => [],
  36       'field_map' => [],
  37       'new_revision' => FALSE,
  38     ];
  39
  40     $bundle = MediaBundle::create($values);
  41     $status = $bundle->save();
  42
  43     $this->assertEqual($status, SAVED_NEW, t('Created media bundle %bundle.', ['%bundle' => $bundle->id()]));
  44
  45     return $bundle;
  46   }
  47
  48 }