2014-03-10 18:19:18 +04:00
|
|
|
<?php
|
|
|
|
/**
|
|
|
|
* Copyright (c) 2014 Vincent Petry <pvince81@owncloud.com>
|
|
|
|
* This file is licensed under the Affero General Public License version 3 or
|
|
|
|
* later.
|
|
|
|
* See the COPYING-README file.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test the storage functions of OC_Helper
|
2015-11-20 13:27:11 +03:00
|
|
|
*
|
|
|
|
* @group DB
|
2014-03-10 18:19:18 +04:00
|
|
|
*/
|
2014-11-12 17:54:41 +03:00
|
|
|
class Test_Helper_Storage extends \Test\TestCase {
|
|
|
|
/** @var string */
|
2014-03-10 18:19:18 +04:00
|
|
|
private $user;
|
2014-11-12 17:54:41 +03:00
|
|
|
/** @var \OC\Files\Storage\Storage */
|
2014-03-10 18:19:18 +04:00
|
|
|
private $storageMock;
|
2014-11-12 17:54:41 +03:00
|
|
|
/** @var \OC\Files\Storage\Storage */
|
|
|
|
private $storage;
|
|
|
|
|
|
|
|
protected function setUp() {
|
|
|
|
parent::setUp();
|
2014-03-10 18:19:18 +04:00
|
|
|
|
2014-11-12 17:54:41 +03:00
|
|
|
$this->user = $this->getUniqueID('user_');
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC_User::createUser($this->user, $this->user);
|
|
|
|
|
2014-11-12 17:54:41 +03:00
|
|
|
$this->storage = \OC\Files\Filesystem::getStorage('/');
|
2014-03-10 18:19:18 +04:00
|
|
|
\OC\Files\Filesystem::tearDown();
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC_User::setUserId($this->user);
|
2014-03-10 18:19:18 +04:00
|
|
|
\OC\Files\Filesystem::init($this->user, '/' . $this->user . '/files');
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC\Files\Filesystem::clearMounts();
|
|
|
|
|
|
|
|
$this->storageMock = null;
|
|
|
|
}
|
|
|
|
|
2014-11-12 17:54:41 +03:00
|
|
|
protected function tearDown() {
|
2014-03-25 19:37:46 +04:00
|
|
|
$this->user = null;
|
2014-03-10 18:19:18 +04:00
|
|
|
|
2014-03-25 19:37:46 +04:00
|
|
|
if ($this->storageMock) {
|
|
|
|
$this->storageMock->getCache()->clear();
|
|
|
|
$this->storageMock = null;
|
|
|
|
}
|
|
|
|
\OC\Files\Filesystem::tearDown();
|
2014-11-12 17:54:41 +03:00
|
|
|
\OC\Files\Filesystem::mount($this->storage, array(), '/');
|
2014-03-25 19:37:46 +04:00
|
|
|
|
|
|
|
\OC_User::setUserId('');
|
|
|
|
\OC_User::deleteUser($this->user);
|
2014-12-04 18:48:07 +03:00
|
|
|
\OC::$server->getConfig()->deleteAllUserValues($this->user);
|
2014-11-12 17:54:41 +03:00
|
|
|
|
|
|
|
parent::tearDown();
|
2014-03-25 19:37:46 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Returns a storage mock that returns the given value as
|
|
|
|
* free space
|
|
|
|
*
|
|
|
|
* @param int $freeSpace free space value
|
|
|
|
* @return \OC\Files\Storage\Storage
|
|
|
|
*/
|
|
|
|
private function getStorageMock($freeSpace = 12) {
|
2014-03-10 18:19:18 +04:00
|
|
|
$this->storageMock = $this->getMock(
|
|
|
|
'\OC\Files\Storage\Temporary',
|
|
|
|
array('free_space'),
|
|
|
|
array('')
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
|
|
$this->storageMock->expects($this->once())
|
|
|
|
->method('free_space')
|
|
|
|
->will($this->returnValue(12));
|
2014-03-25 19:37:46 +04:00
|
|
|
return $this->storageMock;
|
2014-03-10 18:19:18 +04:00
|
|
|
}
|
|
|
|
|
2014-03-25 19:37:46 +04:00
|
|
|
/**
|
|
|
|
* Test getting the storage info
|
|
|
|
*/
|
|
|
|
function testGetStorageInfo() {
|
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
2014-03-10 18:19:18 +04:00
|
|
|
|
2014-03-25 19:37:46 +04:00
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(12, $storageInfo['free']);
|
|
|
|
$this->assertEquals(5, $storageInfo['used']);
|
|
|
|
$this->assertEquals(17, $storageInfo['total']);
|
2014-03-10 18:19:18 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2014-03-25 19:37:46 +04:00
|
|
|
* Test getting the storage info, ignoring extra mount points
|
2014-03-10 18:19:18 +04:00
|
|
|
*/
|
2014-03-25 19:37:46 +04:00
|
|
|
function testGetStorageInfoExcludingExtStorage() {
|
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
|
|
|
|
|
|
|
$extStorage = new \OC\Files\Storage\Temporary(array());
|
|
|
|
$extStorage->file_put_contents('extfile.txt', 'abcdefghijklmnopq');
|
|
|
|
$extStorage->getScanner()->scan(''); // update root size
|
|
|
|
|
|
|
|
\OC\Files\Filesystem::mount($extStorage, array(), '/' . $this->user . '/files/ext');
|
2014-03-10 18:19:18 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(12, $storageInfo['free']);
|
|
|
|
$this->assertEquals(5, $storageInfo['used']);
|
|
|
|
$this->assertEquals(17, $storageInfo['total']);
|
|
|
|
}
|
|
|
|
|
2014-03-25 19:37:46 +04:00
|
|
|
/**
|
|
|
|
* Test getting the storage info, including extra mount points
|
|
|
|
*/
|
|
|
|
function testGetStorageInfoIncludingExtStorage() {
|
|
|
|
$homeStorage = new \OC\Files\Storage\Temporary(array());
|
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
|
|
|
|
|
|
|
$extStorage = new \OC\Files\Storage\Temporary(array());
|
|
|
|
$extStorage->file_put_contents('extfile.txt', 'abcdefghijklmnopq');
|
|
|
|
$extStorage->getScanner()->scan(''); // update root size
|
|
|
|
|
|
|
|
\OC\Files\Filesystem::mount($extStorage, array(), '/' . $this->user . '/files/ext');
|
|
|
|
|
|
|
|
$config = \OC::$server->getConfig();
|
2015-12-02 18:11:38 +03:00
|
|
|
$oldConfig = $config->getSystemValue('quota_include_external_storage', false);
|
|
|
|
$config->setSystemValue('quota_include_external_storage', 'true');
|
|
|
|
|
|
|
|
$config->setUserValue($this->user, 'files', 'quota', '25');
|
2014-03-25 19:37:46 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(3, $storageInfo['free']);
|
|
|
|
$this->assertEquals(22, $storageInfo['used']);
|
|
|
|
$this->assertEquals(25, $storageInfo['total']);
|
|
|
|
|
2015-12-02 18:11:38 +03:00
|
|
|
$config->setSystemValue('quota_include_external_storage', $oldConfig);
|
|
|
|
$config->setUserValue($this->user, 'files', 'quota', 'default');
|
2014-03-25 19:37:46 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test getting the storage info excluding extra mount points
|
|
|
|
* when user has no quota set, even when quota ext storage option
|
|
|
|
* was set
|
|
|
|
*/
|
|
|
|
function testGetStorageInfoIncludingExtStorageWithNoUserQuota() {
|
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
|
|
|
|
|
|
|
$extStorage = new \OC\Files\Storage\Temporary(array());
|
|
|
|
$extStorage->file_put_contents('extfile.txt', 'abcdefghijklmnopq');
|
|
|
|
$extStorage->getScanner()->scan(''); // update root size
|
|
|
|
|
|
|
|
\OC\Files\Filesystem::mount($extStorage, array(), '/' . $this->user . '/files/ext');
|
|
|
|
|
2015-12-02 18:11:38 +03:00
|
|
|
$config = \OC::$server->getConfig();
|
|
|
|
$oldConfig = $config->getSystemValue('quota_include_external_storage', false);
|
|
|
|
$config->setSystemValue('quota_include_external_storage', 'true');
|
2014-03-25 19:37:46 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(12, $storageInfo['free']);
|
|
|
|
$this->assertEquals(5, $storageInfo['used']);
|
|
|
|
$this->assertEquals(17, $storageInfo['total']);
|
|
|
|
|
2015-12-02 18:11:38 +03:00
|
|
|
$config->setSystemValue('quota_include_external_storage', $oldConfig);
|
2014-03-25 19:37:46 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-03-10 18:19:18 +04:00
|
|
|
/**
|
|
|
|
* Test getting the storage info with quota enabled
|
|
|
|
*/
|
|
|
|
function testGetStorageInfoWithQuota() {
|
2014-03-25 19:37:46 +04:00
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
|
|
|
$homeStorage = new \OC\Files\Storage\Wrapper\Quota(
|
2014-03-10 18:19:18 +04:00
|
|
|
array(
|
2014-03-25 19:37:46 +04:00
|
|
|
'storage' => $homeStorage,
|
2014-03-10 18:19:18 +04:00
|
|
|
'quota' => 7
|
|
|
|
)
|
|
|
|
);
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
2014-03-10 18:19:18 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(2, $storageInfo['free']);
|
|
|
|
$this->assertEquals(5, $storageInfo['used']);
|
|
|
|
$this->assertEquals(7, $storageInfo['total']);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test getting the storage info when data exceeds quota
|
|
|
|
*/
|
|
|
|
function testGetStorageInfoWhenSizeExceedsQuota() {
|
2014-03-25 19:37:46 +04:00
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
$homeStorage->file_put_contents('test.txt', '0123456789');
|
|
|
|
$homeStorage = new \OC\Files\Storage\Wrapper\Quota(
|
2014-03-10 18:19:18 +04:00
|
|
|
array(
|
2014-03-25 19:37:46 +04:00
|
|
|
'storage' => $homeStorage,
|
2014-03-10 18:19:18 +04:00
|
|
|
'quota' => 7
|
|
|
|
)
|
|
|
|
);
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
2014-03-10 18:19:18 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(0, $storageInfo['free']);
|
|
|
|
$this->assertEquals(10, $storageInfo['used']);
|
|
|
|
// total = quota
|
|
|
|
$this->assertEquals(7, $storageInfo['total']);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Test getting the storage info when the remaining
|
|
|
|
* free storage space is less than the quota
|
|
|
|
*/
|
|
|
|
function testGetStorageInfoWhenFreeSpaceLessThanQuota() {
|
2014-03-25 19:37:46 +04:00
|
|
|
$homeStorage = $this->getStorageMock(12);
|
|
|
|
$homeStorage->file_put_contents('test.txt', '01234');
|
|
|
|
$homeStorage = new \OC\Files\Storage\Wrapper\Quota(
|
2014-03-10 18:19:18 +04:00
|
|
|
array(
|
2014-03-25 19:37:46 +04:00
|
|
|
'storage' => $homeStorage,
|
2014-03-10 18:19:18 +04:00
|
|
|
'quota' => 18
|
|
|
|
)
|
|
|
|
);
|
2014-03-25 19:37:46 +04:00
|
|
|
\OC\Files\Filesystem::mount($homeStorage, array(), '/' . $this->user . '/files');
|
2014-03-10 18:19:18 +04:00
|
|
|
|
|
|
|
$storageInfo = \OC_Helper::getStorageInfo('');
|
|
|
|
$this->assertEquals(12, $storageInfo['free']);
|
|
|
|
$this->assertEquals(5, $storageInfo['used']);
|
|
|
|
// total = free + used (because quota > total)
|
|
|
|
$this->assertEquals(17, $storageInfo['total']);
|
|
|
|
}
|
|
|
|
}
|