friendica_2021-01/tests/Util/DbaCacheMockTrait.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);
	}
}