Added http chunk test for Coroutine\Http\Client; Optimize write_func …

…tests --filter=[unit]
swoole · Nov 18, 2024 · 3562377 · 3562377
1 parent c4b3271
commit 3562377
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 0 deletions.
diff --git a/tests/swoole_http_client_coro/http_chunk.phpt b/tests/swoole_http_client_coro/http_chunk.phpt
@@ -0,0 +1,52 @@
+--TEST--
+swoole_http_client_coro: http chunk
+--SKIPIF--
+<?php require __DIR__ . '/../include/skipif.inc'; ?>
+--FILE--
+<?php
+require __DIR__ . '/../include/bootstrap.php';
+
+use Swoole\Event;
+
+const N = 8;
+$chunks = [];
+$body = '';
+$n = N;
+while ($n--) {
+    $chunk = base64_encode(random_bytes(random_int(256, 4096)));
+    $body .= $chunk;
+    $chunks[] = $chunk;
+}
+
+$pm = new ProcessManager;
+$pm->parentFunc = function ($pid) use ($pm, $chunks, $body) {
+    Co\run(function () use ($pm, $chunks, $body) {
+        $cli = new Swoole\Coroutine\Http\Client('127.0.0.1', $pm->getFreePort());
+        Assert::assert($cli->get('/'));
+        Assert::eq($cli->getBody(), $body);
+    });
+    $pm->kill();
+    echo "DONE\n";
+};
+
+$pm->childFunc = function () use ($pm, $chunks) {
+    Co\run(function () use ($pm, $chunks) {
+        Event::defer(function () use ($pm) {
+            $pm->wakeup();
+        });
+        $server = new Swoole\Coroutine\Http\Server('127.0.0.1', $pm->getFreePort());
+        $server->handle('/', function ($req, $resp) use ($server, $chunks) {
+            foreach ($chunks as $chunk) {
+                $resp->write($chunk);
+                usleep(mt_rand(10, 50) * 100);
+            }
+        });
+        $server->start();
+    });
+};
+
+$pm->childFirst();
+$pm->run();
+?>
+--EXPECT--
+DONE
diff --git a/tests/swoole_http_client_coro/write_func_1.phpt b/tests/swoole_http_client_coro/write_func_1.phpt
@@ -6,6 +6,8 @@ swoole_http_client_coro: write func 1
 <?php
 require __DIR__ . '/../include/bootstrap.php';
 
+use Swoole\Event;
+
 const N = 8;
 $chunks = [];
 $n = N;
@@ -30,6 +32,9 @@ $pm->parentFunc = function ($pid) use ($pm, $chunks) {
 
 $pm->childFunc = function () use ($pm, $chunks) {
     Co\run(function () use ($pm, $chunks) {
+        Event::defer(function () use ($pm) {
+            $pm->wakeup();
+        });
         $server = new Swoole\Coroutine\Http\Server('127.0.0.1', $pm->getFreePort());
         $server->handle('/', function ($req, $resp) use ($server, $chunks) {
             foreach ($chunks as $chunk) {

diff --git a/tests/swoole_http_client_coro/write_func_2.phpt b/tests/swoole_http_client_coro/write_func_2.phpt
@@ -6,6 +6,8 @@ swoole_http_client_coro: write func 1
 <?php
 require __DIR__ . '/../include/bootstrap.php';
 
+use Swoole\Event;
+
 const N = 16;
 $chunks = [];
 $n = N;
@@ -35,6 +37,9 @@ $pm->parentFunc = function ($pid) use ($pm, $chunks) {
 
 $pm->childFunc = function () use ($pm, $chunks) {
     Co\run(function () use ($pm, $chunks) {
+        Event::defer(function () use ($pm) {
+            $pm->wakeup();
+        });
         $server = new Swoole\Coroutine\Http\Server('127.0.0.1', $pm->getFreePort());
         $server->handle('/', function ($req, $resp) use ($server, $chunks) {
             foreach ($chunks as $chunk) {