MediaSourceFileTest.php in Drupal 9

Namespace

Drupal\Tests\media\Kernel

File

core/modules/media/tests/src/Kernel/MediaSourceFileTest.php

<?php

namespace Drupal\Tests\media\Kernel;


/**
 * Tests the file media source.
 *
 * @group media
 */
class MediaSourceFileTest extends MediaKernelTestBase {

  /**
   * Tests the file extension constraint.
   */
  public function testFileExtensionConstraint() {
    $mediaType = $this
      ->createMediaType('file');

    // Create a random file that should fail.
    $media = $this
      ->generateMedia('test.patch', $mediaType);
    $result = $media
      ->validate();
    $this
      ->assertCount(1, $result);
    $this
      ->assertSame('field_media_file.0', $result
      ->get(0)
      ->getPropertyPath());
    $this
      ->assertStringContainsString('Only files with the following extensions are allowed:', (string) $result
      ->get(0)
      ->getMessage());

    // Create a random file that should pass.
    $media = $this
      ->generateMedia('test.txt', $mediaType);
    $result = $media
      ->validate();
    $this
      ->assertCount(0, $result);
  }

  /**
   * Tests a media file can be deleted.
   */
  public function testFileDeletion() {
    $mediaType = $this
      ->createMediaType('file');
    $media = $this
      ->generateMedia('test.txt', $mediaType);
    $media
      ->save();
    $source_field_name = $mediaType
      ->getSource()
      ->getSourceFieldDefinition($mediaType)
      ->getName();

    /** @var \Drupal\file\FileInterface $file */
    $file = $media
      ->get($source_field_name)->entity;
    $file
      ->delete();
    $this
      ->assertEmpty($this->container
      ->get('entity_type.manager')
      ->getStorage('file')
      ->loadByProperties([
      'filename' => 'test.txt',
    ]));
  }

}

Classes

Name	Description
MediaSourceFileTest	Tests the file media source.