Tests file extension checking.

File

drupal-7.x/modules/file/tests/file.test, line 919

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);
}