412 lines
13 KiB
PHP
412 lines
13 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (c) 2016, ownCloud, Inc.
|
|
*
|
|
* @author Björn Schießle <bjoern@schiessle.org>
|
|
* @author Joas Schilling <coding@schilljs.com>
|
|
* @author Lukas Reschke <lukas@statuscode.ch>
|
|
* @author Morris Jobke <hey@morrisjobke.de>
|
|
* @author Robin Appelman <robin@icewind.nl>
|
|
* @author Thomas Müller <thomas.mueller@tmit.eu>
|
|
*
|
|
* @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\FederatedFileSharing\Tests;
|
|
|
|
use OC\Federation\CloudIdManager;
|
|
use OC\Files\Filesystem;
|
|
use OCA\FederatedFileSharing\DiscoveryManager;
|
|
use OCA\FederatedFileSharing\FederatedShareProvider;
|
|
use OCA\FederatedFileSharing\Controller\RequestHandlerController;
|
|
use OCP\Federation\ICloudIdManager;
|
|
use OCP\Http\Client\IClient;
|
|
use OCP\Http\Client\IClientService;
|
|
use OCP\Http\Client\IResponse;
|
|
use OCP\IUserManager;
|
|
use OCP\Share\IShare;
|
|
|
|
/**
|
|
* Class RequestHandlerTest
|
|
*
|
|
* @package OCA\FederatedFileSharing\Tests
|
|
* @group DB
|
|
*/
|
|
class RequestHandlerControllerTest extends TestCase {
|
|
|
|
const TEST_FOLDER_NAME = '/folder_share_api_test';
|
|
|
|
/**
|
|
* @var \OCP\IDBConnection
|
|
*/
|
|
private $connection;
|
|
|
|
/**
|
|
* @var RequestHandlerController
|
|
*/
|
|
private $s2s;
|
|
|
|
/** @var \OCA\FederatedFileSharing\FederatedShareProvider|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $federatedShareProvider;
|
|
|
|
/** @var \OCA\FederatedFileSharing\Notifications|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $notifications;
|
|
|
|
/** @var \OCA\FederatedFileSharing\AddressHandler|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $addressHandler;
|
|
|
|
/** @var IUserManager|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $userManager;
|
|
|
|
/** @var IShare|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $share;
|
|
|
|
/** @var ICloudIdManager */
|
|
private $cloudIdManager;
|
|
|
|
protected function setUp() {
|
|
parent::setUp();
|
|
|
|
self::loginHelper(self::TEST_FILES_SHARING_API_USER1);
|
|
\OCP\Share::registerBackend('test', 'Test\Share\Backend');
|
|
|
|
$config = $this->getMockBuilder('\OCP\IConfig')
|
|
->disableOriginalConstructor()->getMock();
|
|
$clientService = $this->getMockBuilder('\OCP\Http\Client\IClientService')->getMock();
|
|
$httpHelperMock = $this->getMockBuilder('\OC\HTTPHelper')
|
|
->setConstructorArgs([$config, $clientService])
|
|
->getMock();
|
|
$httpHelperMock->expects($this->any())->method('post')->with($this->anything())->will($this->returnValue(true));
|
|
$this->share = $this->getMockBuilder('\OCP\Share\IShare')->getMock();
|
|
$this->federatedShareProvider = $this->getMockBuilder('OCA\FederatedFileSharing\FederatedShareProvider')
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->federatedShareProvider->expects($this->any())
|
|
->method('isOutgoingServer2serverShareEnabled')->willReturn(true);
|
|
$this->federatedShareProvider->expects($this->any())
|
|
->method('isIncomingServer2serverShareEnabled')->willReturn(true);
|
|
$this->federatedShareProvider->expects($this->any())->method('getShareById')
|
|
->willReturn($this->share);
|
|
|
|
$this->notifications = $this->getMockBuilder('OCA\FederatedFileSharing\Notifications')
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->addressHandler = $this->getMockBuilder('OCA\FederatedFileSharing\AddressHandler')
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->userManager = $this->getMockBuilder('OCP\IUserManager')->getMock();
|
|
|
|
$this->cloudIdManager = new CloudIdManager();
|
|
|
|
$this->registerHttpHelper($httpHelperMock);
|
|
|
|
$this->s2s = new RequestHandlerController(
|
|
'federatedfilesharing',
|
|
\OC::$server->getRequest(),
|
|
$this->federatedShareProvider,
|
|
\OC::$server->getDatabaseConnection(),
|
|
\OC::$server->getShareManager(),
|
|
$this->notifications,
|
|
$this->addressHandler,
|
|
$this->userManager,
|
|
$this->cloudIdManager
|
|
);
|
|
|
|
$this->connection = \OC::$server->getDatabaseConnection();
|
|
}
|
|
|
|
protected function tearDown() {
|
|
$query = \OCP\DB::prepare('DELETE FROM `*PREFIX*share_external`');
|
|
$query->execute();
|
|
|
|
$query = \OCP\DB::prepare('DELETE FROM `*PREFIX*share`');
|
|
$query->execute();
|
|
|
|
$this->restoreHttpHelper();
|
|
|
|
parent::tearDown();
|
|
}
|
|
|
|
/**
|
|
* Register an http helper mock for testing purposes.
|
|
* @param \OC\HTTPHelper $httpHelper helper mock
|
|
*/
|
|
private function registerHttpHelper($httpHelper) {
|
|
$this->oldHttpHelper = \OC::$server->query('HTTPHelper');
|
|
\OC::$server->registerService('HTTPHelper', function ($c) use ($httpHelper) {
|
|
return $httpHelper;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Restore the original http helper
|
|
*/
|
|
private function restoreHttpHelper() {
|
|
$oldHttpHelper = $this->oldHttpHelper;
|
|
\OC::$server->registerService('HTTPHelper', function ($c) use ($oldHttpHelper) {
|
|
return $oldHttpHelper;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @medium
|
|
*/
|
|
function testCreateShare() {
|
|
// simulate a post request
|
|
$_POST['remote'] = 'localhost';
|
|
$_POST['token'] = 'token';
|
|
$_POST['name'] = 'name';
|
|
$_POST['owner'] = 'owner';
|
|
$_POST['shareWith'] = self::TEST_FILES_SHARING_API_USER2;
|
|
$_POST['remoteId'] = 1;
|
|
|
|
$this->s2s->createShare(null);
|
|
|
|
$query = \OCP\DB::prepare('SELECT * FROM `*PREFIX*share_external` WHERE `remote_id` = ?');
|
|
$result = $query->execute(array('1'));
|
|
$data = $result->fetchRow();
|
|
|
|
$this->assertSame('localhost', $data['remote']);
|
|
$this->assertSame('token', $data['share_token']);
|
|
$this->assertSame('/name', $data['name']);
|
|
$this->assertSame('owner', $data['owner']);
|
|
$this->assertSame(self::TEST_FILES_SHARING_API_USER2, $data['user']);
|
|
$this->assertSame(1, (int)$data['remote_id']);
|
|
$this->assertSame(0, (int)$data['accepted']);
|
|
}
|
|
|
|
|
|
function testDeclineShare() {
|
|
|
|
$this->s2s = $this->getMockBuilder('\OCA\FederatedFileSharing\Controller\RequestHandlerController')
|
|
->setConstructorArgs(
|
|
[
|
|
'federatedfilessharing',
|
|
\OC::$server->getRequest(),
|
|
$this->federatedShareProvider,
|
|
\OC::$server->getDatabaseConnection(),
|
|
\OC::$server->getShareManager(),
|
|
$this->notifications,
|
|
$this->addressHandler,
|
|
$this->userManager,
|
|
$this->cloudIdManager
|
|
]
|
|
)->setMethods(['executeDeclineShare', 'verifyShare'])->getMock();
|
|
|
|
$this->s2s->expects($this->once())->method('executeDeclineShare');
|
|
|
|
$this->s2s->expects($this->any())->method('verifyShare')->willReturn(true);
|
|
|
|
$_POST['token'] = 'token';
|
|
|
|
$this->s2s->declineShare(42);
|
|
|
|
}
|
|
|
|
function XtestDeclineShareMultiple() {
|
|
|
|
$this->share->expects($this->any())->method('verifyShare')->willReturn(true);
|
|
|
|
$dummy = \OCP\DB::prepare('
|
|
INSERT INTO `*PREFIX*share`
|
|
(`share_type`, `uid_owner`, `item_type`, `item_source`, `item_target`, `file_source`, `file_target`, `permissions`, `stime`, `token`, `share_with`)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
');
|
|
$dummy->execute(array(\OCP\Share::SHARE_TYPE_REMOTE, self::TEST_FILES_SHARING_API_USER1, 'test', '1', '/1', '1', '/test.txt', '1', time(), 'token1', 'foo@bar'));
|
|
$dummy->execute(array(\OCP\Share::SHARE_TYPE_REMOTE, self::TEST_FILES_SHARING_API_USER1, 'test', '1', '/1', '1', '/test.txt', '1', time(), 'token2', 'bar@bar'));
|
|
|
|
$verify = \OCP\DB::prepare('SELECT * FROM `*PREFIX*share`');
|
|
$result = $verify->execute();
|
|
$data = $result->fetchAll();
|
|
$this->assertCount(2, $data);
|
|
|
|
$_POST['token'] = 'token1';
|
|
$this->s2s->declineShare(array('id' => $data[0]['id']));
|
|
|
|
$verify = \OCP\DB::prepare('SELECT * FROM `*PREFIX*share`');
|
|
$result = $verify->execute();
|
|
$data = $result->fetchAll();
|
|
$this->assertCount(1, $data);
|
|
$this->assertEquals('bar@bar', $data[0]['share_with']);
|
|
|
|
$_POST['token'] = 'token2';
|
|
$this->s2s->declineShare(array('id' => $data[0]['id']));
|
|
|
|
$verify = \OCP\DB::prepare('SELECT * FROM `*PREFIX*share`');
|
|
$result = $verify->execute();
|
|
$data = $result->fetchAll();
|
|
$this->assertEmpty($data);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider dataTestDeleteUser
|
|
*/
|
|
function testDeleteUser($toDelete, $expected, $remainingUsers) {
|
|
$this->createDummyS2SShares();
|
|
|
|
$httpClientService = $this->createMock(IClientService::class);
|
|
$client = $this->createMock(IClient::class);
|
|
$response = $this->createMock(IResponse::class);
|
|
$client
|
|
->expects($this->any())
|
|
->method('get')
|
|
->willReturn($response);
|
|
$client
|
|
->expects($this->any())
|
|
->method('post')
|
|
->willReturn($response);
|
|
$httpClientService
|
|
->expects($this->any())
|
|
->method('newClient')
|
|
->willReturn($client);
|
|
|
|
$discoveryManager = new DiscoveryManager(
|
|
\OC::$server->getMemCacheFactory(),
|
|
$httpClientService
|
|
);
|
|
$manager = new \OCA\Files_Sharing\External\Manager(
|
|
\OC::$server->getDatabaseConnection(),
|
|
Filesystem::getMountManager(),
|
|
Filesystem::getLoader(),
|
|
$httpClientService,
|
|
\OC::$server->getNotificationManager(),
|
|
$discoveryManager,
|
|
$toDelete
|
|
);
|
|
|
|
$manager->removeUserShares($toDelete);
|
|
|
|
$query = $this->connection->prepare('SELECT `user` FROM `*PREFIX*share_external`');
|
|
$query->execute();
|
|
$result = $query->fetchAll();
|
|
|
|
foreach ($result as $r) {
|
|
$remainingShares[$r['user']] = isset($remainingShares[$r['user']]) ? $remainingShares[$r['user']] + 1 : 1;
|
|
}
|
|
|
|
$this->assertSame($remainingUsers, count($remainingShares));
|
|
|
|
foreach ($expected as $key => $value) {
|
|
if ($key === $toDelete) {
|
|
$this->assertArrayNotHasKey($key, $remainingShares);
|
|
} else {
|
|
$this->assertSame($value, $remainingShares[$key]);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
function dataTestDeleteUser() {
|
|
return array(
|
|
array('user1', array('user1' => 0, 'user2' => 3, 'user3' => 3), 2),
|
|
array('user2', array('user1' => 4, 'user2' => 0, 'user3' => 3), 2),
|
|
array('user3', array('user1' => 4, 'user2' => 3, 'user3' => 0), 2),
|
|
array('user4', array('user1' => 4, 'user2' => 3, 'user3' => 3), 3),
|
|
);
|
|
}
|
|
|
|
private function createDummyS2SShares() {
|
|
$query = $this->connection->prepare('
|
|
INSERT INTO `*PREFIX*share_external`
|
|
(`remote`, `share_token`, `password`, `name`, `owner`, `user`, `mountpoint`, `mountpoint_hash`, `remote_id`, `accepted`)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
');
|
|
|
|
$users = array('user1', 'user2', 'user3');
|
|
|
|
for ($i = 0; $i < 10; $i++) {
|
|
$user = $users[$i%3];
|
|
$query->execute(array('remote', 'token', 'password', 'name', 'owner', $user, 'mount point', $i, $i, 0));
|
|
}
|
|
|
|
$query = $this->connection->prepare('SELECT `id` FROM `*PREFIX*share_external`');
|
|
$query->execute();
|
|
$dummyEntries = $query->fetchAll();
|
|
|
|
$this->assertSame(10, count($dummyEntries));
|
|
}
|
|
|
|
/**
|
|
* @dataProvider dataTestGetShare
|
|
*
|
|
* @param bool $found
|
|
* @param bool $correctId
|
|
* @param bool $correctToken
|
|
*/
|
|
public function testGetShare($found, $correctId, $correctToken) {
|
|
|
|
$connection = \OC::$server->getDatabaseConnection();
|
|
$query = $connection->getQueryBuilder();
|
|
$stime = time();
|
|
$query->insert('share')
|
|
->values(
|
|
[
|
|
'share_type' => $query->createNamedParameter(FederatedShareProvider::SHARE_TYPE_REMOTE),
|
|
'uid_owner' => $query->createNamedParameter(self::TEST_FILES_SHARING_API_USER1),
|
|
'uid_initiator' => $query->createNamedParameter(self::TEST_FILES_SHARING_API_USER2),
|
|
'item_type' => $query->createNamedParameter('test'),
|
|
'item_source' => $query->createNamedParameter('1'),
|
|
'item_target' => $query->createNamedParameter('/1'),
|
|
'file_source' => $query->createNamedParameter('1'),
|
|
'file_target' => $query->createNamedParameter('/test.txt'),
|
|
'permissions' => $query->createNamedParameter('1'),
|
|
'stime' => $query->createNamedParameter($stime),
|
|
'token' => $query->createNamedParameter('token'),
|
|
'share_with' => $query->createNamedParameter('foo@bar'),
|
|
]
|
|
)->execute();
|
|
$id = $query->getLastInsertId();
|
|
|
|
$expected = [
|
|
'share_type' => (string)FederatedShareProvider::SHARE_TYPE_REMOTE,
|
|
'uid_owner' => self::TEST_FILES_SHARING_API_USER1,
|
|
'item_type' => 'test',
|
|
'item_source' => '1',
|
|
'item_target' => '/1',
|
|
'file_source' => '1',
|
|
'file_target' => '/test.txt',
|
|
'permissions' => '1',
|
|
'stime' => (string)$stime,
|
|
'token' => 'token',
|
|
'share_with' => 'foo@bar',
|
|
'id' => (string)$id,
|
|
'uid_initiator' => self::TEST_FILES_SHARING_API_USER2,
|
|
'parent' => null,
|
|
'accepted' => '0',
|
|
'expiration' => null,
|
|
'mail_send' => '0'
|
|
];
|
|
|
|
$searchToken = $correctToken ? 'token' : 'wrongToken';
|
|
$searchId = $correctId ? $id : -1;
|
|
|
|
$result = $this->invokePrivate($this->s2s, 'getShare', [$searchId, $searchToken]);
|
|
|
|
if ($found) {
|
|
$this->assertEquals($expected, $result);
|
|
} else {
|
|
$this->assertSame(false, $result);
|
|
}
|
|
}
|
|
|
|
public function dataTestGetShare() {
|
|
return [
|
|
[true, true, true],
|
|
[false, false, true],
|
|
[false, true, false],
|
|
[false, false, false],
|
|
];
|
|
}
|
|
|
|
}
|