Fix security credentials manager test

Signed-off-by: Joas Schilling <coding@schilljs.com>
This commit is contained in:
Joas Schilling 2021-04-20 17:03:29 +02:00
parent 94322971a5
commit c6978bac80
No known key found for this signature in database
GPG Key ID: 7076EA9751AACDDA
1 changed files with 11 additions and 18 deletions

View File

@ -24,6 +24,9 @@ namespace Test\Security;
use OC\DB\ConnectionAdapter;
use OC\Security\CredentialsManager;
use OC\SystemConfig;
use OCP\DB\IResult;
use OCP\DB\QueryBuilder\IExpressionBuilder;
use OCP\DB\QueryBuilder\IQueryBuilder;
use OCP\IDBConnection;
use OCP\ILogger;
use OCP\Security\ICrypto;
@ -39,9 +42,6 @@ class CredentialsManagerTest extends \Test\TestCase {
/** @var IDBConnection */
protected $dbConnection;
/** @var ConnectionAdapter */
protected $dbConnectionAdapter;
/** @var CredentialsManager */
protected $manager;
@ -51,16 +51,11 @@ class CredentialsManagerTest extends \Test\TestCase {
$this->dbConnection = $this->getMockBuilder(IDBConnection::class)
->disableOriginalConstructor()
->getMock();
$this->dbConnectionAdapter = $this->getMockBuilder(ConnectionAdapter::class)
->disableOriginalConstructor()
->getMock();
$this->manager = new CredentialsManager($this->crypto, $this->dbConnection);
}
private function getQueryResult($row) {
$result = $this->getMockBuilder('\Doctrine\DBAL\Driver\Statement')
->disableOriginalConstructor()
->getMock();
$result = $this->createMock(IResult::class);
$result->expects($this->any())
->method('fetch')
@ -98,19 +93,17 @@ class CredentialsManagerTest extends \Test\TestCase {
->with('baz')
->willReturn(json_encode('bar'));
$qb = $this->getMockBuilder(\OC\DB\QueryBuilder\QueryBuilder::class)
->setConstructorArgs([
$this->dbConnectionAdapter,
$this->createMock(SystemConfig::class),
$this->createMock(ILogger::class),
])
->setMethods(['execute'])
->getMock();
$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->dbConnectionAdapter->expects($this->once())
$this->dbConnection->expects($this->once())
->method('getQueryBuilder')
->willReturn($qb);