mirror of https://github.com/friendica/friendica
104 lines
2.7 KiB
PHP
104 lines
2.7 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\Core\Cache\Duration;
|
|
|
|
trait DbaCacheMockTrait
|
|
{
|
|
/**
|
|
* @var
|
|
*/
|
|
protected $dba;
|
|
|
|
public function __construct()
|
|
{
|
|
}
|
|
|
|
protected function mockDelete($key, $return = true, $times = null)
|
|
{
|
|
$this->mockDBADelete('cache', ['k' => $key], $return, $times);
|
|
}
|
|
|
|
protected function mockGet($key, $return = null, $time = null, $times = null)
|
|
{
|
|
if ($time === null) {
|
|
$time = time();
|
|
}
|
|
|
|
$value = @serialize($return);
|
|
|
|
$this->mockSelectFirst('cache', ['v'], ['`k` = ? AND (`expires` >= ? OR `expires` = -1)', $key, $time], ['v' => $value], $times);
|
|
$this->mockIsResult(['v' => $value], isset($return), $times);
|
|
}
|
|
|
|
protected function mockSet($key, $value, $ttl = Duration::FIVE_MINUTES, $time = null, $return = true, $times = null)
|
|
{
|
|
if ($time === null) {
|
|
$time = time();
|
|
}
|
|
|
|
if ($ttl > 0) {
|
|
$this->mockUtc('now + ' . $ttl . 'seconds', $time + $ttl, $times);
|
|
$fields = [
|
|
'v' => serialize($value),
|
|
'expires' => $time + $ttl,
|
|
'updated' => $time
|
|
];
|
|
} else {
|
|
$fields = [
|
|
'v' => serialize($value),
|
|
'expires' => -1,
|
|
'updated' => $time
|
|
];
|
|
}
|
|
|
|
$this->mockDBAUpdate('cache', $fields, ['k' => $key], true, $return, $times);
|
|
}
|
|
|
|
protected function mockClear($outdated = true, $return = true, $times = null)
|
|
{
|
|
if ($outdated) {
|
|
$this->mockDBADelete('cache', ['`expires` < NOW()'], $return, $times);
|
|
} else {
|
|
$this->mockDBADelete('cache', ['`k` IS NOT NULL '], $return, $times);
|
|
}
|
|
}
|
|
|
|
protected function mockGetAllKeys($prefix = null, $return = [], $time = null, $times = null)
|
|
{
|
|
if ($time === null) {
|
|
$time = time();
|
|
}
|
|
|
|
if (empty($prefix)) {
|
|
$where = ['`expires` >= ?', $time];
|
|
} else {
|
|
$where = ['`expires` >= ? AND `k` LIKE CONCAT(?, \'%\')', $time, $prefix];
|
|
}
|
|
|
|
$this->mockSelect('cache', ['k'], $where, $return, $times);
|
|
$this->mockFetchLoop($return, $times);
|
|
$this->mockDbaClose(true, $times);
|
|
}
|
|
}
|