<?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->assertContains('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']));
  }

}
