* * @author Roeland Jago Douma * * @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 . * */ namespace Test\AppFramework\Middleware; use OC\AppFramework\Middleware\AdditionalScriptsMiddleware; use OCP\AppFramework\Controller; use OCP\AppFramework\Http\Events\BeforeTemplateRenderedEvent; use OCP\AppFramework\Http\Response; use OCP\AppFramework\Http\StandaloneTemplateResponse; use OCP\AppFramework\Http\TemplateResponse; use OCP\AppFramework\PublicShareController; use OCP\EventDispatcher\IEventDispatcher; use OCP\IUserSession; use PHPUnit\Framework\MockObject\MockObject; class AdditionalScriptsMiddlewareTest extends \Test\TestCase { /** @var IUserSession|MockObject */ private $userSession; /** @var Controller */ private $controller; /** @var AdditionalScriptsMiddleware */ private $middleWare; /** @var IEventDispatcher|MockObject */ private $dispatcher; protected function setUp(): void { parent::setUp(); $this->userSession = $this->createMock(IUserSession::class); $this->dispatcher = $this->createMock(IEventDispatcher::class); $this->middleWare = new AdditionalScriptsMiddleware( $this->userSession, $this->dispatcher ); $this->controller = $this->createMock(Controller::class); } public function testNoTemplateResponse() { $this->userSession->expects($this->never()) ->method($this->anything()); $this->dispatcher->expects($this->never()) ->method($this->anything()); $this->middleWare->afterController($this->controller, 'myMethod', $this->createMock(Response::class)); } public function testPublicShareController() { $this->userSession->expects($this->never()) ->method($this->anything()); $this->dispatcher->expects($this->never()) ->method($this->anything()); $this->middleWare->afterController($this->createMock(PublicShareController::class), 'myMethod', $this->createMock(Response::class)); } public function testStandaloneTemplateResponse() { $this->userSession->expects($this->never()) ->method($this->anything()); $this->dispatcher->expects($this->once()) ->method('dispatchTyped') ->willReturnCallback(function ($event) { if ($event instanceof BeforeTemplateRenderedEvent && $event->isLoggedIn() === false) { return; } $this->fail('Wrong event dispatched'); }); $this->middleWare->afterController($this->controller, 'myMethod', $this->createMock(StandaloneTemplateResponse::class)); } public function testTemplateResponseNotLoggedIn() { $this->userSession->method('isLoggedIn') ->willReturn(false); $this->dispatcher->expects($this->once()) ->method('dispatchTyped') ->willReturnCallback(function ($event) { if ($event instanceof BeforeTemplateRenderedEvent && $event->isLoggedIn() === false) { return; } $this->fail('Wrong event dispatched'); }); $this->middleWare->afterController($this->controller, 'myMethod', $this->createMock(TemplateResponse::class)); } public function testTemplateResponseLoggedIn() { $events = []; $this->userSession->method('isLoggedIn') ->willReturn(true); $this->dispatcher->expects($this->once()) ->method('dispatchTyped') ->willReturnCallback(function ($event) { if ($event instanceof BeforeTemplateRenderedEvent && $event->isLoggedIn() === true) { return; } $this->fail('Wrong event dispatched'); }); $this->middleWare->afterController($this->controller, 'myMethod', $this->createMock(TemplateResponse::class)); } }