View source
<?php
namespace Drupal\Tests\file\Kernel;
use Drupal\file\Entity\File;
class SpaceUsedTest extends FileManagedUnitTestBase {
protected function setUp() {
parent::setUp();
$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);
$this
->createFileWithSize('public://example5.txt', 1, 2, 0);
$this
->createFileWithSize('public://example6.txt', 3, 3, 0);
}
protected function createFileWithSize($uri, $size, $uid, $status = FILE_STATUS_PERMANENT) {
file_put_contents($uri, $this
->randomMachineName($size));
$file = File::create([
'uri' => $uri,
'uid' => $uid,
'status' => $status,
]);
$file
->save();
return $file;
}
public function testFileSpaceUsed() {
$file = $this->container
->get('entity_type.manager')
->getStorage('file');
$this
->assertEqual($file
->spaceUsed(2), 70);
$this
->assertEqual($file
->spaceUsed(3), 300);
$this
->assertEqual($file
->spaceUsed(), 370);
$this
->assertEqual($file
->spaceUsed(NULL, 0), 4);
$this
->assertEqual($file
->spaceUsed(NULL, FILE_STATUS_PERMANENT), 370);
$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);
}
}