123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154 |
- <?php
- namespace Test\AppFramework\Controller;
- use OCP\AppFramework\AuthPublicShareController;
- use OCP\AppFramework\Http\RedirectResponse;
- use OCP\AppFramework\Http\TemplateResponse;
- use OCP\IRequest;
- use OCP\ISession;
- use OCP\IURLGenerator;
- class AuthPublicShareControllerTest extends \Test\TestCase {
-
- private $request;
-
- private $session;
-
- private $urlGenerator;
-
- private $controller;
- protected function setUp(): void {
- parent::setUp();
- $this->request = $this->createMock(IRequest::class);
- $this->session = $this->createMock(ISession::class);
- $this->urlGenerator = $this->createMock(IURLGenerator::class);
- $this->controller = $this->getMockBuilder(AuthPublicShareController::class)
- ->setConstructorArgs([
- 'app',
- $this->request,
- $this->session,
- $this->urlGenerator
- ])->setMethods([
- 'authFailed',
- 'getPasswordHash',
- 'isAuthenticated',
- 'isPasswordProtected',
- 'isValidToken',
- 'showShare',
- 'verifyPassword',
- 'validateIdentity',
- 'generatePassword'
- ])->getMock();
- }
- public function testShowAuthenticate() {
- $expects = new TemplateResponse('core', 'publicshareauth', [], 'guest');
- $this->assertEquals($expects, $this->controller->showAuthenticate());
- }
- public function testAuthenticateAuthenticated() {
- $this->controller->method('isAuthenticated')
- ->willReturn(true);
- $this->controller->setToken('myToken');
- $this->session->method('get')
- ->willReturnMap(['public_link_authenticate_redirect', ['foo' => 'bar']]);
- $this->urlGenerator->method('linkToRoute')
- ->willReturn('myLink!');
- $result = $this->controller->authenticate('password');
- $this->assertInstanceOf(RedirectResponse::class, $result);
- $this->assertSame('myLink!', $result->getRedirectURL());
- }
- public function testAuthenticateInvalidPassword() {
- $this->controller->setToken('token');
- $this->controller->method('isPasswordProtected')
- ->willReturn(true);
- $this->controller->method('verifyPassword')
- ->with('password')
- ->willReturn(false);
- $this->controller->expects($this->once())
- ->method('authFailed');
- $expects = new TemplateResponse('core', 'publicshareauth', ['wrongpw' => true], 'guest');
- $expects->throttle();
- $result = $this->controller->authenticate('password');
- $this->assertEquals($expects, $result);
- }
- public function testAuthenticateValidPassword() {
- $this->controller->setToken('token');
- $this->controller->method('isPasswordProtected')
- ->willReturn(true);
- $this->controller->method('verifyPassword')
- ->with('password')
- ->willReturn(true);
- $this->controller->method('getPasswordHash')
- ->willReturn('hash');
- $this->session->expects($this->once())
- ->method('regenerateId');
- $this->session->method('get')
- ->willReturnMap(['public_link_authenticate_redirect', ['foo' => 'bar']]);
- $tokenSet = false;
- $hashSet = false;
- $this->session
- ->method('set')
- ->willReturnCallback(function ($key, $value) use (&$tokenSet, &$hashSet) {
- if ($key === 'public_link_authenticated_token' && $value === 'token') {
- $tokenSet = true;
- return true;
- }
- if ($key === 'public_link_authenticated_password_hash' && $value === 'hash') {
- $hashSet = true;
- return true;
- }
- return false;
- });
- $this->urlGenerator->method('linkToRoute')
- ->willReturn('myLink!');
- $result = $this->controller->authenticate('password');
- $this->assertInstanceOf(RedirectResponse::class, $result);
- $this->assertSame('myLink!', $result->getRedirectURL());
- $this->assertTrue($tokenSet);
- $this->assertTrue($hashSet);
- }
- }
|