friendica/tests/Util/DBStructureMockTrait.php

38 lines
791 B
PHP
Raw Normal View History

<?php
namespace Friendica\Test\Util;
/**
* Trait to mock the DBStructure connection status
*/
trait DBStructureMockTrait
{
private $dbStructure;
public function mockUpdate($args = [], $return = true, $times = null)
{
if (!isset($this->dbStructure)) {
$this->dbStructure = \Mockery::mock('alias:Friendica\Database\DBStructure');
}
$this->dbStructure
->shouldReceive('update')
->withArgs($args)
->times($times)
->andReturn($return);
}
public function mockExistsTable($tableName, $return = true, $times = null)
{
if (!isset($this->dbStructure)) {
$this->dbStructure = \Mockery::mock('alias:Friendica\Database\DBStructure');
}
$this->dbStructure
->shouldReceive('existsTable')
->with($tableName)
->times($times)
->andReturn($return);
}
}