添加常驻内存型的单元测试 (#29)

Author: hongweipeng

tests/BaseTest.php

@@ -27,7 +27,7 @@ function get_response(\think\Request $request): \think\Response {
         // 设置 throttle 配置
         $app->config->set($this->throttle_config, 'throttle');
 
-        $response =  $app->http->run($request);
+        $response = $app->http->run($request);
         $app->refClear();
         return $response;
     }

tests/ResidentMemoryTest.php

@@ -0,0 +1,51 @@
+<?php
+
+
+namespace tests;
+
+
+/**
+ * 常驻内存型的单元测试，当 TP 运行在常驻内存型的时候。
+ * 一个 App 实例，处理多次请求
+ * Class ResidentMemoryTest
+ * @package tests
+ */
+class ResidentMemoryTest extends BaseTest
+{
+    public function test_resident_memory()
+    {
+        $app = new GCApp(static::$ROOT_PATH);
+        $app->setRuntimePath(static::$RUNTIME_PATH);
+        $app->middleware->import(include $this->middleware_file, $this->middleware_type);
+        $app->config->set($this->get_default_throttle_config(), 'throttle');
+
+        // 处理多个请求
+        $allowCount1 = 0;
+        $allowCount2 = 0;
+
+        for ($i = 0; $i < 200; $i++) {
+            // 受访问频率限制
+            $request = new \think\Request();
+            $request->setMethod('GET');
+            $request->setUrl('/');
+            $response = $app->http->run($request);
+            if ($response->getCode() == 200) {
+                $allowCount1++;
+            }
+
+            // 不受访问频率限制
+            $request = new \think\Request();
+            $request->setMethod('POST');
+            $request->setUrl('/');
+
+            $response = $app->http->run($request);
+            if ($response->getCode() == 200) {
+                $allowCount2++;
+            }
+        }
+
+        $app->refClear();
+        $this->assertEquals(100, $allowCount1);
+        $this->assertEquals(200, $allowCount2);
+    }
+}