2016-10-18 11:36:57 +03:00
|
|
|
<?php
|
|
|
|
/**
|
|
|
|
* @copyright Copyright (c) 2016, Roeland Jago Douma <roeland@famdouma.nl>
|
|
|
|
*
|
|
|
|
* @author Roeland Jago Douma <roeland@famdouma.nl>
|
|
|
|
*
|
|
|
|
* @license GNU AGPL version 3 or any later version
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU Affero General Public License as
|
|
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
|
|
* License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* 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
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
namespace OCA\Files_Sharing\Tests\Controller;
|
|
|
|
|
|
|
|
use OCA\Files_Sharing\Controller\PublicPreviewController;
|
|
|
|
use OCP\AppFramework\Http;
|
|
|
|
use OCP\AppFramework\Http\DataResponse;
|
|
|
|
use OCP\AppFramework\Http\FileDisplayResponse;
|
2019-04-16 17:50:44 +03:00
|
|
|
use OCP\AppFramework\Utility\ITimeFactory;
|
2016-10-18 11:36:57 +03:00
|
|
|
use OCP\Constants;
|
|
|
|
use OCP\Files\File;
|
|
|
|
use OCP\Files\Folder;
|
|
|
|
use OCP\Files\NotFoundException;
|
|
|
|
use OCP\Files\SimpleFS\ISimpleFile;
|
|
|
|
use OCP\IPreview;
|
|
|
|
use OCP\IRequest;
|
2018-06-14 11:04:31 +03:00
|
|
|
use OCP\ISession;
|
2016-10-18 11:36:57 +03:00
|
|
|
use OCP\Share\Exceptions\ShareNotFound;
|
|
|
|
use OCP\Share\IManager;
|
|
|
|
use OCP\Share\IShare;
|
2019-04-16 17:50:44 +03:00
|
|
|
use PHPUnit\Framework\MockObject\MockObject;
|
2016-10-18 11:36:57 +03:00
|
|
|
use Test\TestCase;
|
|
|
|
|
|
|
|
class PublicPreviewControllerTest extends TestCase {
|
|
|
|
|
|
|
|
/** @var IPreview|\PHPUnit_Framework_MockObject_MockObject */
|
|
|
|
private $previewManager;
|
|
|
|
/** @var IManager|\PHPUnit_Framework_MockObject_MockObject */
|
|
|
|
private $shareManager;
|
2019-04-16 17:50:44 +03:00
|
|
|
/** @var ITimeFactory|MockObject */
|
|
|
|
private $timeFactory;
|
2016-10-18 11:36:57 +03:00
|
|
|
|
|
|
|
/** @var PublicPreviewController */
|
|
|
|
private $controller;
|
|
|
|
|
|
|
|
public function setUp() {
|
|
|
|
parent::setUp();
|
|
|
|
|
|
|
|
$this->previewManager = $this->createMock(IPreview::class);
|
|
|
|
$this->shareManager = $this->createMock(IManager::class);
|
2019-04-16 17:50:44 +03:00
|
|
|
$this->timeFactory = $this->createMock(ITimeFactory::class);
|
|
|
|
|
|
|
|
$this->timeFactory->method('getTime')
|
|
|
|
->willReturn(1337);
|
|
|
|
|
|
|
|
$this->overwriteService(ITimeFactory::class, $this->timeFactory);
|
2016-10-18 11:36:57 +03:00
|
|
|
|
|
|
|
$this->controller = new PublicPreviewController(
|
|
|
|
'files_sharing',
|
|
|
|
$this->createMock(IRequest::class),
|
|
|
|
$this->shareManager,
|
2018-06-14 11:04:31 +03:00
|
|
|
$this->createMock(ISession::class),
|
2016-10-18 11:36:57 +03:00
|
|
|
$this->previewManager
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testInvalidToken() {
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('', 'file', 10, 10, '');
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_BAD_REQUEST);
|
|
|
|
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testInvalidWidth() {
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 0);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_BAD_REQUEST);
|
|
|
|
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testInvalidHeight() {
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 0);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_BAD_REQUEST);
|
|
|
|
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testInvalidShare() {
|
|
|
|
$this->shareManager->method('getShareByToken')
|
|
|
|
->with($this->equalTo('token'))
|
|
|
|
->willThrowException(new ShareNotFound());
|
|
|
|
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 10);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_NOT_FOUND);
|
|
|
|
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testShareNotAccessable() {
|
|
|
|
$share = $this->createMock(IShare::class);
|
|
|
|
$this->shareManager->method('getShareByToken')
|
|
|
|
->with($this->equalTo('token'))
|
|
|
|
->willReturn($share);
|
|
|
|
|
|
|
|
$share->method('getPermissions')
|
|
|
|
->willReturn(0);
|
|
|
|
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 10);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_FORBIDDEN);
|
|
|
|
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testPreviewFile() {
|
|
|
|
$share = $this->createMock(IShare::class);
|
|
|
|
$this->shareManager->method('getShareByToken')
|
|
|
|
->with($this->equalTo('token'))
|
|
|
|
->willReturn($share);
|
|
|
|
|
|
|
|
$share->method('getPermissions')
|
|
|
|
->willReturn(Constants::PERMISSION_READ);
|
|
|
|
|
|
|
|
$file = $this->createMock(File::class);
|
|
|
|
$share->method('getNode')
|
|
|
|
->willReturn($file);
|
|
|
|
|
|
|
|
$preview = $this->createMock(ISimpleFile::class);
|
|
|
|
$this->previewManager->method('getPreview')
|
|
|
|
->with($this->equalTo($file), 10, 10, false)
|
|
|
|
->willReturn($preview);
|
|
|
|
|
|
|
|
$preview->method('getMimeType')
|
|
|
|
->willReturn('myMime');
|
|
|
|
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 10, true);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new FileDisplayResponse($preview, Http::STATUS_OK, ['Content-Type' => 'myMime']);
|
2019-01-03 00:27:46 +03:00
|
|
|
$expected->cacheFor(3600 * 24);
|
2016-10-18 11:36:57 +03:00
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testPreviewFolderInvalidFile() {
|
|
|
|
$share = $this->createMock(IShare::class);
|
|
|
|
$this->shareManager->method('getShareByToken')
|
|
|
|
->with($this->equalTo('token'))
|
|
|
|
->willReturn($share);
|
|
|
|
|
|
|
|
$share->method('getPermissions')
|
|
|
|
->willReturn(Constants::PERMISSION_READ);
|
|
|
|
|
|
|
|
$folder = $this->createMock(Folder::class);
|
|
|
|
$share->method('getNode')
|
|
|
|
->willReturn($folder);
|
|
|
|
|
|
|
|
$folder->method('get')
|
|
|
|
->with($this->equalTo('file'))
|
|
|
|
->willThrowException(new NotFoundException());
|
|
|
|
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 10, true);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new DataResponse([], Http::STATUS_NOT_FOUND);
|
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public function testPreviewFolderValidFile() {
|
|
|
|
$share = $this->createMock(IShare::class);
|
|
|
|
$this->shareManager->method('getShareByToken')
|
|
|
|
->with($this->equalTo('token'))
|
|
|
|
->willReturn($share);
|
|
|
|
|
|
|
|
$share->method('getPermissions')
|
|
|
|
->willReturn(Constants::PERMISSION_READ);
|
|
|
|
|
|
|
|
$folder = $this->createMock(Folder::class);
|
|
|
|
$share->method('getNode')
|
|
|
|
->willReturn($folder);
|
|
|
|
|
|
|
|
$file = $this->createMock(File::class);
|
|
|
|
$folder->method('get')
|
|
|
|
->with($this->equalTo('file'))
|
|
|
|
->willReturn($file);
|
|
|
|
|
|
|
|
$preview = $this->createMock(ISimpleFile::class);
|
|
|
|
$this->previewManager->method('getPreview')
|
|
|
|
->with($this->equalTo($file), 10, 10, false)
|
|
|
|
->willReturn($preview);
|
|
|
|
|
|
|
|
$preview->method('getMimeType')
|
|
|
|
->willReturn('myMime');
|
|
|
|
|
2018-06-14 11:04:31 +03:00
|
|
|
$res = $this->controller->getPreview('token', 'file', 10, 10, true);
|
2016-10-18 11:36:57 +03:00
|
|
|
$expected = new FileDisplayResponse($preview, Http::STATUS_OK, ['Content-Type' => 'myMime']);
|
2019-01-03 00:27:46 +03:00
|
|
|
$expected->cacheFor(3600 * 24);
|
2016-10-18 11:36:57 +03:00
|
|
|
$this->assertEquals($expected, $res);
|
|
|
|
}
|
|
|
|
}
|