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\Entity\MediaType;
Chris@0: use Drupal\media\MediaTypeInterface;
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@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@0:    * Create a media type for a source plugin.
Chris@0:    *
Chris@0:    * @param string $media_source_name
Chris@0:    *   The name of the media source.
Chris@0:    *
Chris@0:    * @return \Drupal\media\MediaTypeInterface
Chris@0:    *   A media type.
Chris@0:    */
Chris@0:   protected function createMediaType($media_source_name) {
Chris@0:     $id = strtolower($this->randomMachineName());
Chris@0:     $media_type = MediaType::create([
Chris@0:       'id' => $id,
Chris@0:       'label' => $id,
Chris@0:       'source' => $media_source_name,
Chris@0:       'new_revision' => FALSE,
Chris@0:     ]);
Chris@0:     $media_type->save();
Chris@0:     $source_field = $media_type->getSource()->createSourceField($media_type);
Chris@0:     // The media type form creates a source field if it does not exist yet. The
Chris@0:     // same must be done in a kernel test, since it does not use that form.
Chris@0:     // @see \Drupal\media\MediaTypeForm::save()
Chris@0:     $source_field->getFieldStorageDefinition()->save();
Chris@0:     // The source field storage has been created, now the field can be saved.
Chris@0:     $source_field->save();
Chris@0:     $media_type->set('source_configuration', [
Chris@0:       'source_field' => $source_field->getName(),
Chris@0:     ])->save();
Chris@0:     return $media_type;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Helper to generate media entity.
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@0:    *   A media entity.
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: }