Added test cases for cookies

Author: limingxinleo

examples/http_server_v2.php

@@ -9,11 +9,14 @@
  * @contact  group@hyperf.io
  * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
  */
+use Hyperf\Codec\Json;
 use Hyperf\Contract\StdoutLoggerInterface;
 use Hyperf\Engine\Coroutine;
 use Hyperf\Engine\Http\ServerFactory;
 use Hyperf\Engine\Http\Stream;
 use Hyperf\Engine\ResponseEmitter;
+use Hyperf\HttpMessage\Cookie\Cookie;
+use Hyperf\HttpMessage\Server\Request;
 use Hyperf\HttpMessage\Server\Response;
 use Psr\Http\Message\RequestInterface;
 use Swoole\Http\Response as SwooleResponse;
@@ -35,12 +38,34 @@
     $server = (new ServerFactory($logger))->make('0.0.0.0', 9505);
 
     $server->handle(static function (RequestInterface $request, SwooleResponse $response) use ($emitter) {
-        $body = (string) $request->getBody();
-        $ret = 'Hello World.';
-        if ($body) {
-            $ret = 'Received: ' . $body;
-        }
-        $emitter->emit((new Response())->withBody(new Stream($ret)), $response);
+        $path = $request->getUri()->getPath();
+
+        $callback = match ($path) {
+            '/set-cookies' => function (Request $request) {
+                $cookies = $request->getCookieParams();
+                $json = Json::decode((string) $request->getBody());
+
+                $response = new Response();
+                if (! empty($json['id'])) {
+                    $response->setCookie(new Cookie('id', $json['id']));
+                }
+                if (! empty($json['id2'])) {
+                    $response->setCookie(new Cookie('id2', $json['id2']));
+                }
+
+                return $response->withBody(new Stream(Json::encode($cookies)));
+            },
+            default => function (RequestInterface $request) {
+                $body = (string) $request->getBody();
+                $ret = 'Hello World.';
+                if ($body) {
+                    $ret = 'Received: ' . $body;
+                }
+
+                return (new Response())->withBody(new Stream($ret));
+            }
+        };
+        $emitter->emit($callback($request), $response);
     });
 
     $server->start();

tests/Cases/HttpServerTest.php

@@ -12,6 +12,7 @@
 
 namespace HyperfTest\Cases;
 
+use Hyperf\Codec\Json;
 use Hyperf\Engine\Http\Client;
 
 /**
@@ -45,4 +46,30 @@ public function testHttpServerReceived()
             $this->assertSame('Received: Hyperf', $response->body);
         });
     }
+
+    /**
+     * @group Server
+     */
+    public function testHttpServerCookies()
+    {
+        $this->runInCoroutine(function () {
+            $client = new Client('127.0.0.1', 9505);
+
+            $client->setCookies(['key' => 'value']);
+
+            $response = $client->request('POST', '/set-cookies', ['user_id' => uniqid()], Json::encode(['id' => $id = uniqid()]));
+            $this->assertSame(200, $response->statusCode);
+            $this->assertSame(1, count($response->getHeaders()['set-cookie']));
+            $this->assertStringStartsWith('id=' . $id, $response->getHeaders()['set-cookie'][0]);
+            $json = Json::decode((string) $response->getBody());
+            $this->assertSame(['key' => 'value'], $json);
+
+            $response = $client->request('POST', '/set-cookies', [], Json::encode(['id2' => $id2 = uniqid()]));
+            $this->assertSame(200, $response->statusCode);
+            $this->assertSame(1, count($response->getHeaders()['set-cookie']));
+            $this->assertStringStartsWith('id2=' . $id2, $response->getHeaders()['set-cookie'][0]);
+            $json = Json::decode((string) $response->getBody());
+            $this->assertSame(['key' => 'value', 'id' => $id], $json);
+        });
+    }
 }