<?php

/**
 * @file
 * Tests for the file usage in entity fields with the Media filter markup.
 */

class MediaFileUsageTest extends DrupalWebTestCase {
  /**
   * Provide test information.
   */
  public static function getInfo() {
    return array(
      'name' => t('File usage tracking'),
      'description' => t('Tests tracking of usage for files in text fields.'),
      'group' => t('Media'),
    );
  }

  /**
   * Enable media and file entity modules for testing.
   */
  public function setUp() {
    parent::setUp(array('media', 'file_entity'));

    // Create and log in a user.
    $account = $this->drupalCreateUser(array('administer nodes', 'create article content'));
    $this->drupalLogin($account);
  }

  /**
    * Generates markup to be inserted for a file.
    *
    * This is a PHP version of InsertMedia.insert() from js/wysiwyg-media.js.
    *
    * @param int $fid
    *   Drupal file id
    * @param int $count
    *   Quantity of markup to insert
    *
    * @return string
    *   Filter markup.
    */
  private function generateFileMarkup($fid, $count = 1) {
    $file_usage_markup = '';

    // Build the data that is used in a media tag.
    $data = array(
      'fid' => $fid,
      'type' => 'media',
      'view_mode' => 'preview',
      'attributes' => array(
        'height' => 100,
        'width' => 100,
        'classes' => 'media-element file_preview',
      )
    );

    // Create the file usage markup.
    for ($i = 1; $i <= $count; $i++) {
      $file_usage_markup .= '<p>[[' . drupal_json_encode($data) . ']]</p>';
    }

    return $file_usage_markup;
  }

  /**
   * Utility function to create a test node.
   *
   * @param int $fid
   *   Create the node with media markup in the body field
   *
   * @return int
   *   Returns the node id
   */
  private function createNode($fid = FALSE) {
    $markup = '';
    if (! empty($fid)) {
      $markup = $this->generateFileMarkup($fid);
    }

    // Create an article node with file markup in the body field.
    $edit = array(
      'title' => $this->randomName(8),
      'body[und][0][value]' => $markup,
    );
    // Save the article node. First argument is the URL, then the value array
    // and the third is the label the button that should be "clicked".
    $this->drupalPost('node/add/article', $edit, t('Save'));

    // Get the article node that was saved by the unique title.
    $node = $this->drupalGetNodeByTitle($edit['title']);
    return $node->nid;
  }

  /**
   * Tests the tracking of file usages for files submitted via the WYSIWYG editor.
   */
  public function testFileUsageIncrementing() {
    // Create a file.
    $files = $this->drupalGetTestFiles('image');
    $file = file_save(file_uri_to_object($files[0]->uri));
    $fid = $file->fid;

    // There should be zero usages of this file prior to node creation,
    $file_uses = file_usage_list($file);
    $this->assertEqual(empty($file_uses), TRUE, t('Created a new file with zero uses.'));

    // Create a node to test with.
    $nid = $this->createNode($fid);

    // Get the new file usage count.
    $file_uses = file_usage_list($file);

    $this->assertEqual($file_uses['media']['node'][$nid], 1, t('File usage increases when added to a new node.'));

    // Create a new revision that has the file on it. File usage will be 2.
    $node = node_load($nid);
    $node->revision = TRUE;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $revisions = count(node_revision_list($node));
    // Keep track of this VID to test deletion later on.
    $delete_one = $node->vid;

    $this->assertEqual($revisions, 2, t('Node save created a second revision'));
    $this->assertEqual($file_uses['media']['node'][$nid], 2, t('File usage incremented with a new node revision.'));

    // Create a new revision that has two instances of the file. File usage will
    // be 4.
    $node = node_load($nid);
    $node->body[LANGUAGE_NONE][0]['value'] = $this->generateFileMarkup($fid, 2);
    $node->revision = TRUE;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $revisions = count(node_revision_list($node));
    // Keep track of this VID to test deletion later on.
    $delete_two = $node->vid;

    $this->assertEqual($revisions, 3, t('Node save created a third revision.'));
    $this->assertEqual($file_uses['media']['node'][$nid], 4, t('File usage incremented with multiple files and a new node revision.'));

    // Create a new revision that has no file on it. File usage will be 4.
    $node = node_load($nid);
    $node->body[LANGUAGE_NONE][0]['value'] = '';
    $node->revision = TRUE;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $revisions = count(node_revision_list($node));
    // Keep track of this VID to test deletion later on.
    $delete_zero = $node->vid;

    $this->assertEqual($revisions, 4, t('Node save created a fourth revision.'));
    $this->assertEqual($file_uses['media']['node'][$nid], 4, t('File usage does not change with a new revision of the node without the file'));

    // Create a new revision that has the file on it. File usage will be 5.
    $node = node_load($nid);
    $node->body[LANGUAGE_NONE][0]['value'] = $this->generateFileMarkup($fid, 1);
    $node->revision = TRUE;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $revisions = count(node_revision_list($node));

    $this->assertEqual($revisions, 5, t('Node save created a new revision.'));
    $this->assertEqual($file_uses['media']['node'][$nid], 5, t('File usage incremented with a single file on a new node revision.'));

    // Delete a revision that has the file on it once. File usage will be 4.
    node_revision_delete($delete_one);
    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $this->assertEqual($file_uses['media']['node'][$nid], 4, t('Deleting revision with file decreases file usage'));

    // Delete a revision that has no file on it. File usage will be 4.
    node_revision_delete($delete_zero);
    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $this->assertEqual($file_uses['media']['node'][$nid], 4, t('Deleting revision without a file does not change file usage.'));

    // Delete a revision that has the file on it twice. File usage will be 2.
    node_revision_delete($delete_two);
    $node = node_load($nid);
    $file_uses = file_usage_list($file);
    $this->assertEqual($file_uses['media']['node'][$nid], 2, t('Deleting revision with file decreases file usage'));

    // Create a new revision with the file on it twice. File usage will be 4.
    $node = node_load($nid);
    $node->body[LANGUAGE_NONE][0]['value'] = $this->generateFileMarkup($fid, 2);
    $node->revision = TRUE;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);

    $this->assertEqual($file_uses['media']['node'][$nid], 4,  t('File usage incremented with files on a new node revision.'));

    // Re-save current revision with file on it once instead of twice. File
    // usage will be 3.
    $node = node_load($nid);
    $node->body[LANGUAGE_NONE][0]['value'] = $this->generateFileMarkup($fid, 1);
    $saved_vid = $node->vid;
    node_save($node);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);

    $this->assertEqual($node->vid, $saved_vid, t('Resaved node revision does not create new revision.'));
    $this->assertEqual($file_uses['media']['node'][$nid], 3, t('Resaved node revision with fewer files reduces file usage.'));

    // Delete the node. File usage will be 0.
    $node = node_load($nid);
    node_delete($nid);

    $node = node_load($nid);
    $file_uses = file_usage_list($file);

    $this->assertEqual(empty($node), TRUE, t('Node has been deleted.'));
    $this->assertEqual(empty($file_uses), TRUE, t('Deleting the node removes all file uses.'));
  }

  /**
   * Tests the behavior of node and file deletion.
   */
  public function testFileUsageIncrementingDelete() {
    // Create a node with file markup in the body field with a new file.
    $files = $this->drupalGetTestFiles('image');
    $file = file_save(file_uri_to_object($files[1]->uri));
    $fid = $file->fid;
    $file_uses = file_usage_list($file);

    $this->assertEqual(empty($file_uses), TRUE, t('Created a new file with zero uses.'));

    // Create a new node with file markup.
    $nid = $this->createNode($fid);
    $file_uses = file_usage_list($file);

    $this->assertEqual($file_uses['media']['node'][$nid], 1, t('Incremented file usage on node save.'));

    // Try to delete the file. file_delete() should return file_usage().
    $deleted = file_delete($file);
    $this->assertTrue(is_array($deleted), t('File cannot be deleted while in use by a node.'));

    // Delete the node.
    node_delete($nid);
    $node = node_load($nid);
    $file_uses = file_usage_list($file);

    $this->assertEqual(empty($node), TRUE, t('Node has been deleted.'));
    $this->assertEqual(empty($file_uses), TRUE, t('File has zero usage after node is deleted.'));

    $deleted = file_delete($file);
    $this->assertTrue($deleted, t('File can be deleted with no usage.'));

    $file = file_load($fid);
    $this->assertTrue(empty($file), t('File no longer exists after delete.'));
  }
}