353 lines
9.5 KiB
PHP
353 lines
9.5 KiB
PHP
<?php
|
|
/**
|
|
* ownCloud
|
|
*
|
|
* @author Joas Schilling
|
|
* @copyright 2014 Joas Schilling nickvergessen@owncloud.com
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
* License as published by the Free Software Foundation; either
|
|
* version 3 of the License, or any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU AFFERO GENERAL PUBLIC LICENSE for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public
|
|
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
namespace Test;
|
|
|
|
use OC\Command\QueueBus;
|
|
use OC\Files\Filesystem;
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IDBConnection;
|
|
use OCP\Security\ISecureRandom;
|
|
|
|
abstract class TestCase extends \PHPUnit_Framework_TestCase {
|
|
/** @var \OC\Command\QueueBus */
|
|
private $commandBus;
|
|
|
|
/** @var IDBConnection */
|
|
static private $realDatabase;
|
|
|
|
protected function getTestTraits() {
|
|
$traits = [];
|
|
$class = $this;
|
|
do {
|
|
$traits = array_merge(class_uses($class), $traits);
|
|
} while ($class = get_parent_class($class));
|
|
foreach ($traits as $trait => $same) {
|
|
$traits = array_merge(class_uses($trait), $traits);
|
|
}
|
|
$traits = array_unique($traits);
|
|
return array_filter($traits, function ($trait) {
|
|
return substr($trait, 0, 5) === 'Test\\';
|
|
});
|
|
}
|
|
|
|
protected function setUp() {
|
|
// detect database access
|
|
if (!$this->IsDatabaseAccessAllowed()) {
|
|
if (is_null(self::$realDatabase)) {
|
|
self::$realDatabase = \OC::$server->getDatabaseConnection();
|
|
}
|
|
\OC::$server->registerService('DatabaseConnection', function () {
|
|
$this->fail('Your test case is not allowed to access the database.');
|
|
});
|
|
}
|
|
|
|
// overwrite the command bus with one we can run ourselves
|
|
$this->commandBus = new QueueBus();
|
|
\OC::$server->registerService('AsyncCommandBus', function () {
|
|
return $this->commandBus;
|
|
});
|
|
|
|
$traits = $this->getTestTraits();
|
|
foreach ($traits as $trait) {
|
|
$methodName = 'setUp' . basename(str_replace('\\', '/', $trait));
|
|
if (method_exists($this, $methodName)) {
|
|
call_user_func([$this, $methodName]);
|
|
}
|
|
}
|
|
}
|
|
|
|
protected function tearDown() {
|
|
// restore database connection
|
|
if (!$this->IsDatabaseAccessAllowed()) {
|
|
\OC::$server->registerService('DatabaseConnection', function () {
|
|
return self::$realDatabase;
|
|
});
|
|
}
|
|
|
|
// further cleanup
|
|
$hookExceptions = \OC_Hook::$thrownExceptions;
|
|
\OC_Hook::$thrownExceptions = [];
|
|
\OC::$server->getLockingProvider()->releaseAll();
|
|
if (!empty($hookExceptions)) {
|
|
throw $hookExceptions[0];
|
|
}
|
|
|
|
$traits = $this->getTestTraits();
|
|
foreach ($traits as $trait) {
|
|
$methodName = 'tearDown' . basename(str_replace('\\', '/', $trait));
|
|
if (method_exists($this, $methodName)) {
|
|
call_user_func([$this, $methodName]);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Allows us to test private methods/properties
|
|
*
|
|
* @param $object
|
|
* @param $methodName
|
|
* @param array $parameters
|
|
* @return mixed
|
|
*/
|
|
protected static function invokePrivate($object, $methodName, array $parameters = array()) {
|
|
$reflection = new \ReflectionClass(get_class($object));
|
|
|
|
if ($reflection->hasMethod($methodName)) {
|
|
$method = $reflection->getMethod($methodName);
|
|
|
|
$method->setAccessible(true);
|
|
|
|
return $method->invokeArgs($object, $parameters);
|
|
} elseif ($reflection->hasProperty($methodName)) {
|
|
$property = $reflection->getProperty($methodName);
|
|
|
|
$property->setAccessible(true);
|
|
|
|
if (!empty($parameters)) {
|
|
$property->setValue($object, array_pop($parameters));
|
|
}
|
|
|
|
return $property->getValue($object);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Returns a unique identifier as uniqid() is not reliable sometimes
|
|
*
|
|
* @param string $prefix
|
|
* @param int $length
|
|
* @return string
|
|
*/
|
|
protected static function getUniqueID($prefix = '', $length = 13) {
|
|
return $prefix . \OC::$server->getSecureRandom()->getLowStrengthGenerator()->generate(
|
|
$length,
|
|
// Do not use dots and slashes as we use the value for file names
|
|
ISecureRandom::CHAR_DIGITS . ISecureRandom::CHAR_LOWER . ISecureRandom::CHAR_UPPER
|
|
);
|
|
}
|
|
|
|
public static function tearDownAfterClass() {
|
|
$dataDir = \OC::$server->getConfig()->getSystemValue('datadirectory', \OC::$SERVERROOT . '/data-autotest');
|
|
$queryBuilder = \OC::$server->getDatabaseConnection()->getQueryBuilder();
|
|
|
|
self::tearDownAfterClassCleanShares($queryBuilder);
|
|
self::tearDownAfterClassCleanStorages($queryBuilder);
|
|
self::tearDownAfterClassCleanFileCache($queryBuilder);
|
|
self::tearDownAfterClassCleanStrayDataFiles($dataDir);
|
|
self::tearDownAfterClassCleanStrayHooks();
|
|
self::tearDownAfterClassCleanStrayLocks();
|
|
|
|
parent::tearDownAfterClass();
|
|
}
|
|
|
|
/**
|
|
* Remove all entries from the share table
|
|
*
|
|
* @param IQueryBuilder $queryBuilder
|
|
*/
|
|
static protected function tearDownAfterClassCleanShares(IQueryBuilder $queryBuilder) {
|
|
$queryBuilder->delete('share')
|
|
->execute();
|
|
}
|
|
|
|
/**
|
|
* Remove all entries from the storages table
|
|
*
|
|
* @param IQueryBuilder $queryBuilder
|
|
*/
|
|
static protected function tearDownAfterClassCleanStorages(IQueryBuilder $queryBuilder) {
|
|
$queryBuilder->delete('storages')
|
|
->execute();
|
|
}
|
|
|
|
/**
|
|
* Remove all entries from the filecache table
|
|
*
|
|
* @param IQueryBuilder $queryBuilder
|
|
*/
|
|
static protected function tearDownAfterClassCleanFileCache(IQueryBuilder $queryBuilder) {
|
|
$queryBuilder->delete('filecache')
|
|
->execute();
|
|
}
|
|
|
|
/**
|
|
* Remove all unused files from the data dir
|
|
*
|
|
* @param string $dataDir
|
|
*/
|
|
static protected function tearDownAfterClassCleanStrayDataFiles($dataDir) {
|
|
$knownEntries = array(
|
|
'owncloud.log' => true,
|
|
'owncloud.db' => true,
|
|
'.ocdata' => true,
|
|
'..' => true,
|
|
'.' => true,
|
|
);
|
|
|
|
if ($dh = opendir($dataDir)) {
|
|
while (($file = readdir($dh)) !== false) {
|
|
if (!isset($knownEntries[$file])) {
|
|
self::tearDownAfterClassCleanStrayDataUnlinkDir($dataDir . '/' . $file);
|
|
}
|
|
}
|
|
closedir($dh);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Recursive delete files and folders from a given directory
|
|
*
|
|
* @param string $dir
|
|
*/
|
|
static protected function tearDownAfterClassCleanStrayDataUnlinkDir($dir) {
|
|
if ($dh = @opendir($dir)) {
|
|
while (($file = readdir($dh)) !== false) {
|
|
if (\OC\Files\Filesystem::isIgnoredDir($file)) {
|
|
continue;
|
|
}
|
|
$path = $dir . '/' . $file;
|
|
if (is_dir($path)) {
|
|
self::tearDownAfterClassCleanStrayDataUnlinkDir($path);
|
|
} else {
|
|
@unlink($path);
|
|
}
|
|
}
|
|
closedir($dh);
|
|
}
|
|
@rmdir($dir);
|
|
}
|
|
|
|
/**
|
|
* Clean up the list of hooks
|
|
*/
|
|
static protected function tearDownAfterClassCleanStrayHooks() {
|
|
\OC_Hook::clear();
|
|
}
|
|
|
|
/**
|
|
* Clean up the list of locks
|
|
*/
|
|
static protected function tearDownAfterClassCleanStrayLocks() {
|
|
\OC::$server->getLockingProvider()->releaseAll();
|
|
}
|
|
|
|
/**
|
|
* Login and setup FS as a given user,
|
|
* sets the given user as the current user.
|
|
*
|
|
* @param string $user user id or empty for a generic FS
|
|
*/
|
|
static protected function loginAsUser($user = '') {
|
|
self::logout();
|
|
\OC\Files\Filesystem::tearDown();
|
|
\OC_User::setUserId($user);
|
|
\OC_Util::setupFS($user);
|
|
if (\OC_User::userExists($user)) {
|
|
\OC::$server->getUserFolder($user);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Logout the current user and tear down the filesystem.
|
|
*/
|
|
static protected function logout() {
|
|
\OC_Util::tearDownFS();
|
|
\OC_User::setUserId('');
|
|
// needed for fully logout
|
|
\OC::$server->getUserSession()->setUser(null);
|
|
}
|
|
|
|
/**
|
|
* Run all commands pushed to the bus
|
|
*/
|
|
protected function runCommands() {
|
|
// get the user for which the fs is setup
|
|
$view = Filesystem::getView();
|
|
if ($view) {
|
|
list(, $user) = explode('/', $view->getRoot());
|
|
} else {
|
|
$user = null;
|
|
}
|
|
|
|
\OC_Util::tearDownFS(); // command cant reply on the fs being setup
|
|
$this->commandBus->run();
|
|
\OC_Util::tearDownFS();
|
|
|
|
if ($user) {
|
|
\OC_Util::setupFS($user);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check if the given path is locked with a given type
|
|
*
|
|
* @param \OC\Files\View $view view
|
|
* @param string $path path to check
|
|
* @param int $type lock type
|
|
* @param bool $onMountPoint true to check the mount point instead of the
|
|
* mounted storage
|
|
*
|
|
* @return boolean true if the file is locked with the
|
|
* given type, false otherwise
|
|
*/
|
|
protected function isFileLocked($view, $path, $type, $onMountPoint = false) {
|
|
// Note: this seems convoluted but is necessary because
|
|
// the format of the lock key depends on the storage implementation
|
|
// (in our case mostly md5)
|
|
|
|
if ($type === \OCP\Lock\ILockingProvider::LOCK_SHARED) {
|
|
// to check if the file has a shared lock, try acquiring an exclusive lock
|
|
$checkType = \OCP\Lock\ILockingProvider::LOCK_EXCLUSIVE;
|
|
} else {
|
|
// a shared lock cannot be set if exclusive lock is in place
|
|
$checkType = \OCP\Lock\ILockingProvider::LOCK_SHARED;
|
|
}
|
|
try {
|
|
$view->lockFile($path, $checkType, $onMountPoint);
|
|
// no exception, which means the lock of $type is not set
|
|
// clean up
|
|
$view->unlockFile($path, $checkType, $onMountPoint);
|
|
return false;
|
|
} catch (\OCP\Lock\LockedException $e) {
|
|
// we could not acquire the counter-lock, which means
|
|
// the lock of $type was in place
|
|
return true;
|
|
}
|
|
}
|
|
|
|
private function IsDatabaseAccessAllowed() {
|
|
// on travis-ci.org we allow database access in any case - otherwise
|
|
// this will break all apps right away
|
|
if (true == getenv('TRAVIS')) {
|
|
return true;
|
|
}
|
|
$annotations = $this->getAnnotations();
|
|
if (isset($annotations['class']['group']) && in_array('DB', $annotations['class']['group'])) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|