function FileFieldValidateTestCase::testFileExtension in Drupal 7

Tests file extension checking.

File

modules/file/tests/file.test, line 1290

Tests for file.module.

Class

FileFieldValidateTestCase

Tests various validations.

Code

function testFileExtension() {
  $type_name = 'article';
  $field_name = strtolower($this
    ->randomName());
  $this
    ->createFileField($field_name, $type_name);
  $field = field_info_field($field_name);
  $instance = field_info_instance('node', $field_name, $type_name);
  $test_file = $this
    ->getTestFile('image');
  list(, $test_file_extension) = explode('.', $test_file->filename);

  // Disable extension checking.
  $this
    ->updateFileField($field_name, $type_name, array(
    'file_extensions' => '',
  ));

  // Check that the file can be uploaded with no extension checking.
  $nid = $this
    ->uploadNodeFile($test_file, $field_name, $type_name);
  $node = node_load($nid, NULL, TRUE);
  $node_file = (object) $node->{$field_name}[LANGUAGE_NONE][0];
  $this
    ->assertFileExists($node_file, 'File exists after uploading a file with no extension checking.');
  $this
    ->assertFileEntryExists($node_file, 'File entry exists after uploading a file with no extension checking.');

  // Enable extension checking for text files.
  $this
    ->updateFileField($field_name, $type_name, array(
    'file_extensions' => 'txt',
  ));

  // Check that the file with the wrong extension cannot be uploaded.
  $nid = $this
    ->uploadNodeFile($test_file, $field_name, $type_name);
  $error_message = t('Only files with the following extensions are allowed: %files-allowed.', array(
    '%files-allowed' => 'txt',
  ));
  $this
    ->assertRaw($error_message, 'Node save failed when file uploaded with the wrong extension.');

  // Enable extension checking for text and image files.
  $this
    ->updateFileField($field_name, $type_name, array(
    'file_extensions' => "txt {$test_file_extension}",
  ));

  // Check that the file can be uploaded with extension checking.
  $nid = $this
    ->uploadNodeFile($test_file, $field_name, $type_name);
  $node = node_load($nid, NULL, TRUE);
  $node_file = (object) $node->{$field_name}[LANGUAGE_NONE][0];
  $this
    ->assertFileExists($node_file, 'File exists after uploading a file with extension checking.');
  $this
    ->assertFileEntryExists($node_file, 'File entry exists after uploading a file with extension checking.');

  // Remove our file field.
  field_delete_field($field_name);
}