--- /dev/null
+<?php
+
+namespace Drupal\Tests\media\Functional;
+
+use Drupal\media\Entity\Media;
+
+/**
+ * Tests media template suggestions.
+ *
+ * @group media
+ */
+class MediaTemplateSuggestionsTest extends MediaFunctionalTestBase {
+
+ /**
+ * Modules to install.
+ *
+ * @var array
+ */
+ public static $modules = ['media'];
+
+ /**
+ * Tests template suggestions from media_theme_suggestions_media().
+ */
+ public function testMediaThemeHookSuggestions() {
+ $media_type = $this->createMediaType([
+ 'new_revision' => FALSE,
+ 'queue_thumbnail_downloads' => FALSE,
+ ]);
+
+ // Create media item to be rendered.
+ $media = Media::create([
+ 'bundle' => $media_type->id(),
+ 'name' => 'Unnamed',
+ ]);
+ $media->save();
+ $view_mode = 'full';
+
+ // Simulate theming of the media item.
+ $build = \Drupal::entityTypeManager()->getViewBuilder('media')->view($media, $view_mode);
+
+ $variables['elements'] = $build;
+ $suggestions = \Drupal::moduleHandler()->invokeAll('theme_suggestions_media', [$variables]);
+ $this->assertEquals($suggestions, ['media__full', 'media__' . $media_type->id(), 'media__' . $media_type->id() . '__full'], 'Found expected media suggestions.');
+ }
+
+}