123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298 |
- <?php
- namespace OCA\Federation\Tests\BackgroundJob;
- use GuzzleHttp\Exception\ConnectException;
- use OCA\Federation\BackgroundJob\GetSharedSecret;
- use OCA\Federation\TrustedServers;
- use OCA\Files_Sharing\Tests\TestCase;
- use OCP\AppFramework\Http;
- use OCP\AppFramework\Utility\ITimeFactory;
- use OCP\BackgroundJob\IJobList;
- use OCP\Http\Client\IClient;
- use OCP\Http\Client\IClientService;
- use OCP\Http\Client\IResponse;
- use OCP\IURLGenerator;
- use OCP\OCS\IDiscoveryService;
- use Psr\Log\LoggerInterface;
- class GetSharedSecretTest extends TestCase {
-
- private $httpClient;
-
- private $httpClientService;
-
- private $jobList;
-
- private $urlGenerator;
-
- private $trustedServers;
-
- private $logger;
-
- private $response;
-
- private $discoverService;
-
- private $timeFactory;
- private GetSharedSecret $getSharedSecret;
- protected function setUp(): void {
- 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->logger = $this->getMockBuilder(LoggerInterface::class)->getMock();
- $this->response = $this->getMockBuilder(IResponse::class)->getMock();
- $this->discoverService = $this->getMockBuilder(IDiscoveryService::class)->getMock();
- $this->timeFactory = $this->createMock(ITimeFactory::class);
- $this->discoverService->expects($this->any())->method('discover')->willReturn([]);
- $this->httpClientService->expects($this->any())->method('newClient')->willReturn($this->httpClient);
- $this->getSharedSecret = new GetSharedSecret(
- $this->httpClientService,
- $this->urlGenerator,
- $this->jobList,
- $this->trustedServers,
- $this->logger,
- $this->discoverService,
- $this->timeFactory
- );
- }
-
- public function testExecute(bool $isTrustedServer, bool $retainBackgroundJob): void {
-
- $getSharedSecret = $this->getMockBuilder(GetSharedSecret::class)
- ->setConstructorArgs(
- [
- $this->httpClientService,
- $this->urlGenerator,
- $this->jobList,
- $this->trustedServers,
- $this->logger,
- $this->discoverService,
- $this->timeFactory
- ]
- )->setMethods(['parentStart'])->getMock();
- $this->invokePrivate($getSharedSecret, 'argument', [['url' => 'url', 'token' => 'token']]);
- $this->trustedServers->expects($this->once())->method('isTrustedServer')
- ->with('url')->willReturn($isTrustedServer);
- if ($isTrustedServer) {
- $getSharedSecret->expects($this->once())->method('parentStart');
- } else {
- $getSharedSecret->expects($this->never())->method('parentStart');
- }
- $this->invokePrivate($getSharedSecret, 'retainJob', [$retainBackgroundJob]);
- $this->jobList->expects($this->once())->method('remove');
- $this->timeFactory->method('getTime')->willReturn(42);
- if ($retainBackgroundJob) {
- $this->jobList->expects($this->once())
- ->method('add')
- ->with(
- GetSharedSecret::class,
- [
- 'url' => 'url',
- 'token' => 'token',
- 'created' => 42,
- ]
- );
- } else {
- $this->jobList->expects($this->never())->method('add');
- }
- $getSharedSecret->start($this->jobList);
- }
- public function dataTestExecute() {
- return [
- [true, true],
- [true, false],
- [false, false],
- ];
- }
-
- public function testRun($statusCode) {
- $target = 'targetURL';
- $source = 'sourceURL';
- $token = 'token';
- $argument = ['url' => $target, 'token' => $token];
- $this->timeFactory->method('getTime')
- ->willReturn(42);
- $this->urlGenerator->expects($this->once())->method('getAbsoluteURL')->with('/')
- ->willReturn($source);
- $this->httpClient->expects($this->once())->method('get')
- ->with(
- $target . '/ocs/v2.php/apps/federation/api/v1/shared-secret',
- [
- 'query' =>
- [
- 'url' => $source,
- 'token' => $token,
- 'format' => 'json',
- ],
- 'timeout' => 3,
- 'connect_timeout' => 3,
- ]
- )->willReturn($this->response);
- $this->response->expects($this->once())->method('getStatusCode')
- ->willReturn($statusCode);
- if ($statusCode === Http::STATUS_OK) {
- $this->response->expects($this->once())->method('getBody')
- ->willReturn('{"ocs":{"data":{"sharedSecret":"secret"}}}');
- $this->trustedServers->expects($this->once())->method('addSharedSecret')
- ->with($target, 'secret');
- } else {
- $this->trustedServers->expects($this->never())->method('addSharedSecret');
- }
- $this->invokePrivate($this->getSharedSecret, 'run', [$argument]);
- if (
- $statusCode !== Http::STATUS_OK
- && $statusCode !== Http::STATUS_FORBIDDEN
- ) {
- $this->assertTrue($this->invokePrivate($this->getSharedSecret, 'retainJob'));
- } else {
- $this->assertFalse($this->invokePrivate($this->getSharedSecret, 'retainJob'));
- }
- }
- public function dataTestRun() {
- return [
- [Http::STATUS_OK],
- [Http::STATUS_FORBIDDEN],
- [Http::STATUS_CONFLICT],
- ];
- }
- public function testRunExpired() {
- $target = 'targetURL';
- $source = 'sourceURL';
- $token = 'token';
- $created = 42;
- $argument = [
- 'url' => $target,
- 'token' => $token,
- 'created' => $created,
- ];
- $this->urlGenerator->expects($this->once())
- ->method('getAbsoluteURL')
- ->with('/')
- ->willReturn($source);
- $this->timeFactory->method('getTime')
- ->willReturn($created + 2592000 + 1);
- $this->trustedServers->expects($this->once())
- ->method('setServerStatus')
- ->with(
- $target,
- TrustedServers::STATUS_FAILURE
- );
- $this->invokePrivate($this->getSharedSecret, 'run', [$argument]);
- }
- public function testRunConnectionError() {
- $target = 'targetURL';
- $source = 'sourceURL';
- $token = 'token';
- $argument = ['url' => $target, 'token' => $token];
- $this->timeFactory->method('getTime')
- ->willReturn(42);
- $this->urlGenerator
- ->expects($this->once())
- ->method('getAbsoluteURL')
- ->with('/')
- ->willReturn($source);
- $this->httpClient->expects($this->once())->method('get')
- ->with(
- $target . '/ocs/v2.php/apps/federation/api/v1/shared-secret',
- [
- 'query' =>
- [
- 'url' => $source,
- 'token' => $token,
- 'format' => 'json',
- ],
- 'timeout' => 3,
- 'connect_timeout' => 3,
- ]
- )->willThrowException($this->createMock(ConnectException::class));
- $this->trustedServers->expects($this->never())->method('addSharedSecret');
- $this->invokePrivate($this->getSharedSecret, 'run', [$argument]);
- $this->assertTrue($this->invokePrivate($this->getSharedSecret, 'retainJob'));
- }
- }
|