78 lines
2.2 KiB
PHP
78 lines
2.2 KiB
PHP
<?php
|
|
/**
|
|
* @copyright Copyright (C) 2010-2021, the Friendica project
|
|
*
|
|
* @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\DBStructure;
|
|
use Mockery\MockInterface;
|
|
|
|
/**
|
|
* Trait to mock the DBStructure connection status
|
|
*/
|
|
trait DBStructureMockTrait
|
|
{
|
|
/**
|
|
* @var MockInterface The mocking interface of Friendica\Database\DBStructure
|
|
*/
|
|
private $dbStructure;
|
|
|
|
/**
|
|
* Mocking DBStructure::update()
|
|
* @see DBStructure::update();
|
|
*
|
|
* @param array $args The arguments for the update call
|
|
* @param bool $return True, if the connect was successful, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockUpdate(array $args = [], bool $return = true, int $times = null)
|
|
{
|
|
if (!isset($this->dbStructure)) {
|
|
$this->dbStructure = \Mockery::mock('alias:' . DBStructure::class);
|
|
}
|
|
|
|
$this->dbStructure
|
|
->shouldReceive('update')
|
|
->withArgs($args)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
|
|
/**
|
|
* Mocking DBStructure::existsTable()
|
|
*
|
|
* @param string $tableName The name of the table to check
|
|
* @param bool $return True, if the connect was successful, otherwise false
|
|
* @param null|int $times How often the method will get used
|
|
*/
|
|
public function mockExistsTable(string $tableName, bool $return = true, int $times = null)
|
|
{
|
|
if (!isset($this->dbStructure)) {
|
|
$this->dbStructure = \Mockery::mock('alias:' . DBStructure::class);
|
|
}
|
|
|
|
$this->dbStructure
|
|
->shouldReceive('existsTable')
|
|
->with($tableName)
|
|
->times($times)
|
|
->andReturn($return);
|
|
}
|
|
}
|