function SaveUploadFormTest::testHandleExtension

Tests extension handling.

File

core/modules/file/tests/src/Functional/SaveUploadFormTest.php, line 152

Class

SaveUploadFormTest
Tests the _file_save_upload_from_form() function.

Namespace

Drupal\Tests\file\Functional

Code

public function testHandleExtension() {
  /** @var \Drupal\Core\File\FileSystemInterface $file_system */
  $file_system = \Drupal::service('file_system');
  // The file being tested is a .gif which is in the default safe list
  // of extensions to allow when the extension validator isn't used. This is
  // implicitly tested at the testNormal() test. Here we tell
  // _file_save_upload_from_form() to only allow ".foo".
  $extensions = 'foo';
  $edit = [
    'file_test_replace' => FileSystemInterface::EXISTS_REPLACE,
    'files[file_test_upload][]' => $file_system->realpath($this->image
      ->getFileUri()),
    'extensions' => $extensions,
  ];
  $this->drupalPostForm('file-test/save_upload_from_form_test', $edit, t('Submit'));
  $this->assertSession()
    ->statusCodeEquals(200);
  $message = t('Only files with the following extensions are allowed:') . ' <em class="placeholder">' . $extensions . '</em>';
  $this->assertRaw($message, 'Cannot upload a disallowed extension');
  $this->assertRaw(t('Epic upload FAIL!'), 'Found the failure message.');
  // Check that the correct hooks were called.
  $this->assertFileHooksCalled([
    'validate',
  ]);
  // Reset the hook counters.
  file_test_reset();
  $extensions = 'foo ' . $this->imageExtension;
  // Now tell _file_save_upload_from_form() to allow the extension of our test image.
  $edit = [
    'file_test_replace' => FileSystemInterface::EXISTS_REPLACE,
    'files[file_test_upload][]' => $file_system->realpath($this->image
      ->getFileUri()),
    'extensions' => $extensions,
  ];
  $this->drupalPostForm('file-test/save_upload_from_form_test', $edit, t('Submit'));
  $this->assertSession()
    ->statusCodeEquals(200);
  $this->assertNoRaw(t('Only files with the following extensions are allowed:'), 'Can upload an allowed extension.');
  $this->assertRaw(t('You WIN!'), 'Found the success message.');
  // Check that the correct hooks were called.
  $this->assertFileHooksCalled([
    'validate',
    'load',
    'update',
  ]);
  // Reset the hook counters.
  file_test_reset();
  // Now tell _file_save_upload_from_form() to allow any extension.
  $edit = [
    'file_test_replace' => FileSystemInterface::EXISTS_REPLACE,
    'files[file_test_upload][]' => $file_system->realpath($this->image
      ->getFileUri()),
    'allow_all_extensions' => 'empty_array',
  ];
  $this->drupalPostForm('file-test/save_upload_from_form_test', $edit, t('Submit'));
  $this->assertSession()
    ->statusCodeEquals(200);
  $this->assertNoRaw(t('Only files with the following extensions are allowed:'), 'Can upload any extension.');
  $this->assertRaw(t('You WIN!'), 'Found the success message.');
  // Check that the correct hooks were called.
  $this->assertFileHooksCalled([
    'validate',
    'load',
    'update',
  ]);
}

Buggy or inaccurate documentation? Please file an issue. Need support? Need help programming? Connect with the Drupal community.