From fa56140606e9df69aeefa1008fdd927bf845130a Mon Sep 17 00:00:00 2001
From: hantianfeng <rango@swoole.com>
Date: Wed, 11 Sep 2024 14:58:19 +0800
Subject: [PATCH] Optimize tests

---
 core-tests/src/os/async.cpp                   |  2 +-
 tests/swoole_server/task/task_ipc_mode_2.phpt | 19 ++++++++++++++-----
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/core-tests/src/os/async.cpp b/core-tests/src/os/async.cpp
index 8dad920991e..8a122069db7 100644
--- a/core-tests/src/os/async.cpp
+++ b/core-tests/src/os/async.cpp
@@ -84,7 +84,7 @@ TEST(async, schedule) {
                 break;
             } else if (count == N - 1) {
                 ASSERT_EQ(SwooleTG.async_threads->get_worker_num(), 4);
-                ASSERT_EQ(SwooleTG.async_threads->get_queue_size(), 1);
+                ASSERT_LE(SwooleTG.async_threads->get_queue_size(), 1);
                 ASSERT_EQ(SwooleTG.async_threads->get_task_num(), 1);
             } else if (count < N / 2) {
                 ASSERT_GT(SwooleTG.async_threads->get_worker_num(), 4);
diff --git a/tests/swoole_server/task/task_ipc_mode_2.phpt b/tests/swoole_server/task/task_ipc_mode_2.phpt
index f0013f390b5..bb92da15b0e 100644
--- a/tests/swoole_server/task/task_ipc_mode_2.phpt
+++ b/tests/swoole_server/task/task_ipc_mode_2.phpt
@@ -5,31 +5,40 @@ swoole_server/task: task_ipc_mode = 2
 --FILE--
 <?php
 require __DIR__ . '/../../include/bootstrap.php';
+
+use Swoole\Atomic;
+use Swoole\Http\Request;
+use Swoole\Http\Response;
+
 $pm = new SwooleTest\ProcessManager;
+$atomic = new Atomic;
 $pm->parentFunc = function ($pid) use ($pm) {
     go(function () use ($pm) {
         echo httpGetBody("http://127.0.0.1:{$pm->getFreePort()}/");
         $pm->kill();
     });
 };
-$pm->childFunc = function () use ($pm) {
+$pm->childFunc = function () use ($pm, $atomic) {
     $server = new Swoole\Http\Server('127.0.0.1', $pm->getFreePort(), SERVER_MODE_RANDOM);
     $server->set([
         'log_file' => '/dev/null',
         'open_tcp_nodelay' => true,
         'task_worker_num' => 4,
+        'worker_num' => 2,
         'task_ipc_mode' => 2,
         'dispatch_mode' => 2
     ]);
-    $server->on('workerStart', function () use ($pm) {
-        $pm->wakeup();
+    $server->on('workerStart', function () use ($pm, $atomic) {
+        if ($atomic->add() == 6) {
+            $pm->wakeup();
+        }
     });
-    $server->on('request', function (Swoole\Http\Request $request, Swoole\Http\Response $response) use ($server) {
+    $server->on('request', function (Request $request, Response $response) use ($server) {
         $response->detach();
         $server->task($response->fd);
     });
     $server->on('task', function ($server, $task_id, $worker_id, string $fd) {
-        $response = Swoole\Http\Response::create($fd);
+        $response = Response::create($fd);
         $response->end("Hello Swoole!\n");
     });
     $server->on('finish', function () { });