Version 1

[yaffs-website] / web / modules / contrib / media_entity / src / Tests / MediaTestTrait.php

diff --git a/web/modules/contrib/media_entity/src/Tests/MediaTestTrait.php b/web/modules/contrib/media_entity/src/Tests/MediaTestTrait.php
new file mode 100644 (file)
index 0000000..f7646aa
--- /dev/null
+++ b/web/modules/contrib/media_entity/src/Tests/MediaTestTrait.php
@@ -0,0 +1,48 @@
+<?php
+
+namespace Drupal\media_entity\Tests;
+
+use Drupal\media_entity\Entity\MediaBundle;
+
+/**
+ * Provides common functionality for media entity test classes.
+ */
+trait MediaTestTrait {
+
+  /**
+   * Creates media bundle.
+   *
+   * @param array $values
+   *   The media bundle values.
+   * @param string $type_name
+   *   (optional) The media type provider plugin that is responsible for
+   *   additional logic related to this media).
+   *
+   * @return \Drupal\Core\Entity\EntityInterface
+   *   Returns newly created media bundle.
+   */
+  protected function drupalCreateMediaBundle(array $values = [], $type_name = 'generic') {
+    if (!isset($values['bundle'])) {
+      $id = strtolower($this->randomMachineName());
+    }
+    else {
+      $id = $values['bundle'];
+    }
+    $values += [
+      'id' => $id,
+      'label' => $id,
+      'type' => $type_name,
+      'type_configuration' => [],
+      'field_map' => [],
+      'new_revision' => FALSE,
+    ];
+
+    $bundle = MediaBundle::create($values);
+    $status = $bundle->save();
+
+    $this->assertEqual($status, SAVED_NEW, t('Created media bundle %bundle.', ['%bundle' => $bundle->id()]));
+
+    return $bundle;
+  }
+
+}