123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198 |
- <?php
- /**
- * @author Roeland Jago Douma <rullzer@owncloud.com>
- *
- * @copyright Copyright (c) 2016, ownCloud, Inc.
- * @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 <http://www.gnu.org/licenses/>
- *
- */
- namespace Test\Repair;
- use OC\Repair\RemoveRootShares;
- use OCP\Files\IRootFolder;
- use OCP\IDBConnection;
- use OCP\IUserManager;
- use OCP\Migration\IOutput;
- use Test\Traits\UserTrait;
- /**
- * Class RemoveOldSharesTest
- *
- * @package Test\Repair
- * @group DB
- */
- class RemoveRootSharesTest extends \Test\TestCase {
- use UserTrait;
- /** @var RemoveRootShares */
- protected $repair;
- /** @var IDBConnection */
- protected $connection;
- /** @var IOutput */
- private $outputMock;
- /** @var IUserManager */
- private $userManager;
- /** @var IRootFolder */
- private $rootFolder;
- protected function setUp() {
- parent::setUp();
- $this->outputMock = $this->getMockBuilder('\OCP\Migration\IOutput')
- ->disableOriginalConstructor()
- ->getMock();
- $this->userManager = \OC::$server->getUserManager();
- $this->rootFolder = \OC::$server->getRootFolder();
- $this->connection = \OC::$server->getDatabaseConnection();
- $this->repair = new RemoveRootShares($this->connection, $this->userManager, $this->rootFolder);
- }
- protected function tearDown() {
- $qb = $this->connection->getQueryBuilder();
- $qb->delete('share');
- $qb->execute();
- return parent::tearDown();
- }
- public function testRootSharesExist() {
- //Add test user
- $user = $this->userManager->createUser('test', 'test');
- $userFolder = $this->rootFolder->getUserFolder('test');
- $fileId = $userFolder->getId();
- //Now insert cyclic share
- $qb = $this->connection->getQueryBuilder();
- $qb->insert('share')
- ->values([
- 'share_type' => $qb->createNamedParameter(0),
- 'share_with' => $qb->createNamedParameter('foo'),
- 'uid_owner' => $qb->createNamedParameter('owner'),
- 'item_type' => $qb->createNamedParameter('file'),
- 'item_source' => $qb->createNamedParameter($fileId),
- 'item_target' => $qb->createNamedParameter('/target'),
- 'file_source' => $qb->createNamedParameter($fileId),
- 'file_target' => $qb->createNamedParameter('/target'),
- 'permissions' => $qb->createNamedParameter(1),
- ]);
- $qb->execute();
- $res = $this->invokePrivate($this->repair, 'rootSharesExist', []);
- $this->assertTrue($res);
- $user->delete();
- }
- public function testRootSharesDontExist() {
- //Add test user
- $user = $this->userManager->createUser('test', 'test');
- $userFolder = $this->rootFolder->getUserFolder('test');
- $fileId = $userFolder->getId();
- $user->updateLastLoginTimestamp();
- //Now insert cyclic share
- $qb = $this->connection->getQueryBuilder();
- $qb->insert('share')
- ->values([
- 'share_type' => $qb->createNamedParameter(0),
- 'share_with' => $qb->createNamedParameter('foo'),
- 'uid_owner' => $qb->createNamedParameter('owner'),
- 'item_type' => $qb->createNamedParameter('file'),
- 'item_source' => $qb->createNamedParameter($fileId+1),
- 'item_target' => $qb->createNamedParameter('/target'),
- 'file_source' => $qb->createNamedParameter($fileId+1),
- 'file_target' => $qb->createNamedParameter('/target'),
- 'permissions' => $qb->createNamedParameter(1),
- ]);
- $qb->execute();
- $res = $this->invokePrivate($this->repair, 'rootSharesExist', []);
- $this->assertFalse($res);
- $user->delete();
- }
- public function testRun() {
- //Add test user
- $user1 = $this->userManager->createUser('test1', 'test1');
- $userFolder = $this->rootFolder->getUserFolder('test1');
- $fileId = $userFolder->getId();
- $user1->updateLastLoginTimestamp();
- //Now insert cyclic share
- $qb = $this->connection->getQueryBuilder();
- $qb->insert('share')
- ->values([
- 'share_type' => $qb->createNamedParameter(0),
- 'share_with' => $qb->createNamedParameter('foo'),
- 'uid_owner' => $qb->createNamedParameter('owner'),
- 'item_type' => $qb->createNamedParameter('file'),
- 'item_source' => $qb->createNamedParameter($fileId),
- 'item_target' => $qb->createNamedParameter('/target'),
- 'file_source' => $qb->createNamedParameter($fileId),
- 'file_target' => $qb->createNamedParameter('/target'),
- 'permissions' => $qb->createNamedParameter(1),
- ]);
- $qb->execute();
- //Add test user
- $user2 = $this->userManager->createUser('test2', 'test2');
- $userFolder = $this->rootFolder->getUserFolder('test2');
- $folder = $userFolder->newFolder('foo');
- $fileId = $folder->getId();
- $user2->updateLastLoginTimestamp();
- //Now insert cyclic share
- $qb = $this->connection->getQueryBuilder();
- $qb->insert('share')
- ->values([
- 'share_type' => $qb->createNamedParameter(0),
- 'share_with' => $qb->createNamedParameter('foo'),
- 'uid_owner' => $qb->createNamedParameter('owner'),
- 'item_type' => $qb->createNamedParameter('file'),
- 'item_source' => $qb->createNamedParameter($fileId),
- 'item_target' => $qb->createNamedParameter('/target'),
- 'file_source' => $qb->createNamedParameter($fileId),
- 'file_target' => $qb->createNamedParameter('/target'),
- 'permissions' => $qb->createNamedParameter(1),
- ]);
- $qb->execute();
- $this->repair->run($this->outputMock);
- //Verify
- $qb = $this->connection->getQueryBuilder();
- $qb->selectAlias($qb->createFunction('COUNT(*)'), 'count')
- ->from('share');
- $cursor = $qb->execute();
- $data = $cursor->fetch();
- $cursor->closeCursor();
- $count = (int)$data['count'];
- $this->assertEquals(1, $count);
- $user1->delete();
- $user2->delete();
- }
- }
|