Added test cases for response proxy trait.

Author: limingxinleo

src/Server/ResponseProxyTrait.php

@@ -11,6 +11,7 @@
  */
 namespace Hyperf\HttpMessage\Server;
 
+use Hyperf\HttpMessage\Cookie\Cookie;
 use Psr\Http\Message\ResponseInterface;
 use Psr\Http\Message\StreamInterface;
 use RuntimeException;
@@ -27,6 +28,14 @@ public function setResponse(ResponseInterface $response)
         $this->response = $response;
     }
 
+    public function getResponse(): ResponseInterface
+    {
+        if (! $this->response instanceof ResponseInterface) {
+            throw new RuntimeException('response is invalid.');
+        }
+        return $this->response;
+    }
+
     public function getProtocolVersion()
     {
         return $this->getResponse()->getProtocolVersion();
@@ -103,11 +112,46 @@ public function getReasonPhrase()
         return $this->getResponse()->getReasonPhrase();
     }
 
-    public function getResponse(): ResponseInterface
+    /**
+     * Returns an instance with specified cookies.
+     */
+    public function withCookie(Cookie $cookie): self
     {
-        if (! $this->response instanceof ResponseInterface) {
-            throw new RuntimeException('response is invalid.');
-        }
-        return $this->response;
+        $this->setResponse($this->getResponse()->withCookie($cookie));
+        return $this;
+    }
+
+    /**
+     * Retrieves all cookies.
+     */
+    public function getCookies(): array
+    {
+        return $this->getResponse()->getCookies();
+    }
+
+    /**
+     * Returns an instance with specified trailer.
+     * @param string $value
+     */
+    public function withTrailer(string $key, $value): self
+    {
+        $this->setResponse($this->getResponse()->withTrailer($key, $value));
+        return $this;
+    }
+
+    /**
+     * Retrieves a specified trailer value, returns null if the value does not exists.
+     */
+    public function getTrailer(string $key)
+    {
+        return $this->getResponse()->getTrailer($key);
+    }
+
+    /**
+     * Retrieves all trailers values.
+     */
+    public function getTrailers(): array
+    {
+        return $this->getResponse()->getTrailers();
     }
 }

tests/ResponseProxyTest.php

@@ -0,0 +1,49 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace HyperfTest\HttpMessage;
+
+use Hyperf\HttpMessage\Server\Response;
+use HyperfTest\HttpMessage\Stub\Server\ResponseStub;
+
+/**
+ * @internal
+ * @coversNothing
+ */
+class ResponseProxyTest extends ResponseTest
+{
+    protected function tearDown()
+    {
+        parent::tearDown(); // TODO: Change the autogenerated stub
+    }
+
+    public function testStatusCode()
+    {
+        parent::testStatusCode(); // TODO: Change the autogenerated stub
+    }
+
+    public function testHeaders()
+    {
+        parent::testHeaders(); // TODO: Change the autogenerated stub
+    }
+
+    public function testCookies()
+    {
+        parent::testCookies(); // TODO: Change the autogenerated stub
+    }
+
+    protected function newResponse()
+    {
+        $response = new ResponseStub();
+        $response->setResponse(new Response());
+        return $response;
+    }
+}

tests/ResponseTest.php

@@ -0,0 +1,59 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace HyperfTest\HttpMessage;
+
+use Hyperf\HttpMessage\Cookie\Cookie;
+use Hyperf\HttpMessage\Server\Response;
+use PHPUnit\Framework\TestCase;
+
+/**
+ * @internal
+ * @coversNothing
+ */
+class ResponseTest extends TestCase
+{
+    protected function tearDown()
+    {
+        \Mockery::close();
+    }
+
+    public function testStatusCode()
+    {
+        $response = $this->newResponse();
+        $this->assertSame(200, $response->getStatusCode());
+        $this->assertSame(201, $response->withStatus(201)->getStatusCode());
+    }
+
+    public function testHeaders()
+    {
+        $response = $this->newResponse();
+        $this->assertSame([], $response->getHeaders());
+        $response = $response->withHeader('Server', 'Hyperf');
+        $this->assertSame(['Server' => ['Hyperf']], $response->getHeaders());
+        $this->assertSame(['Hyperf'], $response->getHeader('Server'));
+        $this->assertSame('Hyperf', $response->getHeaderLine('Server'));
+    }
+
+    public function testCookies()
+    {
+        $cookie = new Cookie('test', uniqid(), 3600, '/', 'hyperf.io');
+        $response = $this->newResponse();
+        $this->assertSame([], $response->getCookies());
+        $response = $response->withCookie($cookie);
+        $this->assertSame(['hyperf.io' => ['/' => ['test' => $cookie]]], $response->getCookies());
+    }
+
+    protected function newResponse()
+    {
+        return new Response();
+    }
+}

tests/Stub/Server/ResponseStub.php

@@ -0,0 +1,20 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace HyperfTest\HttpMessage\Stub\Server;
+
+use Hyperf\HttpMessage\Server\ResponseProxyTrait;
+use Psr\Http\Message\ResponseInterface;
+
+class ResponseStub implements ResponseInterface
+{
+    use ResponseProxyTrait;
+}