140 lines
3.5 KiB
PHP
140 lines
3.5 KiB
PHP
<?php
|
|
/**
|
|
* @author Robin McCorkell <rmccorkell@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 Test\Security;
|
|
|
|
use OC\Security\CredentialsManager;
|
|
use OCP\DB\IResult;
|
|
use OCP\DB\QueryBuilder\IExpressionBuilder;
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IDBConnection;
|
|
use OCP\Security\ICrypto;
|
|
|
|
/**
|
|
* @group DB
|
|
*/
|
|
class CredentialsManagerTest extends \Test\TestCase {
|
|
|
|
/** @var ICrypto */
|
|
protected $crypto;
|
|
|
|
/** @var IDBConnection */
|
|
protected $dbConnection;
|
|
|
|
/** @var CredentialsManager */
|
|
protected $manager;
|
|
|
|
protected function setUp(): void {
|
|
parent::setUp();
|
|
$this->crypto = $this->createMock(ICrypto::class);
|
|
$this->dbConnection = $this->getMockBuilder(IDBConnection::class)
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
$this->manager = new CredentialsManager($this->crypto, $this->dbConnection);
|
|
}
|
|
|
|
private function getQueryResult($row) {
|
|
$result = $this->createMock(IResult::class);
|
|
|
|
$result->expects($this->any())
|
|
->method('fetch')
|
|
->willReturn($row);
|
|
|
|
return $result;
|
|
}
|
|
|
|
public function testStore() {
|
|
$userId = 'abc';
|
|
$identifier = 'foo';
|
|
$credentials = 'bar';
|
|
|
|
$this->crypto->expects($this->once())
|
|
->method('encrypt')
|
|
->with(json_encode($credentials))
|
|
->willReturn('baz');
|
|
|
|
$this->dbConnection->expects($this->once())
|
|
->method('setValues')
|
|
->with(CredentialsManager::DB_TABLE,
|
|
['user' => $userId, 'identifier' => $identifier],
|
|
['credentials' => 'baz']
|
|
);
|
|
|
|
$this->manager->store($userId, $identifier, $credentials);
|
|
}
|
|
|
|
public function testRetrieve() {
|
|
$userId = 'abc';
|
|
$identifier = 'foo';
|
|
|
|
$this->crypto->expects($this->once())
|
|
->method('decrypt')
|
|
->with('baz')
|
|
->willReturn(json_encode('bar'));
|
|
|
|
$eb = $this->createMock(IExpressionBuilder::class);
|
|
$qb = $this->createMock(IQueryBuilder::class);
|
|
$qb->method('select')->willReturnSelf();
|
|
$qb->method('from')->willReturnSelf();
|
|
$qb->method('where')->willReturnSelf();
|
|
$qb->method('expr')->willReturn($eb);
|
|
$qb->expects($this->once())
|
|
->method('execute')
|
|
->willReturn($this->getQueryResult(['credentials' => 'baz']));
|
|
|
|
$this->dbConnection->expects($this->once())
|
|
->method('getQueryBuilder')
|
|
->willReturn($qb);
|
|
|
|
$this->manager->retrieve($userId, $identifier);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider credentialsProvider
|
|
*/
|
|
public function testWithDB($userId, $identifier) {
|
|
$credentialsManager = \OC::$server->getCredentialsManager();
|
|
|
|
$secrets = 'Open Sesame';
|
|
|
|
$credentialsManager->store($userId, $identifier, $secrets);
|
|
$received = $credentialsManager->retrieve($userId, $identifier);
|
|
|
|
$this->assertSame($secrets, $received);
|
|
|
|
$removedRows = $credentialsManager->delete($userId, $identifier);
|
|
$this->assertSame(1, $removedRows);
|
|
}
|
|
|
|
public function credentialsProvider() {
|
|
return [
|
|
[
|
|
'alice',
|
|
'privateCredentials'
|
|
],
|
|
[
|
|
'',
|
|
'systemCredentials',
|
|
],
|
|
];
|
|
}
|
|
}
|