Merge 4e5d88a into 7561382

zendframework · Jul 19, 2018 · b58b338 · b58b338
2 parents 7561382 + 4e5d88a
commit b58b338
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 11 deletions.
diff --git a/src/Response.php b/src/Response.php
@@ -113,7 +113,7 @@ class Response implements ResponseInterface
     /**
      * @var string
      */
-    private $reasonPhrase = '';
+    private $reasonPhrase;
 
     /**
      * @var int
@@ -146,12 +146,6 @@ public function getStatusCode()
      */
     public function getReasonPhrase()
     {
-        if (! $this->reasonPhrase
-            && isset($this->phrases[$this->statusCode])
-        ) {
-            $this->reasonPhrase = $this->phrases[$this->statusCode];
-        }
-
         return $this->reasonPhrase;
     }
 
@@ -161,18 +155,18 @@ public function getReasonPhrase()
     public function withStatus($code, $reasonPhrase = '')
     {
         $new = clone $this;
-        $new->setStatusCode($code);
-        $new->reasonPhrase = $reasonPhrase;
+        $new->setStatusCode($code, $reasonPhrase);
         return $new;
     }
 
     /**
      * Set a valid status code.
      *
      * @param int $code
+     * @param string $reasonPhrase
      * @throws InvalidArgumentException on an invalid status code.
      */
-    private function setStatusCode($code)
+    private function setStatusCode($code, $reasonPhrase = '')
     {
         if (! is_numeric($code)
             || is_float($code)
@@ -181,11 +175,24 @@ private function setStatusCode($code)
         ) {
             throw new InvalidArgumentException(sprintf(
                 'Invalid status code "%s"; must be an integer between %d and %d, inclusive',
-                (is_scalar($code) ? $code : gettype($code)),
+                is_scalar($code) ? $code : gettype($code),
                 static::MIN_STATUS_CODE_VALUE,
                 static::MAX_STATUS_CODE_VALUE
             ));
         }
+
+        if (! is_string($reasonPhrase)) {
+            throw new InvalidArgumentException(sprintf(
+                'Unsupported response reason phrase; must be a string, received %s',
+                is_object($reasonPhrase) ? get_class($reasonPhrase) : gettype($reasonPhrase)
+            ));
+        }
+
+        if ($reasonPhrase === '' && isset($this->phrases[$code])) {
+            $reasonPhrase = $this->phrases[$code];
+        }
+
+        $this->reasonPhrase = $reasonPhrase;
         $this->statusCode = $code;
     }
 }
diff --git a/test/ResponseTest.php b/test/ResponseTest.php
@@ -112,6 +112,29 @@ public function testCanSetCustomReasonPhrase()
         $this->assertSame('Foo Bar!', $response->getReasonPhrase());
     }
 
+    public function invalidReasonPhrases()
+    {
+        return [
+            'true' => [ true ],
+            'false' => [ false ],
+            'array' => [ [ 200 ] ],
+            'object' => [ (object) [ 'reasonPhrase' => 'Ok' ] ],
+            'integer' => [99],
+            'float' => [400.5],
+            'null' => [null],
+        ];
+    }
+
+    /**
+     * @dataProvider invalidReasonPhrases
+     */
+    public function testWithStatusRaisesAnExceptionForNonStringReasonPhrases($invalidReasonPhrase)
+    {
+        $this->expectException(InvalidArgumentException::class);
+
+        $this->response->withStatus(422, $invalidReasonPhrase);
+    }
+
     public function testConstructorRaisesExceptionForInvalidStream()
     {
         $this->expectException(InvalidArgumentException::class);