[HttpClient] fix canceling responses in a streaming loop

Author: Kyra Farrow

Response/MockResponse.php

@@ -78,6 +78,15 @@ public function getInfo(string $type = null)
         return null !== $type ? $this->info[$type] ?? null : $this->info;
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function cancel(): void
+    {
+        $this->info['error'] = 'Response has been canceled.';
+        $this->body = null;
+    }
+
     /**
      * {@inheritdoc}
      */
@@ -150,8 +159,11 @@ protected static function perform(ClientState $multi, array &$responses): void
         foreach ($responses as $response) {
             $id = $response->id;
 
-            if (!$response->body) {
-                // Last chunk
+            if (null === $response->body) {
+                // Canceled response
+                $response->body = [];
+            } elseif ([] === $response->body) {
+                // Error chunk
                 $multi->handlesActivity[$id][] = null;
                 $multi->handlesActivity[$id][] = null !== $response->info['error'] ? new TransportException($response->info['error']) : null;
             } elseif (null === $chunk = array_shift($response->body)) {
@@ -242,7 +254,7 @@ private static function readResponse(self $response, array $options, ResponseInt
 
         // populate info related to headers
         $info = $mock->getInfo() ?: [];
-        $response->info['http_code'] = ($info['http_code'] ?? 0) ?: $mock->getStatusCode(false) ?: 200;
+        $response->info['http_code'] = ($info['http_code'] ?? 0) ?: $mock->getStatusCode() ?: 200;
         $response->addResponseHeaders($info['response_headers'] ?? [], $response->info, $response->headers);
         $dlSize = isset($response->headers['content-encoding']) ? 0 : (int) ($response->headers['content-length'][0] ?? 0);
 

Response/ResponseTrait.php

@@ -327,7 +327,7 @@ public static function stream(iterable $responses, float $timeout = null): \Gene
 
                     unset($multi->handlesActivity[$j]);
 
-                    if ($chunk instanceof FirstChunk && null === $response->initializer) {
+                    if ($chunk instanceof FirstChunk && null === $response->initializer && null === $response->info['error']) {
                         // Ensure the HTTP status code is always checked
                         $response->getHeaders(true);
                     } elseif ($chunk instanceof ErrorChunk && !$chunk->didThrow()) {