upgrade.upload.test

<?php

/**
 * Upgrade test for comment.module.
 */
class UploadUpgradePathTestCase extends UpgradePathTestCase {
  public static function getInfo() {
    return array(
      'name' => 'Upload upgrade path',
      'description' => 'Upload upgrade path tests.',
      'group' => 'Upgrade path',
    );
  }
  public function setUp() {

    // Path to the database dump files.
    $this->databaseDumpFiles = array(
      drupal_get_path('module', 'simpletest') . '/tests/upgrade/drupal-6.filled.database.php',
      drupal_get_path('module', 'simpletest') . '/tests/upgrade/drupal-6.upload.database.php',
    );
    parent::setUp();

    // Set a small batch size to test multiple iterations of the batch.
    $this
      ->variable_set('upload_update_batch_size', 2);
    $this
      ->uninstallModulesExcept(array(
      'upload',
    ));
  }

  /**
   * Test a successful upgrade.
   */
  public function testUploadUpgrade() {
    $this
      ->assertTrue($this
      ->performUpgrade(), 'The upgrade was completed successfully.');
    $query = new EntityFieldQuery();
    $query
      ->entityCondition('entity_type', 'node');
    $query
      ->entityCondition('bundle', 'page');
    $query
      ->age(FIELD_LOAD_REVISION);
    $query
      ->fieldCondition('upload');
    $entities = $query
      ->execute();
    $revisions = $entities['node'];

    // Node revision 50 should not have uploaded files, as the entry in {files}
    // is corrupted.
    $this
      ->assertFalse(isset($revisions[50]), 'Nodes with missing files do not contain filefield data.');

    // Node revisions 51-53 should have uploaded files.
    $this
      ->assertTrue(isset($revisions[51]) && isset($revisions[52]) && isset($revisions[53]), 'Nodes with uploaded files now contain filefield data.');

    // The test database lists uploaded filenames in the body of each node with
    // uploaded files attached. Make sure all files are there in the same order.
    foreach ($revisions as $vid => $revision) {
      $node = node_load($revision->nid, $vid);

      // Assemble a list of the filenames as recorded in the node body before
      // the upgrade.
      $recorded_filenames = preg_split('/\\s+/', $node->body[LANGUAGE_NONE][0]['value']);

      // The first line of the node body should be "Attachments:"
      if (strstr($recorded_filenames[0], "Attachments:")) {
        unset($recorded_filenames[0]);
      }
      $recorded_filenames = array_values($recorded_filenames);
      $files = $node->upload[LANGUAGE_NONE];

      // Assemble a list of the filenames as they exist after the upgrade.
      $filenames = array();
      foreach ($files as $file) {
        $filenames[] = $file['filename'];
      }
      $this
        ->assertIdentical($filenames, $recorded_filenames, 'The uploaded files are present in the same order after the upgrade.');
    }

    // Test for the file with repeating basename to only have the streaming
    // path replaced.
    $node = node_load(40, 53);
    $repeated_basename_file = $node->upload[LANGUAGE_NONE][4];
    $this
      ->assertEqual($repeated_basename_file['uri'], 'private://drupal-6/file/directory/path/crazy-basename.png', "The file with the repeated basename path only had the stream portion replaced");

    // Ensure that filepaths are deduplicated.
    $node0 = node_load(41, 54);
    $node1 = node_load(41, 55);

    // Ensure that both revisions point to the same file ID.
    $items0 = field_get_items('node', $node0, 'upload');
    $this
      ->assertEqual(count($items0), 1);
    $items1 = field_get_items('node', $node1, 'upload');
    $this
      ->assertEqual(count($items1), 2);
    $this
      ->assertEqual($items0[0]['fid'], $items1[0]['fid']);
    $this
      ->assertEqual($items0[0]['fid'], $items1[1]['fid']);

    // The revision with more than one reference to the same file should retain
    // the original settings for each reference.
    $this
      ->assertEqual($items1[0]['description'], 'first description');
    $this
      ->assertEqual($items1[0]['display'], 0);
    $this
      ->assertEqual($items1[1]['description'], 'second description');
    $this
      ->assertEqual($items1[1]['display'], 1);

    // Ensure that the latest version of the files are used.
    $this
      ->assertEqual($items1[0]['filesize'], 316);
    $this
      ->assertEqual($items1[1]['filesize'], 316);

    // No duplicate files should remain on the Drupal 7 site.
    $this
      ->assertEqual(0, db_query("SELECT COUNT(*) FROM {file_managed} GROUP BY uri HAVING COUNT(fid) > 1")
      ->fetchField());

    // Make sure the file settings were properly migrated.
    $d6_file_directory_temp = '/drupal-6/file/directory/temp';
    $d6_file_directory_path = '/drupal-6/file/directory/path';
    $d6_file_downloads = 2;

    // FILE_DOWNLOADS_PRIVATE
    $this
      ->assertNull(variable_get('file_directory_temp', NULL), "The 'file_directory_temp' variable was properly removed.");
    $this
      ->assertEqual(variable_get('file_temporary_path', 'drupal-7-bogus'), $d6_file_directory_temp, "The 'file_temporary_path' setting was properly migrated.");
    $this
      ->assertEqual(variable_get('file_default_scheme', 'drupal-7-bogus'), 'private', "The 'file_default_scheme' setting was properly migrated.");
    $this
      ->assertEqual(variable_get('file_private_path', 'drupal-7-bogus'), $d6_file_directory_path, "The 'file_private_path' setting was properly migrated.");
  }

}