213 lines
6.3 KiB
PHP
213 lines
6.3 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (c) 2016, ownCloud, Inc.
|
|
*
|
|
* @author Björn Schießle <bjoern@schiessle.org>
|
|
* @author Joas Schilling <coding@schilljs.com>
|
|
*
|
|
* @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\Federation\Tests\BackgroundJob;
|
|
|
|
|
|
use OCA\Federation\BackgroundJob\RequestSharedSecret;
|
|
use OCA\Federation\DbHandler;
|
|
use OCA\Federation\TrustedServers;
|
|
use OCP\AppFramework\Http;
|
|
use OCP\BackgroundJob\IJobList;
|
|
use OCP\Http\Client\IClient;
|
|
use OCP\Http\Client\IClientService;
|
|
use OCP\Http\Client\IResponse;
|
|
use OCP\ILogger;
|
|
use OCP\IURLGenerator;
|
|
use OCP\OCS\IDiscoveryService;
|
|
use Test\TestCase;
|
|
|
|
class RequestSharedSecretTest extends TestCase {
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IClientService */
|
|
private $httpClientService;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IClient */
|
|
private $httpClient;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IJobList */
|
|
private $jobList;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IURLGenerator */
|
|
private $urlGenerator;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | DbHandler */
|
|
private $dbHandler;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | TrustedServers */
|
|
private $trustedServers;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IResponse */
|
|
private $response;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | IDiscoveryService */
|
|
private $discoveryService;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject | ILogger */
|
|
private $logger;
|
|
|
|
/** @var RequestSharedSecret */
|
|
private $requestSharedSecret;
|
|
|
|
public function setUp() {
|
|
parent::setUp();
|
|
|
|
$this->httpClientService = $this->createMock(IClientService::class);
|
|
$this->httpClient = $this->getMockBuilder(IClient::class)->getMock();
|
|
$this->jobList = $this->getMockBuilder(IJobList::class)->getMock();
|
|
$this->urlGenerator = $this->getMockBuilder(IURLGenerator::class)->getMock();
|
|
$this->trustedServers = $this->getMockBuilder(TrustedServers::class)
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->dbHandler = $this->getMockBuilder(DbHandler::class)
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->response = $this->getMockBuilder(IResponse::class)->getMock();
|
|
$this->discoveryService = $this->getMockBuilder(IDiscoveryService::class)->getMock();
|
|
$this->logger = $this->createMock(ILogger::class);
|
|
|
|
$this->discoveryService->expects($this->any())->method('discover')->willReturn([]);
|
|
$this->httpClientService->expects($this->any())->method('newClient')->willReturn($this->httpClient);
|
|
|
|
$this->requestSharedSecret = new RequestSharedSecret(
|
|
$this->httpClientService,
|
|
$this->urlGenerator,
|
|
$this->jobList,
|
|
$this->trustedServers,
|
|
$this->dbHandler,
|
|
$this->discoveryService,
|
|
$this->logger
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider dataTestExecute
|
|
*
|
|
* @param bool $isTrustedServer
|
|
* @param bool $retainBackgroundJob
|
|
*/
|
|
public function testExecute($isTrustedServer, $retainBackgroundJob) {
|
|
/** @var RequestSharedSecret |\PHPUnit_Framework_MockObject_MockObject $requestSharedSecret */
|
|
$requestSharedSecret = $this->getMockBuilder('OCA\Federation\BackgroundJob\RequestSharedSecret')
|
|
->setConstructorArgs(
|
|
[
|
|
$this->httpClientService,
|
|
$this->urlGenerator,
|
|
$this->jobList,
|
|
$this->trustedServers,
|
|
$this->dbHandler,
|
|
$this->discoveryService,
|
|
$this->logger
|
|
]
|
|
)->setMethods(['parentExecute', 'reAddJob'])->getMock();
|
|
$this->invokePrivate($requestSharedSecret, 'argument', [['url' => 'url']]);
|
|
|
|
$this->trustedServers->expects($this->once())->method('isTrustedServer')
|
|
->with('url')->willReturn($isTrustedServer);
|
|
if ($isTrustedServer) {
|
|
$requestSharedSecret->expects($this->once())->method('parentExecute');
|
|
} else {
|
|
$requestSharedSecret->expects($this->never())->method('parentExecute');
|
|
}
|
|
$this->invokePrivate($requestSharedSecret, 'retainJob', [$retainBackgroundJob]);
|
|
$this->jobList->expects($this->once())->method('remove');
|
|
if ($retainBackgroundJob) {
|
|
$requestSharedSecret->expects($this->once())->method('reAddJob');
|
|
} else {
|
|
$requestSharedSecret->expects($this->never())->method('reAddJob');
|
|
}
|
|
|
|
$requestSharedSecret->execute($this->jobList);
|
|
|
|
}
|
|
|
|
public function dataTestExecute() {
|
|
return [
|
|
[true, true],
|
|
[true, false],
|
|
[false, false],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @dataProvider dataTestRun
|
|
*
|
|
* @param int $statusCode
|
|
*/
|
|
public function testRun($statusCode) {
|
|
|
|
$target = 'targetURL';
|
|
$source = 'sourceURL';
|
|
$token = 'token';
|
|
|
|
$argument = ['url' => $target, 'token' => $token];
|
|
|
|
$this->urlGenerator->expects($this->once())->method('getAbsoluteURL')->with('/')
|
|
->willReturn($source);
|
|
$this->httpClient->expects($this->once())->method('post')
|
|
->with(
|
|
$target . '/ocs/v2.php/apps/federation/api/v1/request-shared-secret?format=json',
|
|
[
|
|
'body' =>
|
|
[
|
|
'url' => $source,
|
|
'token' => $token
|
|
],
|
|
'timeout' => 3,
|
|
'connect_timeout' => 3,
|
|
]
|
|
)->willReturn($this->response);
|
|
|
|
$this->response->expects($this->once())->method('getStatusCode')
|
|
->willReturn($statusCode);
|
|
|
|
if (
|
|
$statusCode !== Http::STATUS_OK
|
|
&& $statusCode !== Http::STATUS_FORBIDDEN
|
|
) {
|
|
$this->dbHandler->expects($this->never())->method('addToken');
|
|
}
|
|
|
|
if ($statusCode === Http::STATUS_FORBIDDEN) {
|
|
$this->dbHandler->expects($this->once())->method('addToken')->with($target, '');
|
|
}
|
|
|
|
$this->invokePrivate($this->requestSharedSecret, 'run', [$argument]);
|
|
if (
|
|
$statusCode !== Http::STATUS_OK
|
|
&& $statusCode !== Http::STATUS_FORBIDDEN
|
|
) {
|
|
$this->assertTrue($this->invokePrivate($this->requestSharedSecret, 'retainJob'));
|
|
} else {
|
|
$this->assertFalse($this->invokePrivate($this->requestSharedSecret, 'retainJob'));
|
|
}
|
|
}
|
|
|
|
public function dataTestRun() {
|
|
return [
|
|
[Http::STATUS_OK],
|
|
[Http::STATUS_FORBIDDEN],
|
|
[Http::STATUS_CONFLICT],
|
|
];
|
|
}
|
|
}
|