88 lines
2.7 KiB
PHP
88 lines
2.7 KiB
PHP
|
<?php
|
|||
|
|
declare(strict_types=1);
|
|||
|
|
namespace tests;
|
|||
|
|
|
|||
|
|
use PHPUnit\Framework\TestCase;
|
|||
|
|
|
|||
|
|
abstract class Base extends TestCase {
|
|||
|
|
static $ROOT_PATH = __DIR__ . "/../vendor/topthink/think";
|
|||
|
|
static $RUNTIME_PATH = __DIR__ . "/../runtime/";
|
|||
|
|
|
|||
|
|
protected $app;
|
|||
|
|
protected $throttle_config = [];
|
|||
|
|
protected $middleware_file = __DIR__ . "/config/global-middleware.php";
|
|||
|
|
protected $middleware_type = 'global';
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* thinkphp 一般运行在 php-fpm 模式下，每次处理请求都要重新加载配置文件
|
|||
|
|
* @param \think\Request $request
|
|||
|
|
* @return \think\Response
|
|||
|
|
*/
|
|||
|
|
function get_response(\think\Request $request): \think\Response {
|
|||
|
|
// 创建 \think\App 对象，设置配置
|
|||
|
|
$app = new GCApp(static::$ROOT_PATH);
|
|||
|
|
$app->setRuntimePath(static::$RUNTIME_PATH);
|
|||
|
|
|
|||
|
|
// 加载中间件
|
|||
|
|
$app->middleware->import(include $this->middleware_file, $this->middleware_type);
|
|||
|
|
// 设置 throttle 配置
|
|||
|
|
$app->config->set($this->throttle_config, 'throttle');
|
|||
|
|
|
|||
|
|
$response = $app->http->run($request);
|
|||
|
|
$app->refClear();
|
|||
|
|
return $response;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected function tearDown(): void
|
|||
|
|
{
|
|||
|
|
parent::tearDown();
|
|||
|
|
// 每次测试完毕都需要清理 runtime cache 目录，避免影响其他单元测试
|
|||
|
|
$cache_dir = static::$RUNTIME_PATH . "cache";
|
|||
|
|
$dirs = glob($cache_dir . '/*', GLOB_ONLYDIR);
|
|||
|
|
foreach ($dirs as $dir) {
|
|||
|
|
$files = glob($dir . '/*.php');
|
|||
|
|
foreach ($files as $file) {
|
|||
|
|
unlink($file);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 删除 cache 下的空目录
|
|||
|
|
foreach ($dirs as $dir) {
|
|||
|
|
rmdir($dir);
|
|||
|
|
}
|
|||
|
|
unset($cache_dir);
|
|||
|
|
unset($dirs);
|
|||
|
|
gc_collect_cycles(); // 进行垃圾回收
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取默认的 throttle 基础配置信息
|
|||
|
|
* @return array
|
|||
|
|
*/
|
|||
|
|
function get_default_throttle_config(): array {
|
|||
|
|
static $config = []; // 默认配置从文件中读取，可以设置为静态变量
|
|||
|
|
if (!$config) {
|
|||
|
|
$config = include dirname(__DIR__) . "/src/config.php";
|
|||
|
|
}
|
|||
|
|
return $config;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 设置中间件配置文件
|
|||
|
|
* @param string $file 文件的路径 eg: $this->app->getBasePath() . 'middleware.php'
|
|||
|
|
* @param string $type 类型：global 全局；route 路由；controller 控制器
|
|||
|
|
*/
|
|||
|
|
function set_middleware(string $file, string $type = 'global') {
|
|||
|
|
$this->middleware_file = $file;
|
|||
|
|
$this->middleware_type = $type;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 设置 throttle 配置
|
|||
|
|
* @param array $config
|
|||
|
|
*/
|
|||
|
|
function set_throttle_config(array $config) {
|
|||
|
|
$this->throttle_config = $config;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|