namespace Friendica\Test\Util;
+use Friendica\Database\DBA;
use Mockery\MockInterface;
+class DBAStub
+{
+ public static $connected = true;
+}
+
/**
* Trait to mock the DBA connection status
*/
*/
private $dbaMock;
+ private function checkMock()
+ {
+ if (!isset($this->dbaMock)) {
+ $this->dbaMock = \Mockery::namedMock(DBA::class, DBAStub::class);
+ }
+ }
+
/**
* Mocking DBA::connect()
*
*/
public function mockConnect($return = true, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('connect')
*/
public function mockConnected($return = true, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('connected')
*/
public function mockFetchFirst($arg, $return = true, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('fetchFirst')
->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()
*/
public function mockSelect($tableName, $select = [], $where = [], $return = null, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('select')
}
/**
- * Mocking DBA::selectFirst()
+ * Mocking DBA::delete()
*
* @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 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($tableName, $select = [], $where = [], $return = [], $times = null)
+ public function mockSelectFirst($expTableName, $expSelect = [], $expWhere = [], $return = [], $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
+
+ $closure = function ($tableName, $select = [], $where = []) use ($expTableName, $expSelect, $expWhere) {
+ return $tableName === $expTableName
+ && $select === $expSelect
+ && $where === $expWhere;
+ };
$this->dbaMock
->shouldReceive('selectFirst')
- ->with($tableName, $select, $where)
+ ->withArgs($closure)
->times($times)
->andReturn($return);
}
*/
public function mockIsResult($record, $return = true, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('isResult')
*/
public function mockToArray($record = null, $return = [], $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
$this->dbaMock
->shouldReceive('toArray')
->andReturn($return);
}
-
/**
* Mocking DBA::p()
*
*/
public function mockP($sql = null, $return = null, $times = null)
{
- if (!isset($this->dbaMock)) {
- $this->dbaMock = \Mockery::mock('alias:Friendica\Database\DBA');
- }
+ $this->checkMock();
if (!isset($sql)) {
$this->dbaMock
->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);
+ }
}
projects / friendica.git / blobdiff