* @author Joas Schilling * @author John Molakvoæ (skjnldsv) * @author Jörn Friedrich Dreyer * @author Lukas Reschke * @author Morris Jobke * @author Robin Appelman * @author Robin McCorkell * @author Roeland Jago Douma * @author Thomas Müller * @author Vincent Petry * * @license AGPL-3.0 * * This code is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program 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, version 3, * along with this program. If not, see * */ namespace OCA\Files_Sharing\Tests; use OC\Files\Cache\Scanner; use OC\Files\Filesystem; use OCA\Files_Sharing\AppInfo\Application; use OCP\Share\IShare; use Test\Traits\MountProviderTrait; /** * Class TestCase * * @group DB * * Base class for sharing tests. */ abstract class TestCase extends \Test\TestCase { use MountProviderTrait; const TEST_FILES_SHARING_API_USER1 = "test-share-user1"; const TEST_FILES_SHARING_API_USER2 = "test-share-user2"; const TEST_FILES_SHARING_API_USER3 = "test-share-user3"; const TEST_FILES_SHARING_API_USER4 = "test-share-user4"; const TEST_FILES_SHARING_API_GROUP1 = "test-share-group1"; public $filename; public $data; /** * @var \OC\Files\View */ public $view; public $folder; public $subfolder; /** @var \OCP\Share\IManager */ protected $shareManager; /** @var \OCP\Files\IRootFolder */ protected $rootFolder; public static function setUpBeforeClass(): void { parent::setUpBeforeClass(); new Application(); // reset backend \OC_User::clearBackends(); \OC::$server->getGroupManager()->clearBackends(); // clear share hooks \OC_Hook::clear('OCP\\Share'); \OC::registerShareHooks(); // create users $backend = new \Test\Util\User\Dummy(); \OC_User::useBackend($backend); $backend->createUser(self::TEST_FILES_SHARING_API_USER1, self::TEST_FILES_SHARING_API_USER1); $backend->createUser(self::TEST_FILES_SHARING_API_USER2, self::TEST_FILES_SHARING_API_USER2); $backend->createUser(self::TEST_FILES_SHARING_API_USER3, self::TEST_FILES_SHARING_API_USER3); $backend->createUser(self::TEST_FILES_SHARING_API_USER4, self::TEST_FILES_SHARING_API_USER4); // create group $groupBackend = new \Test\Util\Group\Dummy(); $groupBackend->createGroup(self::TEST_FILES_SHARING_API_GROUP1); $groupBackend->createGroup('group'); $groupBackend->createGroup('group1'); $groupBackend->createGroup('group2'); $groupBackend->createGroup('group3'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER1, 'group'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER2, 'group'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER3, 'group'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER2, 'group1'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER3, 'group2'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER4, 'group3'); $groupBackend->addToGroup(self::TEST_FILES_SHARING_API_USER2, self::TEST_FILES_SHARING_API_GROUP1); \OC::$server->getGroupManager()->addBackend($groupBackend); } protected function setUp(): void { parent::setUp(); //login as user1 self::loginHelper(self::TEST_FILES_SHARING_API_USER1); $this->data = 'foobar'; $this->view = new \OC\Files\View('/' . self::TEST_FILES_SHARING_API_USER1 . '/files'); $this->shareManager = \OC::$server->getShareManager(); $this->rootFolder = \OC::$server->getRootFolder(); } protected function tearDown(): void { $qb = \OC::$server->getDatabaseConnection()->getQueryBuilder(); $qb->delete('share'); $qb->execute(); $qb = \OC::$server->getDatabaseConnection()->getQueryBuilder(); $qb->delete('mounts'); $qb->execute(); $qb = \OC::$server->getDatabaseConnection()->getQueryBuilder(); $qb->delete('filecache'); $qb->execute(); parent::tearDown(); } public static function tearDownAfterClass(): void { // cleanup users $user = \OC::$server->getUserManager()->get(self::TEST_FILES_SHARING_API_USER1); if ($user !== null) { $user->delete(); } $user = \OC::$server->getUserManager()->get(self::TEST_FILES_SHARING_API_USER2); if ($user !== null) { $user->delete(); } $user = \OC::$server->getUserManager()->get(self::TEST_FILES_SHARING_API_USER3); if ($user !== null) { $user->delete(); } // delete group $group = \OC::$server->getGroupManager()->get(self::TEST_FILES_SHARING_API_GROUP1); if ($group) { $group->delete(); } \OC_Util::tearDownFS(); \OC_User::setUserId(''); Filesystem::tearDown(); // reset backend \OC_User::clearBackends(); \OC_User::useBackend('database'); \OC::$server->getGroupManager()->clearBackends(); \OC::$server->getGroupManager()->addBackend(new \OC\Group\Database()); parent::tearDownAfterClass(); } /** * @param string $user * @param bool $create * @param bool $password */ protected static function loginHelper($user, $create = false, $password = false) { if ($password === false) { $password = $user; } if ($create) { $userManager = \OC::$server->getUserManager(); $groupManager = \OC::$server->getGroupManager(); $userObject = $userManager->createUser($user, $password); $group = $groupManager->createGroup('group'); if ($group and $userObject) { $group->addUser($userObject); } } self::resetStorage(); \OC_Util::tearDownFS(); \OC\Files\Cache\Storage::getGlobalCache()->clearCache(); \OC::$server->getUserSession()->setUser(null); \OC\Files\Filesystem::tearDown(); \OC::$server->getUserSession()->login($user, $password); \OC::$server->getUserFolder($user); \OC_Util::setupFS($user); } /** * reset init status for the share storage */ protected static function resetStorage() { $storage = new \ReflectionClass('\OCA\Files_Sharing\SharedStorage'); $isInitialized = $storage->getProperty('initialized'); $isInitialized->setAccessible(true); $isInitialized->setValue($storage, false); $isInitialized->setAccessible(false); } /** * get some information from a given share * @param int $shareID * @return array with: item_source, share_type, share_with, item_type, permissions */ protected function getShareFromId($shareID) { $qb = \OC::$server->getDatabaseConnection()->getQueryBuilder(); $qb->select('item_source', '`share_type', 'share_with', 'item_type', 'permissions') ->from('share') ->where( $qb->expr()->eq('id', $qb->createNamedParameter($shareID)) ); $result = $qb->execute(); $share = $result->fetch(); $result->closeCursor(); return $share; } /** * @param int $type The share type * @param string $path The path to share relative to $initiators root * @param string $initiator * @param string $recipient * @param int $permissions * @return \OCP\Share\IShare */ protected function share($type, $path, $initiator, $recipient, $permissions) { $userFolder = $this->rootFolder->getUserFolder($initiator); $node = $userFolder->get($path); $share = $this->shareManager->newShare(); $share->setShareType($type) ->setSharedWith($recipient) ->setSharedBy($initiator) ->setNode($node) ->setPermissions($permissions); $share = $this->shareManager->createShare($share); $share->setStatus(IShare::STATUS_ACCEPTED); $share = $this->shareManager->updateShare($share); return $share; } }