Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\Tests\media\Kernel;
Chris@0: 
Chris@0: use Drupal\file\Entity\File;
Chris@0: use Drupal\KernelTests\KernelTestBase;
Chris@0: use Drupal\media\Entity\Media;
Chris@0: use Drupal\media\MediaTypeInterface;
Chris@17: use Drupal\Tests\media\Traits\MediaTypeCreationTrait;
Chris@0: use Drupal\user\Entity\User;
Chris@0: use org\bovigo\vfs\vfsStream;
Chris@0: 
Chris@0: /**
Chris@0:  * Base class for Media kernel tests.
Chris@0:  */
Chris@0: abstract class MediaKernelTestBase extends KernelTestBase {
Chris@0: 
Chris@17:   use MediaTypeCreationTrait;
Chris@17: 
Chris@0:   /**
Chris@0:    * Modules to install.
Chris@0:    *
Chris@0:    * @var array
Chris@0:    */
Chris@0:   public static $modules = [
Chris@0:     'media',
Chris@0:     'media_test_source',
Chris@0:     'image',
Chris@0:     'user',
Chris@0:     'field',
Chris@0:     'system',
Chris@0:     'file',
Chris@0:   ];
Chris@0: 
Chris@0:   /**
Chris@0:    * The test media type.
Chris@0:    *
Chris@0:    * @var \Drupal\media\MediaTypeInterface
Chris@0:    */
Chris@0:   protected $testMediaType;
Chris@0: 
Chris@0:   /**
Chris@0:    * The test media type with constraints.
Chris@0:    *
Chris@0:    * @var \Drupal\media\MediaTypeInterface
Chris@0:    */
Chris@0:   protected $testConstraintsMediaType;
Chris@0: 
Chris@0:   /**
Chris@0:    * A user.
Chris@0:    *
Chris@0:    * @var \Drupal\user\UserInterface
Chris@0:    */
Chris@0:   protected $user;
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function setUp() {
Chris@0:     parent::setUp();
Chris@0: 
Chris@0:     $this->installEntitySchema('user');
Chris@0:     $this->installEntitySchema('file');
Chris@0:     $this->installSchema('file', 'file_usage');
Chris@0:     $this->installSchema('system', 'sequences');
Chris@0:     $this->installEntitySchema('media');
Chris@0:     $this->installConfig(['field', 'system', 'image', 'file', 'media']);
Chris@0: 
Chris@0:     // Create a test media type.
Chris@0:     $this->testMediaType = $this->createMediaType('test');
Chris@0:     // Create a test media type with constraints.
Chris@0:     $this->testConstraintsMediaType = $this->createMediaType('test_constraints');
Chris@0: 
Chris@0:     $this->user = User::create([
Chris@0:       'name' => 'username',
Chris@0:       'status' => 1,
Chris@0:     ]);
Chris@0:     $this->user->save();
Chris@0:     $this->container->get('current_user')->setAccount($this->user);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@14:    * Helper to generate a media item.
Chris@0:    *
Chris@0:    * @param string $filename
Chris@0:    *   String filename with extension.
Chris@0:    * @param \Drupal\media\MediaTypeInterface $media_type
Chris@0:    *   The the media type.
Chris@0:    *
Chris@0:    * @return \Drupal\media\Entity\Media
Chris@14:    *   A media item.
Chris@0:    */
Chris@0:   protected function generateMedia($filename, MediaTypeInterface $media_type) {
Chris@0:     vfsStream::setup('drupal_root');
Chris@0:     vfsStream::create([
Chris@0:       'sites' => [
Chris@0:         'default' => [
Chris@0:           'files' => [
Chris@0:             $filename => str_repeat('a', 3000),
Chris@0:           ],
Chris@0:         ],
Chris@0:       ],
Chris@0:     ]);
Chris@0: 
Chris@0:     $file = File::create([
Chris@0:       'uri' => 'vfs://drupal_root/sites/default/files/' . $filename,
Chris@0:       'uid' => $this->user->id(),
Chris@0:     ]);
Chris@0:     $file->setPermanent();
Chris@0:     $file->save();
Chris@0: 
Chris@0:     return Media::create([
Chris@0:       'bundle' => $media_type->id(),
Chris@0:       'name' => 'Mr. Jones',
Chris@0:       'field_media_file' => [
Chris@0:         'target_id' => $file->id(),
Chris@0:       ],
Chris@0:     ]);
Chris@0:   }
Chris@0: 
Chris@0: }