diff core/modules/media/tests/src/Kernel/MediaKernelTestBase.php @ 0:c75dbcec494b

Initial commit from drush-created site

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/core/modules/media/tests/src/Kernel/MediaKernelTestBase.php	Thu Jul 05 14:24:15 2018 +0000
@@ -0,0 +1,150 @@
+<?php
+
+namespace Drupal\Tests\media\Kernel;
+
+use Drupal\file\Entity\File;
+use Drupal\KernelTests\KernelTestBase;
+use Drupal\media\Entity\Media;
+use Drupal\media\Entity\MediaType;
+use Drupal\media\MediaTypeInterface;
+use Drupal\user\Entity\User;
+use org\bovigo\vfs\vfsStream;
+
+/**
+ * Base class for Media kernel tests.
+ */
+abstract class MediaKernelTestBase extends KernelTestBase {
+
+  /**
+   * Modules to install.
+   *
+   * @var array
+   */
+  public static $modules = [
+    'media',
+    'media_test_source',
+    'image',
+    'user',
+    'field',
+    'system',
+    'file',
+  ];
+
+  /**
+   * The test media type.
+   *
+   * @var \Drupal\media\MediaTypeInterface
+   */
+  protected $testMediaType;
+
+  /**
+   * The test media type with constraints.
+   *
+   * @var \Drupal\media\MediaTypeInterface
+   */
+  protected $testConstraintsMediaType;
+
+  /**
+   * A user.
+   *
+   * @var \Drupal\user\UserInterface
+   */
+  protected $user;
+
+  /**
+   * {@inheritdoc}
+   */
+  protected function setUp() {
+    parent::setUp();
+
+    $this->installEntitySchema('user');
+    $this->installEntitySchema('file');
+    $this->installSchema('file', 'file_usage');
+    $this->installSchema('system', 'sequences');
+    $this->installEntitySchema('media');
+    $this->installConfig(['field', 'system', 'image', 'file', 'media']);
+
+    // Create a test media type.
+    $this->testMediaType = $this->createMediaType('test');
+    // Create a test media type with constraints.
+    $this->testConstraintsMediaType = $this->createMediaType('test_constraints');
+
+    $this->user = User::create([
+      'name' => 'username',
+      'status' => 1,
+    ]);
+    $this->user->save();
+    $this->container->get('current_user')->setAccount($this->user);
+  }
+
+  /**
+   * Create a media type for a source plugin.
+   *
+   * @param string $media_source_name
+   *   The name of the media source.
+   *
+   * @return \Drupal\media\MediaTypeInterface
+   *   A media type.
+   */
+  protected function createMediaType($media_source_name) {
+    $id = strtolower($this->randomMachineName());
+    $media_type = MediaType::create([
+      'id' => $id,
+      'label' => $id,
+      'source' => $media_source_name,
+      'new_revision' => FALSE,
+    ]);
+    $media_type->save();
+    $source_field = $media_type->getSource()->createSourceField($media_type);
+    // The media type form creates a source field if it does not exist yet. The
+    // same must be done in a kernel test, since it does not use that form.
+    // @see \Drupal\media\MediaTypeForm::save()
+    $source_field->getFieldStorageDefinition()->save();
+    // The source field storage has been created, now the field can be saved.
+    $source_field->save();
+    $media_type->set('source_configuration', [
+      'source_field' => $source_field->getName(),
+    ])->save();
+    return $media_type;
+  }
+
+  /**
+   * Helper to generate a media item.
+   *
+   * @param string $filename
+   *   String filename with extension.
+   * @param \Drupal\media\MediaTypeInterface $media_type
+   *   The the media type.
+   *
+   * @return \Drupal\media\Entity\Media
+   *   A media item.
+   */
+  protected function generateMedia($filename, MediaTypeInterface $media_type) {
+    vfsStream::setup('drupal_root');
+    vfsStream::create([
+      'sites' => [
+        'default' => [
+          'files' => [
+            $filename => str_repeat('a', 3000),
+          ],
+        ],
+      ],
+    ]);
+
+    $file = File::create([
+      'uri' => 'vfs://drupal_root/sites/default/files/' . $filename,
+      'uid' => $this->user->id(),
+    ]);
+    $file->setPermanent();
+    $file->save();
+
+    return Media::create([
+      'bundle' => $media_type->id(),
+      'name' => 'Mr. Jones',
+      'field_media_file' => [
+        'target_id' => $file->id(),
+      ],
+    ]);
+  }
+
+}