use Drupal\Core\Entity\EntityInterface;
use Drupal\media\Entity\Media;
-use Drupal\system\Tests\Entity\EntityWithUriCacheTagsTestBase;
+use Drupal\Tests\media\Traits\MediaTypeCreationTrait;
+use Drupal\Tests\system\Functional\Entity\EntityWithUriCacheTagsTestBase;
/**
* Tests the media items cache tags.
*/
class MediaCacheTagsTest extends EntityWithUriCacheTagsTestBase {
- use MediaFunctionalTestCreateMediaTypeTrait;
+ use MediaTypeCreationTrait;
/**
* {@inheritdoc}
*/
protected function createEntity() {
// Create a media type.
- $mediaType = $this->createMediaType();
+ $mediaType = $this->createMediaType('test');
// Create a media item.
$media = Media::create([