diff options
Diffstat (limited to 'apps/files_trashbin/tests/Command/CleanUpTest.php')
| -rw-r--r-- | apps/files_trashbin/tests/Command/CleanUpTest.php | 207 |
1 files changed, 207 insertions, 0 deletions
diff --git a/apps/files_trashbin/tests/Command/CleanUpTest.php b/apps/files_trashbin/tests/Command/CleanUpTest.php new file mode 100644 index 00000000000..d71c4ecd85d --- /dev/null +++ b/apps/files_trashbin/tests/Command/CleanUpTest.php @@ -0,0 +1,207 @@ +<?php +/** + * @author Björn Schießle <schiessle@owncloud.com> + * @author Joas Schilling <nickvergessen@owncloud.com> + * @author Thomas Müller <thomas.mueller@tmit.eu> + * + * @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 OCA\Files_Trashbin\Tests\Command; + + +use OCA\Files_Trashbin\Command\CleanUp; +use Test\TestCase; +use OC\User\Manager; +use OCP\Files\IRootFolder; + +/** + * Class CleanUpTest + * + * @group DB + * + * @package OCA\Files_Trashbin\Tests\Command + */ +class CleanUpTest extends TestCase { + + /** @var CleanUp */ + protected $cleanup; + + /** @var \PHPUnit_Framework_MockObject_MockObject | Manager */ + protected $userManager; + + /** @var \PHPUnit_Framework_MockObject_MockObject | IRootFolder */ + protected $rootFolder; + + /** @var \OC\DB\Connection */ + protected $dbConnection; + + /** @var string */ + protected $trashTable = 'files_trash'; + + /** @var string */ + protected $user0 = 'user0'; + + public function setUp() { + parent::setUp(); + $this->rootFolder = $this->getMockBuilder('OCP\Files\IRootFolder') + ->disableOriginalConstructor()->getMock(); + $this->userManager = $this->getMockBuilder('OC\User\Manager') + ->disableOriginalConstructor()->getMock(); + + $this->dbConnection = \OC::$server->getDatabaseConnection(); + + $this->cleanup = new CleanUp($this->rootFolder, $this->userManager, $this->dbConnection); + } + + /** + * populate files_trash table with 10 dummy values + */ + public function initTable() { + $query = $this->dbConnection->getQueryBuilder(); + $query->delete($this->trashTable)->execute(); + for ($i = 0; $i < 10; $i++) { + $query->insert($this->trashTable) + ->values(array( + 'id' => $query->expr()->literal('file'.$i), + 'timestamp' => $query->expr()->literal($i), + 'location' => $query->expr()->literal('.'), + 'user' => $query->expr()->literal('user'.$i%2) + ))->execute(); + } + $getAllQuery = $this->dbConnection->getQueryBuilder(); + $result = $getAllQuery->select('id') + ->from($this->trashTable) + ->execute() + ->fetchAll(); + $this->assertSame(10, count($result)); + } + + /** + * @dataProvider dataTestRemoveDeletedFiles + * @param boolean $nodeExists + */ + public function testRemoveDeletedFiles($nodeExists) { + $this->initTable(); + $this->rootFolder->expects($this->once()) + ->method('nodeExists') + ->with('/' . $this->user0 . '/files_trashbin') + ->willReturn($nodeExists); + if($nodeExists) { + $this->rootFolder->expects($this->once()) + ->method('get') + ->with('/' . $this->user0 . '/files_trashbin') + ->willReturn($this->rootFolder); + $this->rootFolder->expects($this->once()) + ->method('delete'); + } else { + $this->rootFolder->expects($this->never())->method('get'); + $this->rootFolder->expects($this->never())->method('delete'); + } + $this->invokePrivate($this->cleanup, 'removeDeletedFiles', [$this->user0]); + + if ($nodeExists) { + // if the delete operation was execute only files from user1 + // should be left. + $query = $this->dbConnection->getQueryBuilder(); + $result = $query->select('user') + ->from($this->trashTable) + ->execute()->fetchAll(); + $this->assertSame(5, count($result)); + foreach ($result as $r) { + $this->assertSame('user1', $r['user']); + } + } else { + // if no delete operation was execute we should still have all 10 + // database entries + $getAllQuery = $this->dbConnection->getQueryBuilder(); + $result = $getAllQuery->select('id') + ->from($this->trashTable) + ->execute() + ->fetchAll(); + $this->assertSame(10, count($result)); + } + + } + public function dataTestRemoveDeletedFiles() { + return array( + array(true), + array(false) + ); + } + + /** + * test remove deleted files from users given as parameter + */ + public function testExecuteDeleteListOfUsers() { + $userIds = ['user1', 'user2', 'user3']; + $instance = $this->getMockBuilder('OCA\Files_Trashbin\Command\CleanUp') + ->setMethods(['removeDeletedFiles']) + ->setConstructorArgs([$this->rootFolder, $this->userManager, $this->dbConnection]) + ->getMock(); + $instance->expects($this->exactly(count($userIds))) + ->method('removeDeletedFiles') + ->willReturnCallback(function ($user) use ($userIds) { + $this->assertTrue(in_array($user, $userIds)); + }); + $this->userManager->expects($this->exactly(count($userIds))) + ->method('userExists')->willReturn(true); + $inputInterface = $this->getMockBuilder('\Symfony\Component\Console\Input\InputInterface') + ->disableOriginalConstructor()->getMock(); + $inputInterface->expects($this->once())->method('getArgument') + ->with('user_id') + ->willReturn($userIds); + $outputInterface = $this->getMockBuilder('\Symfony\Component\Console\Output\OutputInterface') + ->disableOriginalConstructor()->getMock(); + $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]); + } + + /** + * test remove deleted files of all users + */ + public function testExecuteAllUsers() { + $userIds = []; + $backendUsers = ['user1', 'user2']; + $instance = $this->getMockBuilder('OCA\Files_Trashbin\Command\CleanUp') + ->setMethods(['removeDeletedFiles']) + ->setConstructorArgs([$this->rootFolder, $this->userManager, $this->dbConnection]) + ->getMock(); + $backend = $this->getMockBuilder('OC_User_Interface') + ->disableOriginalConstructor()->getMock(); + $backend->expects($this->once())->method('getUsers') + ->with('', 500, 0) + ->willReturn($backendUsers); + $instance->expects($this->exactly(count($backendUsers))) + ->method('removeDeletedFiles') + ->willReturnCallback(function ($user) use ($backendUsers) { + $this->assertTrue(in_array($user, $backendUsers)); + }); + $inputInterface = $this->getMockBuilder('\Symfony\Component\Console\Input\InputInterface') + ->disableOriginalConstructor()->getMock(); + $inputInterface->expects($this->once())->method('getArgument') + ->with('user_id') + ->willReturn($userIds); + $outputInterface = $this->getMockBuilder('\Symfony\Component\Console\Output\OutputInterface') + ->disableOriginalConstructor()->getMock(); + $this->userManager->expects($this->once()) + ->method('getBackends') + ->willReturn([$backend]); + $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]); + } + +} |