SpaceUsedTest.php in Zircon Profile 8

Contains \Drupal\file\Tests\SpaceUsedTest.

Namespace

Drupal\file\Tests

File

core/modules/file/src/Tests/SpaceUsedTest.php

<?php

/**
 * @file
 * Contains \Drupal\file\Tests\SpaceUsedTest.
 */
namespace Drupal\file\Tests;


/**
 * Tests the spaceUsed() function.
 *
 * @group file
 */
class SpaceUsedTest extends FileManagedUnitTestBase {
  protected function setUp() {
    parent::setUp();

    // Create records for a couple of users with different sizes.
    $this
      ->createFileWithSize('public://example1.txt', 50, 2);
    $this
      ->createFileWithSize('public://example2.txt', 20, 2);
    $this
      ->createFileWithSize('public://example3.txt', 100, 3);
    $this
      ->createFileWithSize('public://example4.txt', 200, 3);

    // Now create some non-permanent files.
    $this
      ->createFileWithSize('public://example5.txt', 1, 2, 0);
    $this
      ->createFileWithSize('public://example6.txt', 3, 3, 0);
  }

  /**
   * Creates a file with a given size.
   *
   * @param string $uri
   *   URI of the file to create.
   * @param int $size
   *   Size of the file.
   * @param int $uid
   *   File owner ID.
   * @param int $status
   *   Whether the file should be permanent or temporary.
   *
   * @return \Drupal\Core\Entity\EntityInterface
   *   The file entity.
   */
  protected function createFileWithSize($uri, $size, $uid, $status = FILE_STATUS_PERMANENT) {
    file_put_contents($uri, $this
      ->randomMachineName($size));
    $file = entity_create('file', array(
      'uri' => $uri,
      'uid' => $uid,
      'status' => $status,
    ));
    $file
      ->save();
    return $file;
  }

  /**
   * Test different users with the default status.
   */
  function testFileSpaceUsed() {
    $file = $this->container
      ->get('entity.manager')
      ->getStorage('file');

    // Test different users with default status.
    $this
      ->assertEqual($file
      ->spaceUsed(2), 70);
    $this
      ->assertEqual($file
      ->spaceUsed(3), 300);
    $this
      ->assertEqual($file
      ->spaceUsed(), 370);

    // Test the status fields
    $this
      ->assertEqual($file
      ->spaceUsed(NULL, 0), 4);
    $this
      ->assertEqual($file
      ->spaceUsed(NULL, FILE_STATUS_PERMANENT), 370);

    // Test both the user and status.
    $this
      ->assertEqual($file
      ->spaceUsed(1, 0), 0);
    $this
      ->assertEqual($file
      ->spaceUsed(1, FILE_STATUS_PERMANENT), 0);
    $this
      ->assertEqual($file
      ->spaceUsed(2, 0), 1);
    $this
      ->assertEqual($file
      ->spaceUsed(2, FILE_STATUS_PERMANENT), 70);
    $this
      ->assertEqual($file
      ->spaceUsed(3, 0), 3);
    $this
      ->assertEqual($file
      ->spaceUsed(3, FILE_STATUS_PERMANENT), 300);
  }

}

Classes

Name	Description
SpaceUsedTest	Tests the spaceUsed() function.