Allow return a Promise in onRejected then

Author: ezimuel

spec/Promise/HttpRejectedPromiseSpec.php

@@ -4,6 +4,7 @@
 
 use Http\Client\Exception;
 use Http\Client\Exception\TransferException;
+use Http\Client\Promise\HttpFulfilledPromise;
 use Http\Promise\Promise;
 use PhpSpec\ObjectBehavior;
 use Prophecy\Argument;
@@ -87,4 +88,19 @@ function it_throws_not_http_exception()
             throw new \Exception();
         });
     }
+
+    function it_returns_a_promise_when_rejected(ResponseInterface $response)
+    {
+        $exception = new TransferException();
+        $this->beConstructedWith($exception);
+
+        $promise = $this->then(null, function (Exception $exceptionReceived) use($exception, $response) {
+            return new HttpFulfilledPromise($response->getWrappedObject());
+        });
+
+        $promise->shouldHaveType('Http\Promise\Promise');
+        $promise->shouldHaveType('Http\Client\Promise\HttpFulfilledPromise');
+        $promise->getState()->shouldReturn(Promise::FULFILLED);
+        $promise->wait()->shouldReturnAnInstanceOf('Psr\Http\Message\ResponseInterface');
+    }
 }

src/Promise/HttpRejectedPromise.php

@@ -4,6 +4,7 @@
 
 use Http\Client\Exception;
 use Http\Promise\Promise;
+use Psr\Http\Message\ResponseInterface;
 
 final class HttpRejectedPromise implements Promise
 {
@@ -27,7 +28,11 @@ public function then(callable $onFulfilled = null, callable $onRejected = null)
         }
 
         try {
-            return new HttpFulfilledPromise($onRejected($this->exception));
+            $result = $onRejected($this->exception);
+            if ($result instanceof Promise) {
+                return $result;
+            }
+            return new HttpFulfilledPromise($result);
         } catch (Exception $e) {
             return new self($e);
         }