MediaTrait.php in Entity Reference Hierarchy 8.2
File
modules/entity_hierarchy_microsite/tests/src/Traits/MediaTrait.php
View source
<?php
namespace Drupal\Tests\entity_hierarchy_microsite\Traits;
use Drupal\file\Entity\File;
use Drupal\media\Entity\Media;
use Drupal\Tests\media\Traits\MediaTypeCreationTrait;
use Drupal\Tests\TestFileCreationTrait;
trait MediaTrait {
use MediaTypeCreationTrait;
use TestFileCreationTrait;
protected function createImageMedia(array $values = [], File $file = NULL) {
if (!$file) {
$image = $this
->getTestFiles('image')[0];
$file = $this
->createFile($image->uri);
}
$values = $values + [
'bundle' => 'image',
'field_media_image' => [
'target_id' => $file
->id(),
],
];
$media = Media::create($values);
$media
->save();
return $media;
}
protected function createFile($uri, $status = 1) {
$file = File::create([
'uri' => $uri,
'status' => $status,
]);
$file
->save();
return $file;
}
}
Traits
Name |
Description |
MediaTrait |
Defines a class for media utilties in tests. |