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',
        ]));
    }

}