354 lines
8.9 KiB
PHP
354 lines
8.9 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (C) 2020, Friendica
|
|
*
|
|
* @license GNU AGPL version 3 or any later version
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* 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
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
namespace Friendica\Test\Util;
|
|
|
|
use Friendica\Database\DBA;
|
|
use Mockery\MockInterface;
|
|
|
|
class DBAStub
|
|
{
|
|
public static $connected = true;
|
|
}
|
|
|
|
/**
|
|
* Trait to mock the DBA connection status
|
|
*/
|
|
trait DBAMockTrait
|
|
{
|
|
/**
|
|
* @var MockInterface The mocking interface of Friendica\Database\DBA
|
|
*/
|
|
private $dbaMock;
|
|
|
|
private function checkMock()
|
|
{
|
|
if (!isset($this->dbaMock)) {
|
|
$this->dbaMock = \Mockery::namedMock(DBA::class, DBAStub::class);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::connect()
|
|
*
|
|
* @param bool $return True, if the connect was successful, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockConnect($return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('connect')
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::connected()
|
|
*
|
|
* @param bool $return True, if the DB is connected, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockConnected($return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('connected')
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::fetchFirst()
|
|
*
|
|
* @param string $arg The argument of fetchFirst
|
|
* @param bool $return True, if the DB is connected, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockFetchFirst($arg, $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('fetchFirst')
|
|
->with($arg)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking each DBA::fetch() call of an statement
|
|
*
|
|
* @param array $stmt The result statement (array)
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockFetchLoop($stmt = [], $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
foreach ($stmt as $item) {
|
|
$this->dbaMock
|
|
->shouldReceive('fetch')
|
|
->times($times)
|
|
->andReturn($item);
|
|
}
|
|
|
|
// The last mock call of a fetch (=> breaking the loop)
|
|
$this->dbaMock
|
|
->shouldReceive('fetch')
|
|
->times($times)
|
|
->andReturn(false);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::close()
|
|
*
|
|
* @param array $return The return per fetch
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDbaClose($return = [], $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('close')
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::select()
|
|
*
|
|
* @param string $tableName The name of the table
|
|
* @param array $select The Select Array (Default is [])
|
|
* @param array $where The Where Array (Default is [])
|
|
* @param object $return The array to return (Default is [])
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockSelect($tableName, $select = [], $where = [], $return = null, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('select')
|
|
->with($tableName, $select, $where)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::delete()
|
|
*
|
|
* @param string $tableName The name of the table
|
|
* @param array $where The Where Array (Default is [])
|
|
* @param bool $return The array to return (Default is true)
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDBADelete($tableName, $where = [], $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('delete')
|
|
->with($tableName, $where)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::update()
|
|
*
|
|
* @param string $expTableName The name of the table
|
|
* @param array $expFields The Fields Array
|
|
* @param array $expCondition The Condition Array
|
|
* @param array $expOld_fields The Old Fieldnames (Default is [])
|
|
* @param bool $return true if the update was successful
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDBAUpdate($expTableName, $expFields, $expCondition, $expOld_fields = [], $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$closure = function ($tableName, $fields, $condition, $old_fields = []) use ($expTableName, $expFields, $expCondition, $expOld_fields) {
|
|
return
|
|
$tableName == $expTableName &&
|
|
$fields == $expFields &&
|
|
$condition == $expCondition &&
|
|
$old_fields == $expOld_fields;
|
|
};
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('update')
|
|
->withArgs($closure)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::insert()
|
|
*
|
|
* @param string $expTableName The name of the table
|
|
* @param array $expParam The Parameters Array
|
|
* @param bool $expOnDuplUpdate Update on a duplicated entry
|
|
* @param bool $return True if the insert was successful
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDBAInsert($expTableName, $expParam, $expOnDuplUpdate = false, $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$closure = function ($tableName, $param, $on_duplicate_update = false) use ($expTableName, $expParam, $expOnDuplUpdate) {
|
|
return $tableName == $expTableName
|
|
&& $param == $expParam
|
|
&& $on_duplicate_update == $expOnDuplUpdate;
|
|
|
|
};
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('insert')
|
|
->withArgs($closure)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::selectFirst()
|
|
*
|
|
* @param string $expTableName The name of the table
|
|
* @param array $expSelect The Select Array (Default is [])
|
|
* @param array $expWhere The Where Array (Default is [])
|
|
* @param array $return The array to return (Default is [])
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockSelectFirst($expTableName, $expSelect = [], $expWhere = [], $return = [], $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$closure = function ($tableName, $select = [], $where = []) use ($expTableName, $expSelect, $expWhere) {
|
|
return $tableName === $expTableName
|
|
&& $select === $expSelect
|
|
&& $where === $expWhere;
|
|
};
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('selectFirst')
|
|
->withArgs($closure)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::isResult()
|
|
*
|
|
* @param object $record The record to test
|
|
* @param bool $return True, if the DB is connected, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockIsResult($record, $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('isResult')
|
|
->with($record)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::isResult()
|
|
*
|
|
* @param object $record The record to test
|
|
* @param array $return The array to return
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockToArray($record = null, $return = [], $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('toArray')
|
|
->with($record)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::p()
|
|
*
|
|
* @param string $sql The SQL statement
|
|
* @param object $return The object to return
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockP($sql = null, $return = null, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
if (!isset($sql)) {
|
|
$this->dbaMock
|
|
->shouldReceive('p')
|
|
->times($times)
|
|
->andReturn($return);
|
|
} else {
|
|
$this->dbaMock
|
|
->shouldReceive('p')
|
|
->with($sql)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::lock()
|
|
*
|
|
* @param string $table The table to lock
|
|
* @param bool $return True, if the lock is set successful
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDbaLock($table, $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('lock')
|
|
->with($table)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBA::unlock()
|
|
*
|
|
* @param bool $return True, if the lock is set successful
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockDbaUnlock( $return = true, $times = null)
|
|
{
|
|
$this->checkMock();
|
|
|
|
$this->dbaMock
|
|
->shouldReceive('unlock')
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
}
|