MediaThumbnailFormatterTest.php

<?php

namespace Drupal\Tests\media\Functional\FieldFormatter;

use Drupal\field\Entity\FieldConfig;
use Drupal\field\Entity\FieldStorageConfig;
use Drupal\file\Entity\File;
use Drupal\media\Entity\Media;
use Drupal\Tests\media\Functional\MediaFunctionalTestBase;
use Drupal\Tests\TestFileCreationTrait;

/**
 * @covers \Drupal\media\Plugin\Field\FieldFormatter\MediaThumbnailFormatter
 *
 * @group media
 */
class MediaThumbnailFormatterTest extends MediaFunctionalTestBase {
  use TestFileCreationTrait;
  
  /**
   * {@inheritdoc}
   */
  protected $defaultTheme = 'stark';
  
  /**
   * Tests the media thumbnail field formatter.
   */
  public function testRender() {
    $this->drupalLogin($this->adminUser);
    /** @var \Drupal\node\NodeStorage $node_storage */
    $node_storage = $this->container
      ->get('entity_type.manager')
      ->getStorage('node');
    // Create an image media type for testing the formatter.
    $this->createMediaType('image', [
      'id' => 'image',
    ]);
    // Create an article content type.
    $this->drupalCreateContentType([
      'type' => 'article',
      'name' => 'Article',
    ]);
    // Creates an entity reference field for media.
    $field_storage = FieldStorageConfig::create([
      'field_name' => 'field_media_reference',
      'type' => 'entity_reference',
      'entity_type' => 'node',
      'cardinality' => 1,
      'settings' => [
        'target_type' => 'media',
      ],
    ]);
    $field_storage->save();
    FieldConfig::create([
      'field_storage' => $field_storage,
      'bundle' => 'article',
      'label' => 'Reference media',
      'translatable' => FALSE,
    ])->save();
    // Alter the form display.
    $this->container
      ->get('entity_display.repository')
      ->getFormDisplay('node', 'article')
      ->setComponent('field_media_reference', [
      'type' => 'entity_reference_autocomplete',
    ])
      ->save();
    // Change the image thumbnail to point into the media.
    $this->changeMediaReferenceFieldLinkType('media');
    // Create and upload a file to the media.
    $file = File::create([
      'uri' => current($this->getTestFiles('image'))->uri,
    ]);
    $file->save();
    $mediaImage = Media::create([
      'bundle' => 'image',
      'name' => 'Test image',
      'field_media_image' => $file->id(),
    ]);
    $mediaImage->save();
    // Save the article node.
    $title = $this->randomMachineName();
    $edit = [
      'title[0][value]' => $title,
    ];
    $edit['field_media_reference[0][target_id]'] = $mediaImage->getName();
    $this->drupalGet('node/add/article');
    $this->submitForm($edit, 'Save');
    // Validate the image being loaded with the media reference.
    $this->assertSession()
      ->responseContains('<a href="' . $mediaImage->toUrl('edit-form')
      ->toString());
    // Retrieve the created node.
    $node = $this->drupalGetNodeByTitle($title);
    $nid = $node->id();
    // Change the image thumbnail to point into the content node.
    $this->changeMediaReferenceFieldLinkType('content');
    $node_storage->resetCache([
      $nid,
    ]);
    $this->drupalGet('node/' . $nid);
    // Validate image being loaded with the content on the link.
    $this->assertSession()
      ->responseContains('<a href="' . $node->toUrl()
      ->toString());
    $this->assertSession()
      ->responseContains('loading="eager"');
  }
  
  /**
   * Helper function to change field display.
   *
   * @param string $type
   *   Image link type.
   */
  private function changeMediaReferenceFieldLinkType(string $type) : void {
    // Change the display to use the media thumbnail formatter with image link.
    $this->container
      ->get('entity_display.repository')
      ->getViewDisplay('node', 'article', 'default')
      ->setComponent('field_media_reference', [
      'type' => 'media_thumbnail',
      'settings' => [
        'image_link' => $type,
        'image_style' => '',
        'image_loading' => [
          'attribute' => 'eager',
        ],
      ],
    ])
      ->save();
  }

}