View source
<?php
namespace Drupal\Tests\media\Kernel;
use Drupal\file\Entity\File;
use Drupal\KernelTests\KernelTestBase;
use Drupal\media\Entity\Media;
use Drupal\media\MediaTypeInterface;
use Drupal\Tests\media\Traits\MediaTypeCreationTrait;
use Drupal\user\Entity\User;
use org\bovigo\vfs\vfsStream;
abstract class MediaKernelTestBase extends KernelTestBase {
use MediaTypeCreationTrait;
protected static $modules = [
'media',
'media_test_source',
'image',
'user',
'field',
'system',
'file',
];
protected $testMediaType;
protected $testConstraintsMediaType;
protected $user;
protected function setUp() : void {
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',
]);
$this->testMediaType = $this
->createMediaType('test');
$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);
}
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(),
],
]);
}
}