2015-10-29 16:27:14 +00:00
|
|
|
<?php
|
|
|
|
/**
|
|
|
|
* @author Björn Schießle <schiessle@owncloud.com>
|
|
|
|
*
|
|
|
|
* @copyright Copyright (c) 2015, ownCloud, Inc.
|
|
|
|
* @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\lib;
|
|
|
|
|
|
|
|
|
|
|
|
use OCA\Federation\DbHandler;
|
2015-11-19 16:49:43 +00:00
|
|
|
use OCA\Federation\TrustedServers;
|
2015-10-29 16:27:14 +00:00
|
|
|
use OCP\IDBConnection;
|
|
|
|
use Test\TestCase;
|
|
|
|
|
2015-11-10 09:50:59 +00:00
|
|
|
/**
|
|
|
|
* @group DB
|
|
|
|
*/
|
2015-10-29 16:27:14 +00:00
|
|
|
class DbHandlerTest extends TestCase {
|
|
|
|
|
|
|
|
/** @var DbHandler */
|
|
|
|
private $dbHandler;
|
|
|
|
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject */
|
|
|
|
private $il10n;
|
|
|
|
|
|
|
|
/** @var IDBConnection */
|
|
|
|
private $connection;
|
|
|
|
|
|
|
|
/** @var string */
|
|
|
|
private $dbTable = 'trusted_servers';
|
|
|
|
|
|
|
|
public function setUp() {
|
|
|
|
parent::setUp();
|
|
|
|
|
|
|
|
$this->connection = \OC::$server->getDatabaseConnection();
|
|
|
|
$this->il10n = $this->getMock('OCP\IL10N');
|
|
|
|
|
|
|
|
$this->dbHandler = new DbHandler(
|
|
|
|
$this->connection,
|
|
|
|
$this->il10n
|
|
|
|
);
|
|
|
|
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertEmpty($result, 'we need to start with a empty trusted_servers table');
|
|
|
|
}
|
|
|
|
|
|
|
|
public function tearDown() {
|
|
|
|
parent::tearDown();
|
|
|
|
$query = $this->connection->getQueryBuilder()->delete($this->dbTable);
|
|
|
|
$query->execute();
|
|
|
|
}
|
|
|
|
|
2015-11-24 12:07:40 +00:00
|
|
|
/**
|
|
|
|
* @dataProvider dataTestAddServer
|
|
|
|
*
|
|
|
|
* @param string $url passed to the method
|
|
|
|
* @param string $expectedUrl the url we expect to be written to the db
|
|
|
|
* @param string $expectedHash the hash value we expect to be written to the db
|
|
|
|
*/
|
|
|
|
public function testAddServer($url, $expectedUrl, $expectedHash) {
|
|
|
|
$id = $this->dbHandler->addServer($url);
|
2015-10-29 16:27:14 +00:00
|
|
|
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
2015-11-24 12:07:40 +00:00
|
|
|
$this->assertSame($expectedUrl, $result[0]['url']);
|
2015-11-19 16:49:43 +00:00
|
|
|
$this->assertSame($id, (int)$result[0]['id']);
|
2015-11-24 12:07:40 +00:00
|
|
|
$this->assertSame($expectedHash, $result[0]['url_hash']);
|
2015-11-19 16:49:43 +00:00
|
|
|
$this->assertSame(TrustedServers::STATUS_PENDING, (int)$result[0]['status']);
|
2015-10-29 16:27:14 +00:00
|
|
|
}
|
|
|
|
|
2015-11-24 12:07:40 +00:00
|
|
|
public function dataTestAddServer() {
|
|
|
|
return [
|
|
|
|
['http://owncloud.org', 'http://owncloud.org', md5('owncloud.org')],
|
|
|
|
['https://owncloud.org', 'https://owncloud.org', md5('owncloud.org')],
|
|
|
|
['http://owncloud.org/', 'http://owncloud.org', md5('owncloud.org')],
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
2015-10-29 16:27:14 +00:00
|
|
|
public function testRemove() {
|
2015-11-10 09:50:59 +00:00
|
|
|
$id1 = $this->dbHandler->addServer('server1');
|
|
|
|
$id2 = $this->dbHandler->addServer('server2');
|
2015-10-29 16:27:14 +00:00
|
|
|
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(2, count($result));
|
|
|
|
$this->assertSame('server1', $result[0]['url']);
|
|
|
|
$this->assertSame('server2', $result[1]['url']);
|
2015-11-19 16:49:43 +00:00
|
|
|
$this->assertSame($id1, (int)$result[0]['id']);
|
|
|
|
$this->assertSame($id2, (int)$result[1]['id']);
|
2015-10-29 16:27:14 +00:00
|
|
|
|
2015-11-10 09:50:59 +00:00
|
|
|
$this->dbHandler->removeServer($id2);
|
2015-10-29 16:27:14 +00:00
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame('server1', $result[0]['url']);
|
2015-11-19 16:49:43 +00:00
|
|
|
$this->assertSame($id1, (int)$result[0]['id']);
|
2015-10-29 16:27:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetAll() {
|
2015-11-10 09:50:59 +00:00
|
|
|
$id1 = $this->dbHandler->addServer('server1');
|
|
|
|
$id2 = $this->dbHandler->addServer('server2');
|
2015-10-29 16:27:14 +00:00
|
|
|
|
2015-11-10 09:50:59 +00:00
|
|
|
$result = $this->dbHandler->getAllServer();
|
2015-10-29 16:27:14 +00:00
|
|
|
$this->assertSame(2, count($result));
|
|
|
|
$this->assertSame('server1', $result[0]['url']);
|
|
|
|
$this->assertSame('server2', $result[1]['url']);
|
2015-11-19 16:49:43 +00:00
|
|
|
$this->assertSame($id1, (int)$result[0]['id']);
|
|
|
|
$this->assertSame($id2, (int)$result[1]['id']);
|
2015-10-29 16:27:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2015-11-19 16:49:43 +00:00
|
|
|
* @dataProvider dataTestServerExists
|
2015-10-29 16:27:14 +00:00
|
|
|
*
|
|
|
|
* @param string $serverInTable
|
|
|
|
* @param string $checkForServer
|
|
|
|
* @param bool $expected
|
|
|
|
*/
|
2015-11-19 16:49:43 +00:00
|
|
|
public function testServerExists($serverInTable, $checkForServer, $expected) {
|
2015-11-10 09:50:59 +00:00
|
|
|
$this->dbHandler->addServer($serverInTable);
|
2015-10-29 16:27:14 +00:00
|
|
|
$this->assertSame($expected,
|
2015-11-10 09:50:59 +00:00
|
|
|
$this->dbHandler->serverExists($checkForServer)
|
2015-10-29 16:27:14 +00:00
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2015-11-19 16:49:43 +00:00
|
|
|
public function dataTestServerExists() {
|
2015-10-29 16:27:14 +00:00
|
|
|
return [
|
|
|
|
['server1', 'server1', true],
|
2015-11-19 16:49:43 +00:00
|
|
|
['server1', 'http://server1', true],
|
2015-10-29 16:27:14 +00:00
|
|
|
['server1', 'server2', false]
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
2015-11-19 16:49:43 +00:00
|
|
|
public function testAddToken() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame(null, $result[0]['token']);
|
|
|
|
$this->dbHandler->addToken('http://server1', 'token');
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame('token', $result[0]['token']);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetToken() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$this->dbHandler->addToken('http://server1', 'token');
|
|
|
|
$this->assertSame('token',
|
|
|
|
$this->dbHandler->getToken('https://server1')
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testAddSharedSecret() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame(null, $result[0]['shared_secret']);
|
|
|
|
$this->dbHandler->addSharedSecret('http://server1', 'secret');
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame('secret', $result[0]['shared_secret']);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetSharedSecret() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$this->dbHandler->addSharedSecret('http://server1', 'secret');
|
|
|
|
$this->assertSame('secret',
|
|
|
|
$this->dbHandler->getSharedSecret('https://server1')
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testSetServerStatus() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame(TrustedServers::STATUS_PENDING, (int)$result[0]['status']);
|
|
|
|
$this->dbHandler->setServerStatus('http://server1', TrustedServers::STATUS_OK);
|
|
|
|
$query = $this->connection->getQueryBuilder()->select('*')->from($this->dbTable);
|
|
|
|
$result = $query->execute()->fetchAll();
|
|
|
|
$this->assertSame(1, count($result));
|
|
|
|
$this->assertSame(TrustedServers::STATUS_OK, (int)$result[0]['status']);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetServerStatus() {
|
|
|
|
$this->dbHandler->addServer('server1');
|
|
|
|
$this->dbHandler->setServerStatus('http://server1', TrustedServers::STATUS_OK);
|
|
|
|
$this->assertSame(TrustedServers::STATUS_OK,
|
|
|
|
$this->dbHandler->getServerStatus('https://server1')
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* hash should always be computed with the normalized URL
|
|
|
|
*
|
|
|
|
* @dataProvider dataTestHash
|
|
|
|
*
|
|
|
|
* @param string $url
|
|
|
|
* @param string $expected
|
|
|
|
*/
|
|
|
|
public function testHash($url, $expected) {
|
|
|
|
$this->assertSame($expected,
|
|
|
|
$this->invokePrivate($this->dbHandler, 'hash', [$url])
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function dataTestHash() {
|
|
|
|
return [
|
|
|
|
['server1', md5('server1')],
|
|
|
|
['http://server1', md5('server1')],
|
|
|
|
['https://server1', md5('server1')],
|
|
|
|
['http://server1/', md5('server1')],
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
2015-11-10 09:50:59 +00:00
|
|
|
/**
|
|
|
|
* @dataProvider dataTestNormalizeUrl
|
|
|
|
*
|
|
|
|
* @param string $url
|
|
|
|
* @param string $expected
|
|
|
|
*/
|
|
|
|
public function testNormalizeUrl($url, $expected) {
|
|
|
|
$this->assertSame($expected,
|
|
|
|
$this->invokePrivate($this->dbHandler, 'normalizeUrl', [$url])
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function dataTestNormalizeUrl() {
|
|
|
|
return [
|
|
|
|
['owncloud.org', 'owncloud.org'],
|
|
|
|
['http://owncloud.org', 'owncloud.org'],
|
|
|
|
['https://owncloud.org', 'owncloud.org'],
|
|
|
|
['https://owncloud.org//mycloud', 'owncloud.org/mycloud'],
|
|
|
|
['https://owncloud.org/mycloud/', 'owncloud.org/mycloud'],
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
2015-10-29 16:27:14 +00:00
|
|
|
}
|