123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239 |
- <?php
- namespace OCA\Files_Trashbin\Tests\Command;
- use OC\User\Manager;
- use OCA\Files_Trashbin\Command\CleanUp;
- use OCP\Files\IRootFolder;
- use OCP\IDBConnection;
- use OCP\UserInterface;
- use Symfony\Component\Console\Exception\InvalidOptionException;
- use Symfony\Component\Console\Input\InputInterface;
- use Symfony\Component\Console\Output\NullOutput;
- use Symfony\Component\Console\Output\OutputInterface;
- use Test\TestCase;
- class CleanUpTest extends TestCase {
-
- protected $cleanup;
-
- protected $userManager;
-
- protected $rootFolder;
-
- protected $dbConnection;
-
- protected $trashTable = 'files_trash';
-
- protected $user0 = 'user0';
- protected function setUp(): void {
- 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);
- }
-
- public function initTable() {
- $query = $this->dbConnection->getQueryBuilder();
- $query->delete($this->trashTable)->execute();
- for ($i = 0; $i < 10; $i++) {
- $query->insert($this->trashTable)
- ->values([
- '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));
- }
-
- public function testRemoveDeletedFiles(bool $nodeExists): void {
- $this->initTable();
- $this->rootFolder
- ->method('nodeExists')
- ->with('/' . $this->user0 . '/files_trashbin')
- ->willReturnOnConsecutiveCalls($nodeExists, false);
- if ($nodeExists) {
- $this->rootFolder
- ->method('get')
- ->with('/' . $this->user0 . '/files_trashbin')
- ->willReturn($this->rootFolder);
- $this->rootFolder
- ->method('delete');
- } else {
- $this->rootFolder->expects($this->never())->method('get');
- $this->rootFolder->expects($this->never())->method('delete');
- }
- $this->invokePrivate($this->cleanup, 'removeDeletedFiles', [$this->user0, new NullOutput(), false]);
- if ($nodeExists) {
-
-
- $query = $this->dbConnection->getQueryBuilder();
- $query->select('user')
- ->from($this->trashTable);
- $qResult = $query->execute();
- $result = $qResult->fetchAll();
- $qResult->closeCursor();
- $this->assertSame(5, count($result));
- foreach ($result as $r) {
- $this->assertSame('user1', $r['user']);
- }
- } else {
-
-
- $getAllQuery = $this->dbConnection->getQueryBuilder();
- $result = $getAllQuery->select('id')
- ->from($this->trashTable)
- ->execute()
- ->fetchAll();
- $this->assertSame(10, count($result));
- }
- }
- public function dataTestRemoveDeletedFiles() {
- return [
- [true],
- [false]
- ];
- }
-
- public function testExecuteDeleteListOfUsers(): void {
- $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): void {
- $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->method('getArgument')
- ->with('user_id')
- ->willReturn($userIds);
- $inputInterface->method('getOption')
- ->willReturnMap([
- ['all-users', false],
- ['verbose', false],
- ]);
- $outputInterface = $this->getMockBuilder('\Symfony\Component\Console\Output\OutputInterface')
- ->disableOriginalConstructor()->getMock();
- $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]);
- }
-
- public function testExecuteAllUsers(): void {
- $userIds = [];
- $backendUsers = ['user1', 'user2'];
- $instance = $this->getMockBuilder('OCA\Files_Trashbin\Command\CleanUp')
- ->setMethods(['removeDeletedFiles'])
- ->setConstructorArgs([$this->rootFolder, $this->userManager, $this->dbConnection])
- ->getMock();
- $backend = $this->createMock(UserInterface::class);
- $backend->method('getUsers')
- ->with('', 500, 0)
- ->willReturn($backendUsers);
- $instance->expects($this->exactly(count($backendUsers)))
- ->method('removeDeletedFiles')
- ->willReturnCallback(function ($user) use ($backendUsers): void {
- $this->assertTrue(in_array($user, $backendUsers));
- });
- $inputInterface = $this->createMock(InputInterface::class);
- $inputInterface->method('getArgument')
- ->with('user_id')
- ->willReturn($userIds);
- $inputInterface->method('getOption')
- ->willReturnMap([
- ['all-users', true],
- ['verbose', false],
- ]);
- $outputInterface = $this->createMock(OutputInterface::class);
- $this->userManager
- ->method('getBackends')
- ->willReturn([$backend]);
- $this->invokePrivate($instance, 'execute', [$inputInterface, $outputInterface]);
- }
- public function testExecuteNoUsersAndNoAllUsers(): void {
- $inputInterface = $this->createMock(InputInterface::class);
- $inputInterface->method('getArgument')
- ->with('user_id')
- ->willReturn([]);
- $inputInterface->method('getOption')
- ->willReturnMap([
- ['all-users', false],
- ['verbose', false],
- ]);
- $outputInterface = $this->createMock(OutputInterface::class);
- $this->expectException(InvalidOptionException::class);
- $this->expectExceptionMessage('Either specify a user_id or --all-users');
- $this->invokePrivate($this->cleanup, 'execute', [$inputInterface, $outputInterface]);
- }
- public function testExecuteUsersAndAllUsers(): void {
- $inputInterface = $this->createMock(InputInterface::class);
- $inputInterface->method('getArgument')
- ->with('user_id')
- ->willReturn(['user1', 'user2']);
- $inputInterface->method('getOption')
- ->willReturnMap([
- ['all-users', true],
- ['verbose', false],
- ]);
- $outputInterface = $this->createMock(OutputInterface::class);
- $this->expectException(InvalidOptionException::class);
- $this->expectExceptionMessage('Either specify a user_id or --all-users');
- $this->invokePrivate($this->cleanup, 'execute', [$inputInterface, $outputInterface]);
- }
- }
|